Home
last modified time | relevance | path

Searched defs:input_dimension (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/src/delegate/npu/
Dnpu_executor.cc116 std::vector<hiai::TensorDimension> input_dimension; in GetIOTensorVec() local
142 int NPUExecutor::UpdateInputTensorVec(const std::vector<hiai::TensorDimension> &input_dimension) { in UpdateInputTensorVec()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Ddepthtospace_cpu_kernel.cc47 size_t input_dimension = input_shape.size(); in Launch() local
Dspacetodepth_cpu_kernel.cc58 size_t input_dimension = input_shape.size(); in Launch() local