Searched refs:output_dtypes (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | single_kernel_graph.cc | 24 const std::vector<TypeId> &output_dtypes, const std::vector<std::vector<size_t>> &output_shapes) { in ConstructKernelGraphBasedOnSingleOp() argument
47 if (output_dtypes.size() != output_shapes.size()) { in ConstructKernelGraphBasedOnSingleOp()
50 AnfAlgo::SetOutputInferTypeAndShape(output_dtypes, output_shapes, cnode.get()); in ConstructKernelGraphBasedOnSingleOp()
|
| D | single_kernel_graph.h | 34 … const std::vector<TypeId> &output_dtypes, const std::vector<std::vector<size_t>> &output_shapes);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | launch_mul.cc | 25 std::vector<TypeId> output_dtypes = {dtype_}; in ObtainMulKernelGraph() local
35 kMulOpName, input_dtypes, input_shapes, output_dtypes, output_shapes); in ObtainMulKernelGraph()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_launch_atomic_clean.cc | 79 std::vector<TypeId> output_dtypes = {}; in ObtainAtomicCleanKernelGraph() local
89 kAtomicAddrCleanOpName, input_dtypes, input_shapes, output_dtypes, output_shapes); in ObtainAtomicCleanKernelGraph()
|
| D | ascend_launch_transdata.cc | 81 std::vector<TypeId> output_dtypes = {dtype_}; in ObtainTransDataKernelGraph() local
88 kTransDataOpName, input_dtypes, input_shapes, output_dtypes, output_shapes); in ObtainTransDataKernelGraph()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | cpu_kernel_factory.cc | 79 auto output_dtypes = outputs_ptr[j]->dtypes(); in SetKernelAttrs() local
81 (void)kernel_attr.AddOutputAttr(kernel::DtypeToTypeId(output_dtypes[i]), output_formats[i]); in SetKernelAttrs()
|