Home
last modified time | relevance | path

Searched defs:output_types (Results 1 – 25 of 27) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dkernel_build_utils.cc53 void GetOutputDtypes(const CNodePtr &kernel_node, std::vector<TypeId> *output_types) { in GetOutputDtypes()
104 …utDtypeMatchedNum(const kernel::KernelAttr &kernel_attr, const std::vector<TypeId> &output_types) { in GetOutputDtypeMatchedNum()
171 … const std::vector<std::string> &output_formats, const std::vector<TypeId> &output_types, in SetKernelBuildInfo()
189 std::vector<TypeId> output_types; in UpdateDynamicKernelBuildInfo() local
217 std::vector<TypeId> output_types; in UpdateCustomKernelBuildInfo() local
229 const std::vector<TypeId> &output_types, in FillNoneInKernelAttr()
291 … const std::vector<size_t> &input_not_cnode_indexes, const std::vector<TypeId> &output_types, in SelectKernel()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dkernel_select_cpu.cc86 void GetOutputDtypes(const CNodePtr &kernel_node, std::vector<TypeId> *output_types) { in GetOutputDtypes()
95 void GetOutputDtypesForRealTuple(const CNodePtr &kernel_node, std::vector<TypeId> *output_types) { in GetOutputDtypesForRealTuple()
138 …utputDtypeMatched(const kernel::KernelAttr &kernel_attr, const std::vector<TypeId> &output_types) { in OutputDtypeMatched()
291 … const std::vector<std::string> &output_formats, const std::vector<TypeId> &output_types, in SetKernelBuildInfo()
307 std::vector<TypeId> output_types; in SetKernelBuildInfoWithSelectedAttr() local
420 std::vector<TypeId> output_types; in UpdateDynamicKernelBuildInfo() local
525 std::vector<TypeId> output_types; in UpdateCustomKernelBuildInfo() local
562 const std::vector<TypeId> &output_types, in FillNoneInKernelAttr()
627 std::vector<TypeId> output_types; in SelectKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/
Dacl_kernel_build.cc57 auto output_types = build_info->GetAllOutputDeviceTypes(); in AclOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dwkv.cc64 std::vector<TypePtr> output_types = {k_type, k_type, k_type, k_type}; in InferType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dgraph_kernel_helper.cc81 const std::vector<TypeId> &output_types) { in BuildSelectKernelBuildInfo()
90 const std::vector<TypeId> &output_types, in BuildSelectKernelBuildInfo()
317 std::vector<TypeId> output_types = {out_type->type_id()}; in CreateCNode() local
Dcast_matmul_fusion.cc35 std::vector<TypeId> output_types = AnfAlgo::GetAllOutputDeviceTypes(matmul_node); in UpdateBuildInfo() local
Dinsert_pad.cc153 std::vector<TypeId> output_types = {input_types.front()}; in InsertPad() local
203 std::vector<TypeId> output_types = AnfAlgo::GetAllOutputDeviceTypes(matmul_node); in UpdateMatmulInfo() local
Ddecrease_compute_precision.cc79 std::vector<TypeId> output_types = {TypeId::kNumberTypeFloat16}; in UpdateOutputInfo() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_nonzero_ext.cc73 TypePtrList output_types; in TEST_P() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dinsert_cast_gpu.cc113 auto output_types = std::vector<TypeId>(output_num, kNumberTypeFloat32); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dadd_cast_for_ge.cc124 std::vector<TypeId> output_types; in AddCastForGeOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Dakg_tensorrt.h40 size_t output_types[LEN_LIMIT]; member
/third_party/mindspore/mindspore-src/source/tests/
Ddataset_mock.py44 def output_types(self): member in MindData
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/
Dcallback_impl.cc214 std::vector<TypeId> output_types; in SetBasicNodeKernelInfo() local
397 std::vector<TypeId> output_types; in SetBasicNodeKernelInfo() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/backend/
Dgraph_optimizer_test_framework.cc127 std::vector<TypeId> output_types(output_num); in SetGeneralBuildInfo() local
/third_party/mindspore/mindspore-src/source/tests/st/pynative/loss_scale/
Dtest_loss_scale.py57 def output_types(self): member in MindData
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dsparse_apply_adagrad_d_a_gpu_kernel.cc228 std::vector<TypeId> output_types(kSparseApplyAdagradDAOutputsNum); in CheckDType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dcustom_op_infer.cc53 std::vector<int32_t> output_types; in InferOffline() local
/third_party/mesa3d/src/gallium/targets/teflon/
Dtest_executor.cpp402 void ***output, size_t **output_sizes, TfLiteType **output_types, in run_model()
Dtest_teflon.cpp56 TfLiteType *output_types; in test_model() local
/third_party/rust/rust/compiler/rustc_codegen_llvm/src/
Dasm.rs38 let mut output_types = vec![]; in codegen_inline_asm() localVariable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dactor_dump.cc1052 …AbstractActor *actor, std::vector<BaseShapePtr> *output_shapes, std::vector<TypePtr> *output_types, in FetchOutputInfo()
1214 std::vector<TypePtr> output_types; in DumpActorInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dkernel_select_ascend.cc495 std::vector<TypeId> output_types; in GenerateKernelBuildInfo() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/pre_activate/pass/
Dinsert_type_transform_op_test.cc259 const std::vector<std::string> &output_formats, const std::vector<TypeId> &output_types, in SetKernelBuildInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dframework_utils.cc444 … const std::vector<std::string> &output_formats, const std::vector<TypeId> &output_types, in SetKernelBuildInfo()

12