| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | dynamic_broadcast_gradient_args_kernel.cc | 123 std::vector<int64_t> GetInputShape(const CNodePtr &cnode, size_t index) { in GetInputShape() function
199 input_shapes[0] = GetInputShape(cnode, 0); in Execute()
200 input_shapes[1] = GetInputShape(cnode, 1); in Execute()
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/include/api/ |
| D | context.h | 143 inline std::string GetInputShape() const;
210 std::string Ascend310DeviceInfo::GetInputShape() const { return CharToString(GetInputShapeChar()); } in GetInputShape() function
|
| /third_party/mindspore/include/api/ |
| D | context.h | 329 inline std::string GetInputShape() const;
418 std::string Ascend310DeviceInfo::GetInputShape() const { return CharToString(GetInputShapeChar()); } in GetInputShape() function
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | node_infershape.h | 44 std::vector<int> GetInputShape(const CNodePtr &cnode, size_t index);
|
| D | transpose_strategy.cc | 61 auto shape = node_infer_shape->GetInputShape(cnode, 1); in JudgeIs4DInput()
64 shape = node_infer_shape->GetInputShape(cnode, kInputIndexTwo); in JudgeIs4DInput()
273 auto shape = node_infer_shape->GetInputShape(cnode, kInputIndexTwo); in ChangeOpSlice()
502 auto shape = node_infer_shape_.GetInputShape(base_node, input_index); in TransposeInsertDependOnShape()
|
| D | node_infershape.cc | 176 std::vector<int> NodeInferShape::GetInputShape(const CNodePtr &cnode, size_t index) { in GetInputShape() function in mindspore::opt::NodeInferShape
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | conv3d_grad_input_gpu_kernel.h | 97 GetInputShape(kernel_node, &input_shape); in Init()
312 void GetInputShape(const CNodePtr &kernel_node, std::vector<size_t> *input_shape) { in GetInputShape() function
|
| D | conv2d_grad_input_gpu_kernel.h | 135 GetInputShape(kernel_node, &input_shape); in Init()
324 void GetInputShape(const CNodePtr &kernel_node, std::vector<size_t> *input_shape) { in GetInputShape() function
|
| D | conv3d_transpose_gpu_kernel.h | 129 GetInputShape(kernel_node, &output_shape); in Init()
394 void GetInputShape(const CNodePtr &kernel_node, std::vector<size_t> *input_shape) { in GetInputShape() function
|
| /third_party/mindspore/tests/ut/cpp/cxx_api/ |
| D | context_test.cc | 104 ASSERT_EQ(ascend310->GetInputShape(), option_1); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_generator.h | 141 std::vector<size_t> GetInputShape(const AnfNodePtr &anf_node, size_t real_index) const;
|
| D | akg_kernel_json_generator.cc | 148 std::vector<size_t> AkgKernelJsonGenerator::GetInputShape(const AnfNodePtr &anf_node, size_t real_i… in GetInputShape() function in mindspore::kernel::AkgKernelJsonGenerator
205 auto input_shape = this->GetInputShape(anf_node, real_input_index); in CreateInputDescJson()
722 auto input_shape = this->GetInputShape(tmp_input.first, tmp_input.second.first); in CreateInputsJson()
|