Searched defs:inputs_type (Results 1 – 18 of 18) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | reorder_ops.h | 30 std::vector<TypeId> inputs_type; member
|
| D | graph_kernel_helper.cc | 288 const std::vector<TypeId> &inputs_type, in BuildSelectKernelBuildInfo()
304 const std::vector<TypeId> &inputs_type, in BuildSelectKernelBuildInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | host_kernel_metadata.cc | 42 std::vector<TypeId> inputs_type{}; in HostMetadataInfo() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | merge.cc | 32 auto inputs_type = input_args[0]->BuildType()->cast<TuplePtr>()->elements(); in MergeInfer() local
|
| D | assert.cc | 71 auto inputs_type = input_args[1]->BuildType()->cast<TuplePtr>()->elements(); in AssertInfer() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.cc | 202 bool IsNeedProcessFormatInfo(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type) { in IsNeedProcessFormatInfo()
247 void UpdateKernelFormatInfo(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type, in UpdateKernelFormatInfo()
344 … PrintUnsupportedTypeException(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type, in PrintUnsupportedTypeException()
407 std::vector<TypeId> inputs_type; in SetKernelInfo() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_metadata.cc | 55 std::vector<TypeId> inputs_type{}; in AicpuMetadataInfoForSpecialNodes() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel_metadata.cc | 88 std::vector<TypeId> inputs_type{}; in HcclMetadataInfo() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | add_relu_grad_v2_fusion.cc | 33 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| D | add_relu_v2_fusion.cc | 33 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| D | matmul_biasadd_fusion.cc | 33 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| D | relu_v2_pass.cc | 46 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| D | adam_fusion.cc | 33 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| D | adam_weight_decay_fusion.cc | 33 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| D | print_reduce_fusion.cc | 32 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | debug_ops.py | 372 def infer_dtype(self, *inputs_type):
|
| D | nn_ops.py | 4492 def infer_dtype(self, inputs_type, rois_type): argument
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/ |
| D | cpu_device_context.cc | 148 std::vector<TypeId> inputs_type; in SetControlOpInfo() local
|