Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/
Dnpu_executor.cc133 std::vector<hiai::TensorDimension> input_dimension; in GetIOTensorVec() local
159 int NPUExecutor::UpdateInputTensorVec(const std::vector<hiai::TensorDimension> &input_dimension) { in UpdateInputTensorVec()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Ddepthtospace_cpu_kernel.cc83 size_t input_dimension = input_shape.size(); in LaunchKernel() local
Dspacetodepth_cpu_kernel.cc86 size_t input_dimension = input_shape.size(); in LaunchKernel() local
Dcpu_kernel.cc431 int input_dimension = SizeToInt(multi_input.size()); in BroadcastShape() local