| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/ |
| D | graph_desc_reporter.cc | 43 element.data_shape_ = AnfAlgo::GetOutputDeviceShape(input_node, input_index); in ReportData()
53 element.data_shape_ = AnfAlgo::GetOutputDeviceShape(node, i); in ReportData()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | ascend_inference_session.cc | 143 auto parameter_shape = AnfAlgo::GetOutputDeviceShape(parameter, 0); in CompareInput()
202 …graph += std::to_string(i) + ": dims " + std::to_string(AnfAlgo::GetOutputDeviceShape(para, 0).siz… in InputsInfo()
203 ", shape " + PrintInputShape(AnfAlgo::GetOutputDeviceShape(para, 0)) + ", data type " + in InputsInfo()
|
| D | gpu_inference_session.cc | 143 auto parameter_shape = AnfAlgo::GetOutputDeviceShape(parameter, 0); in CompareInput()
202 …graph += std::to_string(i) + ": dims " + std::to_string(AnfAlgo::GetOutputDeviceShape(para, 0).siz… in InputsInfo()
203 ", shape " + PrintInputShape(AnfAlgo::GetOutputDeviceShape(para, 0)) + ", data type " + in InputsInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | addn_cpu_kernel.cc | 50 std::vector<size_t> dst_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel()
97 auto dst_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in CheckParam()
|
| D | pooling_cpu_kernel.cc | 39 std::vector<size_t> dst_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel()
|
| D | conv2d_grad_filter_cpu_kernel.cc | 37 std::vector<size_t> weight_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | remove_reshape_pair.cc | 50 auto output_shape = AnfAlgo::GetOutputDeviceShape(out_reshape, 0); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | remove_format_transform_pair.cc | 55 auto transpose2_output_shape = AnfAlgo::GetOutputDeviceShape(node, 0); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | tensoradd_cpu_kernel.cc | 35 output_shape_ = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel()
|
| D | depthtospace_cpu_kernel.cc | 31 output_shape_ = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel()
|
| D | spacetodepth_cpu_kernel.cc | 35 output_shape_ = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel()
|
| D | pack_cpu_kernel.cc | 44 auto output_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel()
|
| D | adam_delta_cpu_kernel.cc | 63 std::vector<size_t> delta_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | host_kernel_mod.cc | 64 std::vector<size_t> shape_i = AnfAlgo::GetOutputDeviceShape(anf_node, i); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_build.cc | 93 std::vector<size_t> shape_i = AnfAlgo::GetOutputDeviceShape(anf_node, i); in SetIOSize()
234 std::vector<size_t> output_shape = AnfAlgo::GetOutputDeviceShape(anf_node, output_index); in SetNodeOutputs()
344 std::vector<size_t> output_shape = AnfAlgo::GetOutputDeviceShape(anf_node, output_index); in SetExtInfoOutputShapeType()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/ |
| D | trt_kernel.cc | 41 auto output_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, j); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | unpack_gpu_kernel.h | 75 auto _shape = AnfAlgo::GetOutputDeviceShape(kernel_node, i); in Init()
|
| D | pack_gpu_kernel.h | 92 auto output_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in Init()
|
| D | meshgrid_gpu_kernel.h | 99 output_shape_ = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in Init()
|
| D | concatv2_gpu_kernel.h | 114 auto output_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/ |
| D | op_tiling_adapter.cc | 85 auto ms_shape = AnfAlgo::GetOutputDeviceShape(input_node, input_index); in ConvertInputShapeAndType()
115 auto ms_shape = AnfAlgo::GetOutputDeviceShape(node, i); in ConvertOutputShapeAndType()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | pad_gpu_kernel.h | 83 std::vector<size_t> output_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | all_to_all_v_calc_param.cc | 63 auto ms_shape = AnfAlgo::GetOutputDeviceShape(cnode, i); in CalcOpParam()
|
| /third_party/mindspore/mindspore/ccsrc/debug/data_dump/ |
| D | dump_utils.cc | 118 auto shape = AnfAlgo::GetOutputDeviceShape(node, index); in GetDumpIntShape()
|
| /third_party/mindspore/tests/ut/cpp/session/ |
| D | anf_runtime_algorithm_test.cc | 357 TEST_F(AnfRuntimeAlgorithmTest, GetOutputDeviceShape) { in TEST_F() argument
378 EXPECT_EQ(AnfAlgo::GetOutputDeviceShape(add, 0)[2], 224); in TEST_F()
379 EXPECT_EQ(AnfAlgo::GetOutputDeviceShape(add, 1)[0], 2); in TEST_F()
381 EXPECT_EQ(AnfAlgo::GetOutputDeviceShape(add, 2), expect_shape); in TEST_F()
383 EXPECT_EQ(AnfAlgo::GetOutputDeviceShape(add, 3), nz_expect_shape); in TEST_F()
|