| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_convert_utils.cc | 81 TypeId DtypeToTypeId(const std::string &dtypes) { in DtypeToTypeId()
97 size_t GetDtypeNbyte(const std::string &dtypes) { in GetDtypeNbyte()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/pyfunc/ |
| D | py_func_cpu_kernel.h | 38 std::vector<TypePtr> dtypes; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | all_to_all_unify_mindir.cc | 82 std::vector<TypeId> dtypes(split_count, dtype); in CreateSplitNode() local
110 std::vector<TypeId> dtypes(split_count, single_type); in CreateAllToAllvNode() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | unique_cpu_kernel.cc | 64 std::vector<TypeId> dtypes(output_num); in Launch() local
|
| D | masked_select_cpu_kernel.cc | 73 std::vector<TypeId> dtypes(output_num); in Launch() local
|
| D | sub_and_filter_cpu_kernel.cc | 81 std::vector<TypeId> dtypes(output_num); in LaunchKernel() local
|
| D | pad_and_shift_cpu_kernel.cc | 89 std::vector<TypeId> dtypes(output_nums); in LaunchKernel() local
|
| D | map_cache_idx_cpu_kernel.cc | 55 std::vector<TypeId> dtypes(output_num); in UpdateShape() local
|
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_tensor_setitem.py | 39 def test_tensor_setitem_net_many_dtype(input_shape, update_shape, slice_tuple, dtypes): argument
|
| D | test_tensor_copy_slices.py | 62 def test_tensor_copy_slices_net_many_dtype(input_shape, update_shape, begin, end, strides, dtypes): argument
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_rl_buffer_net.py | 28 def create_tensor(capcity, shapes, dtypes): argument
|
| D | test_prelu_op.py | 67 dtypes = [mstype.float16, mstype.float32] variable
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_cache.h | 83 std::vector<SignatureEnumDType> dtypes; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | split_inputs_for_reduce_scatter.cc | 34 std::vector<TypeId> dtypes(rank_size, AnfAlgo::GetPrevNodeOutputInferDataType(node, i)); in InsertSplitForInput() local
|
| D | concat_outputs_for_all_gather.cc | 104 const std::vector<TypeId> &dtypes = {std::get<0>(output_info)[i]}; in InsertConcatForOutput() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | do_signature.cc | 153 MaxTypeMap GetMaxDtype(const std::vector<SignatureEnumDType> &dtypes, const std::vector<TypePtr> &i… in GetMaxDtype()
202 std::vector<SignatureEnumDType> dtypes; in DoAutoCast() local
|
| /third_party/vk-gl-cts/external/vulkancts/modules/vulkan/spirv_assembly/ |
| D | vktSpvAsmComputeShaderCase.cpp | 192 …ayout (const DeviceInterface& vkdi, const VkDevice& device, const vector<VkDescriptorType>& dtypes) in createDescriptorSetLayout()
242 …rPool (const DeviceInterface& vkdi, const VkDevice& device, const vector<VkDescriptorType>& dtypes) in createDescriptorPool()
258 …ce, VkDescriptorPool pool, VkDescriptorSetLayout layout, const vector<VkDescriptorType>& dtypes, c… in createDescriptorSet()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 261 TypeId DtypeToTypeId(const std::string &dtypes) { in DtypeToTypeId()
315 std::vector<std::string> dtypes = input->dtypes(); in SetInputKernelBuilderInfo() local
391 std::vector<std::string> dtypes = output->dtypes(); in SetOutputKernelBuilderInfo() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | concat_outputs_for_all_gather.cc | 102 const std::vector<TypeId> &dtypes = {std::get<0>(output_info)[i]}; in InsertConcatForOutput() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | dynamic_gru_v2_grad_fission.cc | 224 …auto dtypes = {AnfAlgo::GetOutputInferDataType(input_h, 0), AnfAlgo::GetOutputInferDataType(input_… in AddHSplitNode() local
321 …auto dtypes = {AnfAlgo::GetOutputInferDataType(dgate_h, 0), AnfAlgo::GetOutputInferDataType(dgate_… in CreateDgateHSplitVDNode() local
|
| D | dynamic_rnn_grad_fission_v2.cc | 301 …auto dtypes = {AnfAlgo::GetOutputInferDataType(origin_input6, 0), AnfAlgo::GetOutputInferDataType(… in CreateSplitV() local
|
| /third_party/mindspore/mindspore/ops/composite/multitype_ops/ |
| D | _constexpr_utils.py | 267 def judge_indexes_types(dtypes, target_type): argument
297 def check_types_valid(dtypes, target_type, op_name): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/ |
| D | opinfo.h | 68 const std::vector<std::string> &dtypes() const { return dtypes_; } in dtypes() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | communication_op_fusion.cc | 346 std::vector<TypeId> dtypes(output_num, AnfAlgo::GetOutputInferDataType(final_node, 0)); in CreateFusedCommunicationOp() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_select.cc | 505 std::vector<std::string> dtypes; member
|