| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | replace_momentum_cast_fusion.cc | 47 std::vector<TypeId> outputs_type; in Process() local
51 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(node, i)); in Process()
54 outputs_type[kGradIndex] = AnfAlgo::GetPrevNodeOutputInferDataType(grad_cast, 0); in Process()
56 AnfAlgo::SetOutputInferTypeAndShape(outputs_type, outputs_shape, node.get()); in Process()
|
| D | replace_addn_fusion.cc | 43 std::vector<TypeId> outputs_type; in Process() local
45 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(A, 0)); in Process()
47 AnfAlgo::SetOutputInferTypeAndShape(outputs_type, outputs_shape, add_new.get()); in Process()
|
| D | add_relu_v2_fusion.cc | 34 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
44 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(node, output_index)); in GenerateKernelBuildInfo()
49 builder.SetOutputsDeviceType(outputs_type); in GenerateKernelBuildInfo()
|
| D | matmul_biasadd_fusion.cc | 34 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
44 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(node, output_index)); in GenerateKernelBuildInfo()
49 builder.SetOutputsDeviceType(outputs_type); in GenerateKernelBuildInfo()
|
| D | add_relu_grad_v2_fusion.cc | 34 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
44 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(node, output_index)); in GenerateKernelBuildInfo()
49 builder.SetOutputsDeviceType(outputs_type); in GenerateKernelBuildInfo()
|
| D | batch_norm_relu_fusion.cc | 80 std::vector<TypeId> outputs_type; in Process() local
84 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(batch_norm, i)); in Process()
87 …AnfAlgo::SetOutputInferTypeAndShape(outputs_type, outputs_shape, fused_batch_norm_with_relu.get()); in Process()
|
| D | batch_norm_add_relu_fusion.cc | 85 std::vector<TypeId> outputs_type; in Process() local
89 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(batch_norm, i)); in Process()
92 …AnfAlgo::SetOutputInferTypeAndShape(outputs_type, outputs_shape, fused_batch_norm_with_add_relu.ge… in Process()
|
| D | post_batch_norm_add_relu_fusion.cc | 85 std::vector<TypeId> outputs_type; in Process() local
89 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(batch_norm, i)); in Process()
92 …AnfAlgo::SetOutputInferTypeAndShape(outputs_type, outputs_shape, fused_batch_norm_with_add_relu.ge… in Process()
|
| D | batch_norm_relu_grad_fusion.cc | 97 std::vector<TypeId> outputs_type; in Process() local
101 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(node, i)); in Process()
104 …AnfAlgo::SetOutputInferTypeAndShape(outputs_type, outputs_shape, fused_batch_norm_grad_with_relu.g… in Process()
|
| D | batch_norm_add_relu_grad_fusion.cc | 70 std::vector<TypeId> outputs_type; in SetShapeAndType() local
74 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(bn_grad, i)); in SetShapeAndType()
78 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(relu_grad, 0)); in SetShapeAndType()
80 AnfAlgo::SetOutputInferTypeAndShape(outputs_type, outputs_shape, bn_add_relu_grad.get()); in SetShapeAndType()
|
| D | relu_v2_pass.cc | 47 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
57 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(node, output_index)); in GenerateKernelBuildInfo()
62 builder.SetOutputsDeviceType(outputs_type); in GenerateKernelBuildInfo()
|
| D | adam_fusion.cc | 34 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
45 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(node, output_index)); in GenerateKernelBuildInfo()
50 builder.SetOutputsDeviceType(outputs_type); in GenerateKernelBuildInfo()
|
| D | adam_weight_decay_fusion.cc | 34 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
45 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(node, output_index)); in GenerateKernelBuildInfo()
50 builder.SetOutputsDeviceType(outputs_type); in GenerateKernelBuildInfo()
|
| D | print_reduce_fusion.cc | 33 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
44 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(node, output_index)); in GenerateKernelBuildInfo()
50 builder.SetOutputsDeviceType(outputs_type); in GenerateKernelBuildInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel_metadata.cc | 95 std::vector<TypeId> outputs_type; in HcclMetadataInfo() local
100 outputs_type.push_back(recv_type); in HcclMetadataInfo()
102 outputs_type.push_back(type); in HcclMetadataInfo()
109 builder.SetOutputsDeviceType(outputs_type); in HcclMetadataInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | host_kernel_metadata.cc | 49 std::vector<TypeId> outputs_type; in HostMetadataInfo() local
53 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(kernel_node, output_index)); in HostMetadataInfo()
59 builder.SetOutputsDeviceType(outputs_type); in HostMetadataInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | reorder_ops.cc | 100 if (node_io_info.outputs_type.empty()) { in SetNodeInfo()
107 …w_abstract = std::make_shared<abstract::AbstractTensor>(TypeIdToType(node_io_info.outputs_type[0]), in SetNodeInfo()
111 …w_abstract = std::make_shared<abstract::AbstractTensor>(TypeIdToType(node_io_info.outputs_type[0]), in SetNodeInfo()
125 info_builder.SetOutputsDeviceType(node_io_info.outputs_type); in SetNodeInfo()
241 cast_io_info.outputs_type.push_back(small_type); in ReorderTypeInsensitiveCastDown()
250 type_insens_io_info.outputs_type.push_back(small_type); in ReorderTypeInsensitiveCastDown()
307 type_insens_io_info.outputs_type.push_back(small_type); in ReorderCastUpTypeInsensitive()
316 cast_io_info.outputs_type.push_back(large_type); in ReorderCastUpTypeInsensitive()
|
| D | reorder_ops.h | 31 std::vector<TypeId> outputs_type; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_metadata.cc | 65 std::vector<TypeId> outputs_type; in AicpuMetadataInfoForSpecialNodes() local
69 (void)outputs_type.emplace_back(AnfAlgo::GetOutputInferDataType(kernel_node, output_index)); in AicpuMetadataInfoForSpecialNodes()
75 builder.SetOutputsDeviceType(outputs_type); in AicpuMetadataInfoForSpecialNodes()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.cc | 345 const std::vector<TypeId> &outputs_type) { in PrintUnsupportedTypeException() argument
351 std::for_each(std::begin(outputs_type), std::end(outputs_type), in PrintUnsupportedTypeException()
414 std::vector<TypeId> outputs_type; in SetKernelInfo() local
418 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(kernel_node, output_index)); in SetKernelInfo()
429 builder->SetOutputsDeviceType(outputs_type); in SetKernelInfo()
445 PrintUnsupportedTypeException(kernel_node, inputs_type, outputs_type); in SetKernelInfo()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | node_info.cc | 201 std::vector<TypePtr> outputs_type; in ExtractOutputTypeByNode() local
212 outputs_type.push_back(ele_element_type); in ExtractOutputTypeByNode()
221 outputs_type.push_back(element_type); in ExtractOutputTypeByNode()
226 return outputs_type; in ExtractOutputTypeByNode()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/ |
| D | cpu_device_context.cc | 155 std::vector<TypeId> outputs_type; in SetControlOpInfo() local
159 outputs_type.push_back(AnfAlgo::GetOutputInferDataType(kernel_node, output_index)); in SetControlOpInfo()
166 builder->SetOutputsDeviceType(outputs_type); in SetControlOpInfo()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_auto_parallel.cc | 311 auto outputs_type = ExtractOutputTypeByNode(cnode); in CreateTheOperatorInfo() local
313 outputs_type_length.reserve(outputs_type.size()); in CreateTheOperatorInfo()
314 std::transform(outputs_type.begin(), outputs_type.end(), std::back_inserter(outputs_type_length), in CreateTheOperatorInfo()
320 if (operator_info->set_outputs_type(outputs_type) != SUCCESS) { in CreateTheOperatorInfo()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.h | 82 Status set_outputs_type(const std::vector<TypePtr> &outputs_type);
83 const std::vector<TypePtr> &outputs_type() const { return outputs_type_; } in outputs_type() function
|
| D | operator_info.cc | 1683 Status OperatorInfo::set_outputs_type(const std::vector<TypePtr> &outputs_type) { in set_outputs_type() argument
1684 if (outputs_type.size() != outputs_shape_.size()) { in set_outputs_type()
1685 MS_LOG(ERROR) << "Outputs type: " << outputs_type.size() in set_outputs_type()
1689 outputs_type_ = outputs_type; in set_outputs_type()
|