Home
last modified time | relevance | path

Searched refs:dtypes (Results 1 – 25 of 48) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_convert_utils.cc81 TypeId DtypeToTypeId(const std::string &dtypes) { in DtypeToTypeId() argument
82 auto iter = type_str_id_maps.find(dtypes); in DtypeToTypeId()
84 MS_LOG(EXCEPTION) << "Illegal input device dtype: " << dtypes; in DtypeToTypeId()
97 size_t GetDtypeNbyte(const std::string &dtypes) { in GetDtypeNbyte() argument
98 auto iter = type_nbyte_maps.find(dtypes); in GetDtypeNbyte()
100 MS_LOG(EXCEPTION) << "Illegal input dtype: " << dtypes; in GetDtypeNbyte()
Dtbe_convert_utils.h29 TypeId DtypeToTypeId(const std::string &dtypes);
33 size_t GetDtypeNbyte(const std::string &dtypes);
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Ddo_signature.cc153 MaxTypeMap GetMaxDtype(const std::vector<SignatureEnumDType> &dtypes, const std::vector<TypePtr> &i… in GetMaxDtype() argument
157 for (size_t i = 0; i < dtypes.size(); ++i) { in GetMaxDtype()
158 auto it = type_indices.find(dtypes[i]); in GetMaxDtype()
160 (void)type_indices.insert(std::make_pair(dtypes[i], std::vector<size_t>{i})); in GetMaxDtype()
202 std::vector<SignatureEnumDType> dtypes; in DoAutoCast() local
203 (void)std::transform(signature.begin(), signature.end(), std::back_inserter(dtypes), in DoAutoCast()
205 …int64_t empty_dtype_count = std::count(dtypes.begin(), dtypes.end(), SignatureEnumDType::kDTypeEmp… in DoAutoCast()
206 if (dtypes.empty() || static_cast<int64_t>(dtypes.size()) == empty_dtype_count) { in DoAutoCast()
210 std::map<SignatureEnumDType, TypeId> dst_type = GetMaxDtype(dtypes, input_types); in DoAutoCast()
213 auto it = dst_type.find(dtypes[i]); in DoAutoCast()
/third_party/mindspore/tests/st/ops/gpu/
Dtest_prelu_op.py67 dtypes = [mstype.float16, mstype.float32] variable
80 for dtype in dtypes:
121 for dtype in dtypes:
140 for dtype in dtypes:
159 for dtype in dtypes:
178 for dtype in dtypes:
201 for dtype in dtypes:
Dtest_hswish_op.py79 dtypes = (np.float32, np.float16)
81 for dtype in dtypes:
Dtest_rl_buffer_net.py28 def create_tensor(capcity, shapes, dtypes): argument
31 buffer.append(Tensor(np.zeros(((capcity,)+shapes[i])), dtypes[i]))
/third_party/mindspore/tests/st/ops/cpu/
Dtest_addn_op.py43 dtypes = (np.int32, np.float32)
44 for dtype in dtypes:
69 dtypes = (np.int32, np.float32)
70 for dtype in dtypes:
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.cc261 TypeId DtypeToTypeId(const std::string &dtypes) { in DtypeToTypeId() argument
262 auto iter = type_id_maps.find(dtypes); in DtypeToTypeId()
266 MS_EXCEPTION(ArgumentError) << "Illegal input device dtype:" << dtypes; in DtypeToTypeId()
310 size_t kernel_info_cnt = inputs[0]->dtypes().size(); in SetInputKernelBuilderInfo()
315 std::vector<std::string> dtypes = input->dtypes(); in SetInputKernelBuilderInfo() local
317 if (dtypes.size() != kernel_info_cnt || formats.size() != kernel_info_cnt) { in SetInputKernelBuilderInfo()
319 << dtypes.size() << ", formats size : " << formats.size(); in SetInputKernelBuilderInfo()
331 auto type_id = DtypeToTypeId(dtypes[builder_idex]); in SetInputKernelBuilderInfo()
338 auto type_id = DtypeToTypeId(dtypes[builder_idex]); in SetInputKernelBuilderInfo()
345 auto type_id = DtypeToTypeId(dtypes[builder_idex]); in SetInputKernelBuilderInfo()
[all …]
Dcommon_utils.h78 TypeId DtypeToTypeId(const std::string &dtypes);
79 std::string Dtype2ShortType(const std::string &dtypes);
81 size_t GetDtypeNbyte(const std::string &dtypes);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dmasked_select_cpu_kernel.cc73 std::vector<TypeId> dtypes(output_num); in Launch() local
75 dtypes[i] = AnfAlgo::GetOutputDeviceDataType(node_, i); in Launch()
77 AnfAlgo::SetOutputInferTypeAndShape(dtypes, {out_shape}, node_.get()); in Launch()
Dsub_and_filter_cpu_kernel.cc81 std::vector<TypeId> dtypes(output_num); in LaunchKernel() local
83 dtypes[i] = AnfAlgo::GetOutputDeviceDataType(node, i); in LaunchKernel()
85 AnfAlgo::SetOutputInferTypeAndShape(dtypes, {out_shape, out_shape}, node.get()); in LaunchKernel()
Dpad_and_shift_cpu_kernel.cc89 std::vector<TypeId> dtypes(output_nums); in LaunchKernel() local
91 dtypes[i] = AnfAlgo::GetOutputDeviceDataType(node_, i); in LaunchKernel()
93 AnfAlgo::SetOutputInferTypeAndShape(dtypes, {out_shape}, node_.get()); in LaunchKernel()
Dunique_cpu_kernel.cc64 std::vector<TypeId> dtypes(output_num); in Launch() local
66 dtypes[i] = AnfAlgo::GetOutputDeviceDataType(node_, i); in Launch()
68 …AnfAlgo::SetOutputInferTypeAndShape(dtypes, {out_shape, AnfAlgo::GetOutputInferShape(node_, 1)}, n… in Launch()
Dcpu_kernel_factory.cc67 auto first_input_dtypes = inputs_ptr[0]->dtypes(); in SetKernelAttrs()
74 auto input_dtypes = inputs_ptr[j]->dtypes(); in SetKernelAttrs()
79 auto output_dtypes = outputs_ptr[j]->dtypes(); in SetKernelAttrs()
Dmap_cache_idx_cpu_kernel.cc55 std::vector<TypeId> dtypes(output_num); in UpdateShape() local
57 dtypes[i] = AnfAlgo::GetOutputDeviceDataType(node, i); in UpdateShape()
59 …AnfAlgo::SetOutputInferTypeAndShape(dtypes, {AnfAlgo::GetOutputInferShape(node, 0), out_shape, out… in UpdateShape()
/third_party/boost/libs/python/doc/numpy/tutorial/
Ddtype.rst1 How to use dtypes
31 We can also print the dtypes of the data members of the ndarray by using the get_dtype method for t…
35 We can also create custom dtypes and build ndarrays with the custom dtypes
/third_party/boost/libs/python/doc/html/numpy/_sources/tutorial/
Ddtype.rst.txt1 How to use dtypes
31 We can also print the dtypes of the data members of the ndarray by using the get_dtype method for t…
35 We can also create custom dtypes and build ndarrays with the custom dtypes
/third_party/vk-gl-cts/external/vulkancts/modules/vulkan/spirv_assembly/
DvktSpvAsmComputeShaderCase.cpp192 …ayout (const DeviceInterface& vkdi, const VkDevice& device, const vector<VkDescriptorType>& dtypes) in createDescriptorSetLayout() argument
196 for (size_t bindingNdx = 0; bindingNdx < dtypes.size(); ++bindingNdx) in createDescriptorSetLayout()
197 builder.addSingleBinding(dtypes[bindingNdx], VK_SHADER_STAGE_COMPUTE_BIT); in createDescriptorSetLayout()
242 …rPool (const DeviceInterface& vkdi, const VkDevice& device, const vector<VkDescriptorType>& dtypes) in createDescriptorPool() argument
246 for (size_t typeNdx = 0; typeNdx < dtypes.size(); ++typeNdx) in createDescriptorPool()
247 builder.addType(dtypes[typeNdx], 1); in createDescriptorPool()
258 …Pool pool, VkDescriptorSetLayout layout, const vector<VkDescriptorType>& dtypes, const vector<VkDe… in createDescriptorSet() argument
260 DE_ASSERT(dtypes.size() == descriptorInfos.size() + descriptorImageInfos.size()); in createDescriptorSet()
277 for (deUint32 descriptorNdx = 0; descriptorNdx < dtypes.size(); ++descriptorNdx) in createDescriptorSet()
279 switch (dtypes[descriptorNdx]) in createDescriptorSet()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc104 inputs_info.empty() ? outputs_info[0]->dtypes().size() : inputs_info[0]->dtypes().size(); in GetCommonPatternKernelInfo()
331 const auto &kernel_build_info_dtype = io_info_item->dtypes()[kernel_build_info_index]; in GenBuilderItem()
402 auto dtype = op_io_info.dtypes(); in CreateNewOpIOInfo()
505 std::vector<std::string> dtypes; member
534 select_input.dtypes = SplitStrToVec(input_dtype_item); in CreateNewOpInfo()
542 select_output.dtypes = SplitStrToVec(input_dtype_item); in CreateNewOpInfo()
558 …CreateNewOpIOInfo(*op_info.inputs_ptr().at(i), inputs.at(i).dtypes, inputs.at(i).formats, input_ne… in CreateNewOpInfo()
563 …CreateNewOpIOInfo(*op_info.outputs_ptr().at(i), outputs.at(i).dtypes, outputs.at(i).formats, outpu… in CreateNewOpInfo()
/third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/
Dtest_tensor_setitem.py39 def test_tensor_setitem_net_many_dtype(input_shape, update_shape, slice_tuple, dtypes): argument
40 for dtype in dtypes:
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/pyfunc/
Dpy_func_cpu_kernel.cc176 result[i] = RawMemoryToScalar(inputs[i]->addr, input_infos.dtypes[i]); in RawMemoryToPyObjects()
199 …return ObjectToRawMemory(object, output_infos.object_types[0], output_infos.dtypes[0], outputs[0]); in PyObjectToRawMemorys()
209 …ObjectToRawMemory(result_tuple[i], output_infos.object_types[i], output_infos.dtypes[i], outputs[i… in PyObjectToRawMemorys()
235 input_infos_.dtypes = in_types; in BuildFuncInfo()
245 output_infos_.dtypes = out_types; in BuildFuncInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dall_to_all_unify_mindir.cc82 std::vector<TypeId> dtypes(split_count, dtype); in CreateSplitNode() local
84 AnfAlgo::SetOutputInferTypeAndShape(dtypes, shapes, split_v.get()); in CreateSplitNode()
110 std::vector<TypeId> dtypes(split_count, single_type); in CreateAllToAllvNode() local
112 AnfAlgo::SetOutputInferTypeAndShape(dtypes, shapes, all_to_all_v.get()); in CreateAllToAllvNode()
/third_party/boost/libs/python/doc/numpy/reference/
Ddtype.rst8 .. _dtype: http://docs.scipy.org/doc/numpy/reference/arrays.dtypes.html#data-type-objects-dtype
10 …t to a numpy.dtype from builtin C++ objects, as well as to create custom dtypes from user defined …
/third_party/boost/libs/python/doc/html/numpy/_sources/reference/
Ddtype.rst.txt8 .. _dtype: http://docs.scipy.org/doc/numpy/reference/arrays.dtypes.html#data-type-objects-dtype
10 …t to a numpy.dtype from builtin C++ objects, as well as to create custom dtypes from user defined …
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dsplit_inputs_for_reduce_scatter.cc34 std::vector<TypeId> dtypes(rank_size, AnfAlgo::GetPrevNodeOutputInferDataType(node, i)); in InsertSplitForInput() local
43 AnfAlgo::SetOutputInferTypeAndShape(dtypes, shapes, split.get()); in InsertSplitForInput()

12