Home
last modified time | relevance | path

Searched refs:SetInputBlock (Results 1 – 2 of 2) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dbatch_to_space_nd_builder.h44 OH_NN_ReturnCode SetInputBlock(const std::shared_ptr<NNTensor>& tensor);
51 {OH_NN_BATCH_TO_SPACE_ND_BLOCKSIZE, &BatchToSpaceNDBuilder::SetInputBlock},
H A Dbatch_to_space_nd_builder.cpp32 OH_NN_ReturnCode BatchToSpaceNDBuilder::SetInputBlock(const std::shared_ptr<NNTensor>& tensor) in SetInputBlock() function in OHOS::NeuralNetworkRuntime::Ops::BatchToSpaceNDBuilder