Home
last modified time | relevance | path

Searched refs:dtype_ (Results 1 – 25 of 111) sorted by relevance

12345

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Delu_grad_cpu_kernel.cc33 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
34 if (dtype_ != AnfAlgo::GetInputDeviceDataType(kernel_node, 1)) { in InitKernel()
43 if (dtype_ == kNumberTypeFloat32 || dtype_ == kNumberTypeFloat) { in Launch()
45 } else if (dtype_ == kNumberTypeFloat16) { in Launch()
48 MS_LOG(EXCEPTION) << "Data type is " << TypeIdLabel(dtype_) << "is not support."; in Launch()
Dunique_with_pad_cpu_kernel.cc32 if (dtype_ == kNumberTypeInt32) { in Launch()
35 } else if (dtype_ == kNumberTypeInt64) { in Launch()
38 } else if (dtype_ == kNumberTypeFloat32 || dtype_ == kNumberTypeFloat16) { in Launch()
42 MS_LOG(EXCEPTION) << "Unsupported input data type: " << dtype_; in Launch()
Dresize_nearest_neighbor_grad_cpu_kernel.cc36 dtype_ = AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, 0); in InitKernel()
63 if (dtype_ == kNumberTypeFloat16) { in Launch()
65 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
67 } else if (dtype_ == kNumberTypeFloat64) { in Launch()
69 } else if (dtype_ == kNumberTypeInt32) { in Launch()
71 } else if (dtype_ == kNumberTypeInt64) { in Launch()
74 MS_LOG(EXCEPTION) << "Unsupported input data type: " << dtype_; in Launch()
Dresize_nearest_neighbor_cpu_kernel.cc36 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
62 if (dtype_ == kNumberTypeFloat16) { in Launch()
64 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
66 } else if (dtype_ == kNumberTypeFloat64) { in Launch()
68 } else if (dtype_ == kNumberTypeInt32) { in Launch()
70 } else if (dtype_ == kNumberTypeInt64) { in Launch()
73 MS_LOG(EXCEPTION) << "Unsupported input data type: " << dtype_; in Launch()
Dcumsum_cpu_kernel.cc34 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
59 if (dtype_ == kNumberTypeFloat32) { in InitInputOutputSize()
61 } else if (dtype_ == kNumberTypeFloat16) { in InitInputOutputSize()
63 } else if (dtype_ == kNumberTypeInt32) { in InitInputOutputSize()
65 } else if (dtype_ == kNumberTypeInt8) { in InitInputOutputSize()
67 } else if (dtype_ == kNumberTypeUInt8) { in InitInputOutputSize()
71 << TypeIdToType(dtype_)->ToString(); in InitInputOutputSize()
81 if (dtype_ == kNumberTypeFloat32) { in Launch()
83 } else if (dtype_ == kNumberTypeFloat16) { in Launch()
85 } else if (dtype_ == kNumberTypeInt32) { in Launch()
[all …]
Dsigmoid_cross_entropy_with_logits_grad_cpu_kernel.cc30 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
42 if (dtype_ == kNumberTypeFloat16) { in Launch()
44 } else if (dtype_ == kNumberTypeFloat32 || dtype_ == kNumberTypeFloat64) { in Launch()
47 MS_LOG(EXCEPTION) << "Unsupported input data type: " << dtype_; in Launch()
Dsigmoid_cross_entropy_with_logits_cpu_kernel.cc30 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
40 if (dtype_ == kNumberTypeFloat16) { in Launch()
42 } else if (dtype_ == kNumberTypeFloat32 || dtype_ == kNumberTypeFloat64) { in Launch()
45 MS_LOG(EXCEPTION) << "Unsupported input data type: " << dtype_; in Launch()
Dunique_cpu_kernel.cc31 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
47 if (dtype_ == kNumberTypeInt32) { in Launch()
49 } else if (dtype_ == kNumberTypeInt64) { in Launch()
51 } else if (dtype_ == kNumberTypeFloat32 || dtype_ == kNumberTypeFloat16) { in Launch()
54 MS_LOG(EXCEPTION) << "Unsupported input data type: " << dtype_; in Launch()
Dpad_cpu_kernel.cc31 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
67 if (dtype_ == kNumberTypeFloat16) { in Launch()
69 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
71 } else if (dtype_ == kNumberTypeFloat64) { in Launch()
73 } else if (dtype_ == kNumberTypeInt32) { in Launch()
76 MS_LOG(EXCEPTION) << "Data type is " << TypeIdLabel(dtype_) << " which is not supported."; in Launch()
Dresize_bilinear_cpu_kernel.cc36 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
57 if (dtype_ == kNumberTypeFloat16) { in Launch()
59 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
62 MS_LOG(EXCEPTION) << "Unsupported input data type: " << dtype_; in Launch()
73 if (dtype_ == kNumberTypeFloat16) { in LaunchKernel()
92 } else if (dtype_ == kNumberTypeFloat32) { in LaunchKernel()
146 if (dtype_ == kNumberTypeFloat16) { in LaunchKernel()
Dunsorted_segment_sum_cpu_kernel.cc31 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
65 if (dtype_ == kNumberTypeInt32 && segment_ids_dtype_ == kNumberTypeInt32) { in Launch()
69 } else if (dtype_ == kNumberTypeFloat32 && segment_ids_dtype_ == kNumberTypeInt32) { in Launch()
73 } else if (dtype_ == kNumberTypeInt32 && segment_ids_dtype_ == kNumberTypeInt64) { in Launch()
77 } else if (dtype_ == kNumberTypeFloat32 && segment_ids_dtype_ == kNumberTypeInt64) { in Launch()
Dmaximum_grad_cpu_kernel.cc43 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
54 if (dtype_ == kNumberTypeInt32) { in Launch()
56 } else if (dtype_ == kNumberTypeUInt32) { in Launch()
58 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
60 } else if (dtype_ == kNumberTypeInt64) { in Launch()
62 } else if (dtype_ == kNumberTypeUInt64) { in Launch()
64 } else if (dtype_ == kNumberTypeFloat64) { in Launch()
66 } else if (dtype_ == kNumberTypeFloat16) { in Launch()
Drange_cpu_kernel.cc30 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
37 if (dtype_ == kNumberTypeInt32) { in Launch()
39 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
42 MS_LOG(EXCEPTION) << "Only support int, float, but actual data type is " << TypeIdLabel(dtype_); in Launch()
Dresize_bilinear_grad_cpu_kernel.cc36 dtype_ = AnfAlgo::GetPrevNodeOutputInferDataType(kernel_node, 0); in InitKernel()
59 if (dtype_ == kNumberTypeFloat16) { in Launch()
61 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
64 MS_LOG(EXCEPTION) << "Unsupported input data type: " << dtype_; in Launch()
79 if (dtype_ == kNumberTypeFloat16) { in LaunchKernel()
105 } else if (dtype_ == kNumberTypeFloat32) { in LaunchKernel()
162 if (dtype_ == kNumberTypeFloat16) { in LaunchKernel()
Ddropout_grad_kernel.cc44 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
53 if (dtype_ == kNumberTypeFloat16) { in InitInputOutputSize()
64 if (dtype_ == kNumberTypeFloat16) { in Launch()
66 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
70 << TypeIdToType(dtype_)->ToString(); in Launch()
Dminimum_grad_cpu_kernel.cc71 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
82 if (dtype_ == kNumberTypeInt32) { in Launch()
84 } else if (dtype_ == kNumberTypeUInt32) { in Launch()
86 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
88 } else if (dtype_ == kNumberTypeInt64) { in Launch()
90 } else if (dtype_ == kNumberTypeUInt64) { in Launch()
92 } else if (dtype_ == kNumberTypeFloat64) { in Launch()
Ddropout_cpu_kernel.cc41 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
51 if (dtype_ == kNumberTypeFloat16) { in Launch()
53 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
57 << TypeIdToType(dtype_)->ToString(); in Launch()
Dmirror_pad_grad_cpu_kernel.cc59 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
127 if (dtype_ == kNumberTypeFloat16) { in Launch()
129 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
131 } else if (dtype_ == kNumberTypeFloat64) { in Launch()
133 } else if (dtype_ == kNumberTypeInt32) { in Launch()
136 MS_LOG(EXCEPTION) << "Data type is " << TypeIdLabel(dtype_) << " which is not supported."; in Launch()
149 if (dtype_ == kNumberTypeFloat16) { in InitInputOutputSize()
151 } else if (dtype_ == kNumberTypeFloat32) { in InitInputOutputSize()
153 } else if (dtype_ == kNumberTypeFloat64) { in InitInputOutputSize()
155 } else if (dtype_ == kNumberTypeInt32) { in InitInputOutputSize()
Dapply_adagrad_cpu_kernel.cc35 dtype_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 0); in InitKernel()
41 if (dtype_ == kNumberTypeFloat16) { in Launch()
43 } else if (dtype_ == kNumberTypeFloat32) { in Launch()
47 << TypeIdToType(dtype_)->ToString(); in Launch()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/vision/
Dnormalize_pad_ir.cc33 : mean_(mean), std_(std), dtype_(dtype) {} in NormalizePadOperation()
41 if (dtype_ != "float32" && dtype_ != "float16") { in ValidateParams()
42 std::string err_msg = "NormalizePad: dtype must be float32 or float16, but got: " + dtype_; in ValidateParams()
54 … std_[dimension_zero], std_[dimension_one], std_[dimension_two], dtype_); in Build()
61 args["dtype"] = dtype_; in to_json()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_mul.cc24 std::vector<TypeId> input_dtypes = {dtype_, dtype_}; in ObtainMulKernelGraph()
25 std::vector<TypeId> output_dtypes = {dtype_}; in ObtainMulKernelGraph()
27 size_t dtype_size = abstract::TypeIdSize(dtype_); in ObtainMulKernelGraph()
68 auto size = abstract::TypeIdSize(dtype_); in ObtainMulInputsAddr()
/third_party/boost/boost/python/numpy/
Dinvoke_matching.hpp49 dtype_template_invoker(dtype const & dtype_, Function func) in dtype_template_invoker()
50 : m_dtype(dtype_), m_func(func) {} in dtype_template_invoker()
71 dtype_template_invoker(dtype const & dtype_, Function & func) in dtype_template_invoker()
72 : m_dtype(dtype_), m_func(func) {} in dtype_template_invoker()
132 void invoke_matching_dtype(dtype const & dtype_, Function f) in invoke_matching_dtype() argument
134 detail::dtype_template_invoker<Function> invoker(dtype_, f); in invoke_matching_dtype()
/third_party/mindspore/mindspore/ccsrc/debug/debugger/offline_debug/
Ddbg_services.h124 : data_size_(data_size), dtype_(dtype), shape_(shape) {} in TensorBaseData()
127 const int dtype() const { return dtype_; } in dtype()
130 int dtype_; member
139 dtype_(dtype), in TensorStatData()
154 const int dtype() const { return dtype_; } in dtype()
169 int dtype_; member
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/control/
Dtensorlist_fromtensor.cc77 dtype_ = in_tensors_[0]->data_type(); in Run()
97 auto data_offset = devision_dim0 * lite::DataTypeSize(dtype_); in Run()
112 out_ptr->set_data_type(dtype_); in Run()
116 output0->set_tensors_data_type(dtype_); in Run()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_transdata.cc80 std::vector<TypeId> input_dtypes = {dtype_}; in ObtainTransDataKernelGraph()
81 std::vector<TypeId> output_dtypes = {dtype_}; in ObtainTransDataKernelGraph()
101 AnfAlgo::SetOutputInferTypeAndShape({dtype_}, {shape_}, transdata_node.get()); in ConstructKernelGraphAndSetAttr()
105 std::vector<TypeId> device_type = {dtype_}; in ConstructKernelGraphAndSetAttr()

12345