Home
last modified time | relevance | path

Searched defs:GetOutputInferShape (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dcallback_impl.cc41 ShapeVector CallbackImpl::GetOutputInferShape(const AnfNodePtr &node, size_t i) { in GetOutputInferShape() function in mindspore::graphkernel::CallbackImpl
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dclip_by_norm_fission.cc32 ShapeVector GetOutputInferShape(const AnfNodePtr &node) { in GetOutputInferShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dascend_clip_by_norm_fission.cc31 ShapeVector GetOutputInferShape(const AnfNodePtr &node) { in GetOutputInferShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/
Dcallback_impl.cc91 ShapeVector CallbackImpl::GetOutputInferShape(const AnfNodePtr &node, size_t i) { in GetOutputInferShape() function in mindspore::graphkernel::CallbackImpl
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dclip_by_norm_fission.cc33 ShapeVector GetOutputInferShape(const AnfNodePtr &node) { in GetOutputInferShape() function
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc322 TEST_F(AnfRuntimeAlgorithmTest, GetOutputInferShape) { in TEST_F() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Danfalgo.cc813 ShapeVector AnfAlgo::GetOutputInferShape(const AnfNodePtr &node, size_t output_idx, bool is_real_sq… in GetOutputInferShape() function in mindspore::common::AnfAlgo