Searched refs:GetOutputReshapeType (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | insert_cast.cc | 66 origin_type, AnfAlgo::GetOutputReshapeType(getitem, 0)); in InsertCastForMultipleOutput()
102 origin_type, AnfAlgo::GetOutputReshapeType(cnode, 0)); in InsertCastForOutput()
|
| D | insert_transpose_for_dynamic_gru_v2.cc | 55 auto padding_axis = AnfAlgo::GetOutputReshapeType(transdata_node, 0); in Insert()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_build_info.h | 69 std::string GetOutputReshapeType(size_t input_index) const;
143 …kernel_build_info_->output_reshape_type_.emplace_back(kernel_build_info->GetOutputReshapeType(inde… in KernelBuildInfoBuilder()
|
| D | kernel_build_info.cc | 90 std::string KernelBuildInfo::GetOutputReshapeType(size_t output_index) const { in GetOutputReshapeType() function in mindspore::kernel::KernelBuildInfo
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | transdata_split.cc | 72 auto padding_axis = AnfAlgo::GetOutputReshapeType(node, 0); in DoSplit()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | output_actor.cc | 35 tensor->set_padding_type(AnfAlgo::GetOutputReshapeType(output_node, output_index)); in CreateOutputTensor()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 203 : AnfAlgo::GetOutputReshapeType(node, insert_index); in AddTransOpNodeToGraph()
290 auto padding_axis = AnfAlgo::GetOutputReshapeType(input, 0); in NewTransOpNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 650 shape = trans::PaddingShape(shape, format, AnfAlgo::GetOutputReshapeType(node, output_index)); in GetOutputTensorMemSize()
803 return GetOutputReshapeType(kernel_with_index.first, kernel_with_index.second); in GetPrevNodeOutputReshapeType()
871 infer_shape = trans::PaddingShape(infer_shape, format, GetOutputReshapeType(node, output_idx)); in GetOutputDeviceShapeForTbeBuild()
884 infer_shape = trans::PaddingShape(infer_shape, format, GetOutputReshapeType(node, output_idx)); in GetOutputDeviceShape()
923 std::string AnfRuntimeAlgorithm::GetOutputReshapeType(const AnfNodePtr &node, size_t output_idx) { in GetOutputReshapeType() function in mindspore::session::AnfRuntimeAlgorithm
940 return build_info->GetOutputReshapeType(output_idx); in GetOutputReshapeType()
|
| D | anf_runtime_algorithm.h | 174 static std::string GetOutputReshapeType(const AnfNodePtr &node, size_t output_idx);
|
| D | session_basic.cc | 230 tensor->set_padding_type(AnfAlgo::GetOutputReshapeType(node, output_index)); in CreateNodeOutputTensor()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 206 tensor->set_padding_type(AnfAlgo::GetOutputReshapeType(output_node, output_index)); in CreateOutputTensor()
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.cc | 911 … host_shape = trans::PaddingShape(host_shape, format, AnfAlgo::GetOutputReshapeType(node, index)); in GetRuntimePaddingShape()
|