Home
last modified time | relevance | path

Searched refs:inputs_type (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc202 bool IsNeedProcessFormatInfo(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type) { in IsNeedProcessFormatInfo() argument
219 if (inputs_type.size() == 0) { in IsNeedProcessFormatInfo()
247 void UpdateKernelFormatInfo(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type, in UpdateKernelFormatInfo() argument
258 auto cal_format = (inputs_type[0] == kNumberTypeFloat16) ? kOpFormat_NHWC : kOpFormat_NCHW; in UpdateKernelFormatInfo()
344 … PrintUnsupportedTypeException(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type, in PrintUnsupportedTypeException() argument
348 std::for_each(std::begin(inputs_type), std::end(inputs_type), in PrintUnsupportedTypeException()
407 std::vector<TypeId> inputs_type; in SetKernelInfo() local
411 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, input_index)); in SetKernelInfo()
421 if (IsNeedProcessFormatInfo(kernel_node, inputs_type)) { in SetKernelInfo()
422 …UpdateKernelFormatInfo(kernel_node, inputs_type, &inputs_format, &outputs_format, &origin_data_for… in SetKernelInfo()
[all …]
/third_party/mindspore/mindspore/core/ops/
Dmerge.cc32 auto inputs_type = input_args[0]->BuildType()->cast<TuplePtr>()->elements(); in MergeInfer() local
35 for (size_t i = 0; i != inputs_type.size(); i++) { in MergeInfer()
36 (void)args.insert(std::make_pair("input[" + std::to_string(i) + "]", inputs_type[i])); in MergeInfer()
Dassert.cc71 auto inputs_type = input_args[1]->BuildType()->cast<TuplePtr>()->elements(); in AssertInfer() local
72 for (auto dtype : inputs_type) { in AssertInfer()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_metadata.cc42 std::vector<TypeId> inputs_type{}; in HostMetadataInfo() local
46 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, input_index)); in HostMetadataInfo()
57 builder.SetInputsDeviceType(inputs_type); in HostMetadataInfo()
/third_party/mindspore/mindspore/ops/operations/
Ddebug_ops.py372 def infer_dtype(self, *inputs_type): argument
373 if len(inputs_type) == 1:
374 return inputs_type[0]
375 return inputs_type
Dnn_ops.py4492 def infer_dtype(self, inputs_type, rois_type): argument
4494 validator.check_tensor_dtype_valid("inputs_type", inputs_type, valid_dtypes, self.name)
4496 return inputs_type
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_metadata.cc55 std::vector<TypeId> inputs_type{}; in AicpuMetadataInfoForSpecialNodes() local
61 … (void)inputs_type.emplace_back(AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, input_index)); in AicpuMetadataInfoForSpecialNodes()
73 builder.SetInputsDeviceType(inputs_type); in AicpuMetadataInfoForSpecialNodes()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dreorder_ops.cc123 info_builder.SetInputsDeviceType(node_io_info.inputs_type); in SetNodeInfo()
183 new_inputs_info->inputs_type.resize(0); in SetTypeInsensitiveNodeInputsInfo()
189 new_inputs_info->inputs_type.push_back(AnfAlgo::GetInputDeviceDataType(node, data_idx)); in SetTypeInsensitiveNodeInputsInfo()
193 … new_inputs_info->inputs_type.push_back(AnfAlgo::GetInputDeviceDataType(input_at_indexes[idx], 0)); in SetTypeInsensitiveNodeInputsInfo()
196 …new_inputs_info->inputs_type.push_back(AnfAlgo::GetOutputDeviceDataType(input_at_indexes[idx], 0)); in SetTypeInsensitiveNodeInputsInfo()
240 cast_io_info.inputs_type.push_back(AnfAlgo::GetInputDeviceDataType(type_insens_node, index)); in ReorderTypeInsensitiveCastDown()
315 cast_io_info.inputs_type.push_back(small_type); in ReorderCastUpTypeInsensitive()
Dgraph_kernel_helper.h67 const std::vector<TypeId> &inputs_type,
71 const std::vector<TypeId> &inputs_type,
Dreorder_ops.h30 std::vector<TypeId> inputs_type; member
Dgraph_kernel_helper.cc288 const std::vector<TypeId> &inputs_type, in BuildSelectKernelBuildInfo() argument
293 graph_info_builder.SetInputsDeviceType(inputs_type); in BuildSelectKernelBuildInfo()
304 const std::vector<TypeId> &inputs_type, in BuildSelectKernelBuildInfo() argument
309 graph_info_builder.SetInputsDeviceType(inputs_type); in BuildSelectKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dadd_relu_v2_fusion.cc33 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
39 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(node, input_index)); in GenerateKernelBuildInfo()
47 builder.SetInputsDeviceType(inputs_type); in GenerateKernelBuildInfo()
Dmatmul_biasadd_fusion.cc33 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
39 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(node, input_index)); in GenerateKernelBuildInfo()
47 builder.SetInputsDeviceType(inputs_type); in GenerateKernelBuildInfo()
Dadd_relu_grad_v2_fusion.cc33 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
39 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(node, input_index)); in GenerateKernelBuildInfo()
47 builder.SetInputsDeviceType(inputs_type); in GenerateKernelBuildInfo()
Drelu_v2_pass.cc46 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
52 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(node, input_index)); in GenerateKernelBuildInfo()
60 builder.SetInputsDeviceType(inputs_type); in GenerateKernelBuildInfo()
Dadam_fusion.cc33 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
40 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(node, input_index)); in GenerateKernelBuildInfo()
48 builder.SetInputsDeviceType(inputs_type); in GenerateKernelBuildInfo()
Dprint_reduce_fusion.cc32 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
39 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(node, input_index)); in GenerateKernelBuildInfo()
49 builder.SetInputsDeviceType(inputs_type); in GenerateKernelBuildInfo()
Dadam_weight_decay_fusion.cc33 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
40 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(node, input_index)); in GenerateKernelBuildInfo()
48 builder.SetInputsDeviceType(inputs_type); in GenerateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhccl_kernel_metadata.cc88 std::vector<TypeId> inputs_type{}; in HcclMetadataInfo() local
92 inputs_type.push_back(type); in HcclMetadataInfo()
107 builder.SetInputsDeviceType(inputs_type); in HcclMetadataInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.cc148 std::vector<TypeId> inputs_type; in SetControlOpInfo() local
152 inputs_type.push_back(AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, input_index)); in SetControlOpInfo()
164 builder->SetInputsDeviceType(inputs_type); in SetControlOpInfo()