Home
last modified time | relevance | path

Searched defs:GetInputShape (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/include/api/
Dcontext.h210 std::string Ascend310DeviceInfo::GetInputShape() const { return CharToString(GetInputShapeChar()); } in GetInputShape() function
/third_party/mindspore/include/api/
Dcontext.h418 std::string Ascend310DeviceInfo::GetInputShape() const { return CharToString(GetInputShapeChar()); } in GetInputShape() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Ddynamic_broadcast_gradient_args_kernel.cc123 std::vector<int64_t> GetInputShape(const CNodePtr &cnode, size_t index) { in GetInputShape() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dconv3d_grad_input_gpu_kernel.h312 void GetInputShape(const CNodePtr &kernel_node, std::vector<size_t> *input_shape) { in GetInputShape() function
Dconv2d_grad_input_gpu_kernel.h324 void GetInputShape(const CNodePtr &kernel_node, std::vector<size_t> *input_shape) { in GetInputShape() function
Dconv3d_transpose_gpu_kernel.h394 void GetInputShape(const CNodePtr &kernel_node, std::vector<size_t> *input_shape) { in GetInputShape() function
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dnode_infershape.cc176 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/akg/
Dakg_kernel_json_generator.cc148 std::vector<size_t> AkgKernelJsonGenerator::GetInputShape(const AnfNodePtr &anf_node, size_t real_i… in GetInputShape() function in mindspore::kernel::AkgKernelJsonGenerator