| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | kernel_select_cpu.cc | 52 std::vector<TypeId> output_types; in UpdatePrevNotCNodeFormatDtype() local
63 void GetOutputDtypes(const CNodePtr &kernel_node, std::vector<TypeId> *output_types) { in GetOutputDtypes()
102 … std::vector<std::string> *output_formats, std::vector<TypeId> *output_types) { in GetOutputFormatsAndDtypes()
125 int GetOutputDtypeMatchedNum(const KernelAttr &kernel_attr, const std::vector<TypeId> &output_types… in GetOutputDtypeMatchedNum()
179 … const std::vector<std::string> &output_formats, const std::vector<TypeId> &output_types, in SetKernelBuildInfo()
230 std::vector<TypeId> output_types; in UpdateDynamicKernelBuildInfoAndAttrs() local
275 … const std::vector<size_t> &input_not_cnode_indexes, const std::vector<TypeId> &output_types, in SelectKernel()
314 std::vector<TypeId> output_types; in SetKernelInfo() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | broadcast.cc | 56 std::vector<TypePtr> output_types; in BroadcastInfer() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/other/ |
| D | gpu_convert_to_dynamic_shape_gpu_kernel.h | 59 std::vector<TypeId> output_types = {AnfAlgo::GetOutputInferDataType(kernel_node_.lock(), 0)}; in PostExecute() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | cast_matmul_fusion.cc | 34 std::vector<TypeId> output_types = AnfAlgo::GetAllOutputDeviceTypes(matmul_node); in UpdateBuildInfo() local
|
| D | insert_pad.cc | 177 std::vector<TypeId> output_types = {input_types.front()}; in InsertPad() local
227 std::vector<TypeId> output_types = AnfAlgo::GetAllOutputDeviceTypes(matmul_node); in UpdateMatmulInfo() local
|
| D | optimize_assign.cc | 180 std::vector<TypeId> output_types = {input_types.back()}; in ReplaceAssignByInplaceAssignInGraphkernel() local
|
| D | decrease_compute_precision.cc | 74 std::vector<TypeId> output_types = {TypeId::kNumberTypeFloat16}; in UpdateOutputInfo() local
|
| D | graph_kernel_helper.cc | 290 … const std::vector<TypeId> &output_types, const AnfNodePtr &node) { in BuildSelectKernelBuildInfo()
306 const std::vector<TypeId> &output_types) { in BuildSelectKernelBuildInfo()
735 std::vector<TypeId> output_types = {out_type->type_id()}; in CreateCNode() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | insert_cast_gpu.cc | 110 auto output_types = std::vector<TypeId>(output_num, kNumberTypeFloat32); in Run() local
|
| /third_party/mindspore/tests/ |
| D | dataset_mock.py | 43 def output_types(self): member in MindData
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | lstm_cpu_kernel.cc | 45 auto output_types = std::vector<TypeId>(output_num, output_type); in InitInputOutputSize() local
|
| /third_party/mindspore/tests/st/pynative/loss_scale/ |
| D | test_loss_scale.py | 57 def output_types(self): member in MindData
|
| /third_party/gstreamer/gstplugins_bad/sys/mediafoundation/ |
| D | gstmftransform.cpp | 1146 GList ** output_types) in gst_mf_transform_get_output_available_types()
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 1568 def output_types(self): member in Dataset
3195 def output_types(self): member in TransferDataset
|