| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | ascend_inference_session.cc | 147 vector<size_t> trans_input; in CompareInput() local
148 (void)std::transform(input_shape.begin(), input_shape.end(), std::back_inserter(trans_input), in CompareInput()
153 …if ((!is_scalar_shape(trans_input) || !is_scalar_shape(parameter_shape)) && (trans_input != parame… in CompareInput()
154 …S_LOG(ERROR) << "Input shape is inconsistent. The actual shape is " << PrintInputShape(trans_input) in CompareInput()
|
| D | gpu_inference_session.cc | 147 vector<size_t> trans_input; in CompareInput() local
148 (void)std::transform(input_shape.begin(), input_shape.end(), std::back_inserter(trans_input), in CompareInput()
153 …if ((!is_scalar_shape(trans_input) || !is_scalar_shape(parameter_shape)) && (trans_input != parame… in CompareInput()
154 …S_LOG(ERROR) << "Input shape is inconsistent. The actual shape is " << PrintInputShape(trans_input) in CompareInput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_winograd_fp32.c | 48 float *trans_input = buffer_list[0]; in ConvWinogardFp32() local
68 … WinogradInputTransform(input_data + in_batch_offset, trans_input + task_id * trans_input_offset, in ConvWinogardFp32()
72 float *src_ptr = trans_input + task_id * trans_input_offset; in ConvWinogardFp32()
|
| D | winograd_transform.h | 31 void WinogradInputTransform(const float *input_data, float *trans_input, float *tmp_data, int cal_n…
|
| D | winograd_transform.c | 21 void WinogradInputTransform(const float *input_data, float *trans_input, float *tmp_data, int cal_n… in WinogradInputTransform() argument
91 float *trans_input_ptr = trans_input + dst_ic4_offset; in WinogradInputTransform()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | split_n_optimizer.cc | 107 auto trans_input = AnfAlgo::VisitKernel(in_node, 0).first; in InputCheck() local
108 MS_EXCEPTION_IF_NULL(trans_input); in InputCheck()
109 …if (in_node_name == kTransDataOpName && (trans_input->isa<Parameter>() || trans_input->isa<ValueNo… in InputCheck()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | infershape_pass.cc | 317 auto trans_input = trans_cnode->input(1); in SetSubGraphOutput() local
318 MS_ASSERT(trans_input != nullptr); in SetSubGraphOutput()
319 auto trans_input_name = trans_input->fullname_with_scope(); in SetSubGraphOutput()
320 if (utils::isa<ParameterPtr>(trans_input)) { in SetSubGraphOutput()
321 trans_input->cast<ParameterPtr>()->set_name(node_name); in SetSubGraphOutput()
322 } else if (utils::isa<CNodePtr>(trans_input)) { in SetSubGraphOutput()
323 trans_input->cast<CNodePtr>()->set_fullname_with_scope(node_name); in SetSubGraphOutput()
|
| D | decrease_transpose_algo.cc | 566 auto trans_input = trans_cnode->input(1); in SetSubGraphOutput() local
567 auto trans_input_name = trans_input->fullname_with_scope(); in SetSubGraphOutput()
568 if (utils::isa<ParameterPtr>(trans_input)) { in SetSubGraphOutput()
569 trans_input->cast<ParameterPtr>()->set_name(node_name); in SetSubGraphOutput()
570 } else if (utils::isa<CNodePtr>(trans_input)) { in SetSubGraphOutput()
571 trans_input->cast<CNodePtr>()->set_fullname_with_scope(node_name); in SetSubGraphOutput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | winograd_transform_fp16.h | 32 void WinogradInputTransformFp16(const float16_t *input_data, float16_t *trans_input, float16_t *tmp…
|
| D | conv_fp16.c | 209 float16_t *trans_input = buffer_list[0]; in ConvWinogardFp16() local
229 …WinogradInputTransformFp16(input_data + in_batch_offset, trans_input + task_id * trans_input_offse… in ConvWinogardFp16()
233 float16_t *src_ptr = trans_input + task_id * trans_input_offset; in ConvWinogardFp16()
|
| D | winograd_transform_fp16.c | 20 void WinogradInputTransformFp16(const float16_t *input_data, float16_t *trans_input, float16_t *tmp… in WinogradInputTransformFp16() argument
121 float16_t *trans_input_ptr = trans_input + dst_ic8_offset; in WinogradInputTransformFp16()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/format/ |
| D | to_format_base.cc | 32 AnfNodePtr trans_input = before ? cnode->input(index) : cnode; in GenNewInput() local
35 auto trans_cnode = opt::GenTransposeNode(func_graph, trans_input, perm, trans_name); in GenNewInput()
44 auto abstract = trans_input->abstract(); in GenNewInput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | conv3x3_int8.c | 781 void Conv3x3Int8InputTransform(const int16_t *input_data, int16_t *trans_input, int16_t *tmp_data, … in Conv3x3Int8InputTransform() argument
820 int16_t *trans_input_ptr = trans_input + dst_ic8_offset; in Conv3x3Int8InputTransform()
|