Searched refs:GetOutputDetailShape (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | layer_norm_grad_split.cc | 50 auto shapes = {AnfAlgo::GetOutputDetailShape(layer_norm_grad, 0), in CreateOutputsOfLayerNormXBackpropV2()
82 auto shapes = {AnfAlgo::GetOutputDetailShape(layer_norm_grad, kLayerNormGradOutputGammaIndex), in CreateOutputsOfLayerNormBetaGammaBackpropV2()
83 AnfAlgo::GetOutputDetailShape(layer_norm_grad, kLayerNormGradOutputBetaIndex)}; in CreateOutputsOfLayerNormBetaGammaBackpropV2()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | insert_cast.cc | 50 const auto origin_shape = AnfAlgo::GetOutputDetailShape(cnode, output_idx); in InsertCastForMultipleOutput()
96 const abstract::BaseShapePtr origin_shape = AnfAlgo::GetOutputDetailShape(cnode, 0); in InsertCastForOutput()
|
| D | deal_ref_and_split_unsupported_transdata.cc | 119 auto detail_shape = AnfAlgo::GetOutputDetailShape(cnode, output_index); in AddAdditionalToRefOutput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | tbe_json_utils.cc | 100 auto out_shape = AnfAlgo::GetOutputDetailShape(anf_node, real_idx); in GetOutputOriShapeForTbeBuild()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 271 auto out_shape_base = AnfAlgo::GetOutputDetailShape(input, 0); in NewTransOpNode()
444 …const abstract::BaseShapePtr origin_shape = AnfAlgo::GetOutputDetailShape(prev_node.first, prev_no… in InsertCastForInput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.h | 211 static abstract::BaseShapePtr GetOutputDetailShape(const AnfNodePtr &node, size_t output_idx);
|
| D | anf_runtime_algorithm.cc | 858 auto output_shape = GetOutputDetailShape(node, output_idx); in GetOutputDeviceShapeForTbeBuild()
1164 abstract::BaseShapePtr AnfRuntimeAlgorithm::GetOutputDetailShape(const AnfNodePtr &node, size_t out… in GetOutputDetailShape() function in mindspore::session::AnfRuntimeAlgorithm
1202 …return AnfRuntimeAlgorithm::GetOutputDetailShape(kernel_with_index.first, kernel_with_index.second… in GetPrevNodeOutputDetailShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_build.cc | 136 auto output_shape = AnfAlgo::GetOutputDetailShape(anf_node, real_index); in GetOutputShapeForTbeBuild()
|