Home
last modified time | relevance | path

Searched defs:input_types (Results 1 – 25 of 38) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dkernel_build_utils.cc68 void GetInputDtypes(const CNodePtr &kernel_node, std::vector<TypeId> *input_types, in GetInputDtypes()
123 …typeFormatMatchedNum(const kernel::KernelAttr &kernel_attr, const std::vector<TypeId> &input_types, in GetInputDtypeFormatMatchedNum()
170 …nelBuildInfo(const std::vector<std::string> &input_formats, const std::vector<TypeId> &input_types, in SetKernelBuildInfo()
186 std::vector<TypeId> input_types; in UpdateDynamicKernelBuildInfo() local
209 std::vector<TypeId> input_types; in UpdateCustomKernelBuildInfo() local
228 …ernelAttr FillNoneInKernelAttr(const CNodePtr &kernel_node, const std::vector<TypeId> &input_types, in FillNoneInKernelAttr()
290 … const std::vector<kernel::KernelAttr> &kernel_attrs, const std::vector<TypeId> &input_types, in SelectKernel()
373 std::vector<TypeId> input_types; in SetKernelInfoWithMsg() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dkernel_select_cpu.cc108 void GetInputDtypes(const CNodePtr &kernel_node, std::vector<TypeId> *input_types) { in GetInputDtypes()
158 …utDtypeFormatMatched(const kernel::KernelAttr &kernel_attr, const std::vector<TypeId> &input_types, in InputDtypeFormatMatched()
290 …nelBuildInfo(const std::vector<std::string> &input_formats, const std::vector<TypeId> &input_types, in SetKernelBuildInfo()
313 std::vector<TypeId> input_types; in SetKernelBuildInfoWithSelectedAttr() local
369 std::vector<TypeId> input_types; in KernelNotSupportWarning() local
418 std::vector<TypeId> input_types; in UpdateDynamicKernelBuildInfo() local
518 std::vector<TypeId> input_types; in UpdateCustomKernelBuildInfo() local
561 …ernelAttr FillNoneInKernelAttr(const CNodePtr &kernel_node, const std::vector<TypeId> &input_types, in FillNoneInKernelAttr()
626 std::vector<TypeId> input_types; in SelectKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/
Dacl_kernel_build.cc55 auto input_types = build_info->GetAllInputDeviceTypes(); in AclOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Drpc_recv.cc51 std::vector<TypePtr> input_types; in InferType() local
Drpc_send.cc51 std::vector<TypePtr> input_types; in InferType() local
Dlower_bound.cc69 std::map<std::string, TypePtr> input_types; in LowerBoundInferType() local
Dupper_bound.cc69 std::map<std::string, TypePtr> input_types; in UpperBoundInferType() local
Dmap_cache_idx.cc80 std::map<std::string, TypePtr> input_types; in MapCacheIdxInferType() local
Dgenerate_eod_mask.cc88 std::map<std::string, TypePtr> input_types; in InferType() local
Dlu_unpack.cc136 std::map<std::string, TypePtr> input_types; in LuUnpackInferType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/
Dcallback_impl.cc53 …tTypeAndFormats(const device::KernelWithIndex &kernel_with_index, std::vector<TypeId> *input_types, in GetTypeAndFormats()
134 …CallbackImpl::CollectInputTypesAndFormats(const AnfNodePtr &node, std::vector<TypeId> *input_types, in CollectInputTypesAndFormats()
204 std::vector<TypeId> input_types; in SetBasicNodeKernelInfo() local
264 std::vector<TypeId> input_types; in ResetKernelInfoInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/composite/
Ddo_signature.cc72 void GetTypeInfo(const std::vector<TypePtr> &input_types, std::vector<TypeId> *args_type_id, in GetTypeInfo()
90 void DoAutoCast(const std::vector<Signature> &signature, const std::vector<TypePtr> &input_types, in DoAutoCast()
204 std::vector<TypePtr> input_types; in GetNewInputsBySignatures() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/rts/
Drt_kernel_info.cc78 std::vector<TypeId> input_types = {}; in GetRtKelInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dapply_rotary_pos_emb.cc57 std::map<std::string, TypePtr> input_types; in InferType() local
Dgenerator.cc62 auto input_types = input_args[kInputsIndex]->GetType()->cast<TuplePtr>(); in CheckValidation() local
Dgroup_norm.cc118 TypePtrList input_types{x_type, weight_type, bias_type}; in InferType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dcast_matmul_fusion.cc30 std::vector<TypeId> input_types = AnfAlgo::GetAllInputDeviceTypes(matmul_node); in UpdateBuildInfo() local
Dinsert_pad.cc149 std::vector<TypeId> input_types = AnfAlgo::GetAllInputDeviceTypes(matmul); in InsertPad() local
201 std::vector<TypeId> input_types = AnfAlgo::GetAllInputDeviceTypes(matmul_node); in UpdateMatmulInfo() local
/third_party/vixl/tools/test_generator/
Dparser.py54 def __init__(self, operand_types, operand_variants, input_types, argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dglu_grad.cc94 const std::set<TypePtr> input_types = {kFloat64, kFloat32, kFloat16}; in GluGradInferType() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/backend/
Dgraph_optimizer_test_framework.cc118 std::vector<TypeId> input_types(input_num); in SetGeneralBuildInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/
Dgpu_kernel_factory.cc199 std::vector<mindspore::TypeId> input_types; in GpuKernelAttrCheck() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/
Dmapper_config_parser.cc130 auto input_types = SplitString(input_type_str, ';'); in ParseInputType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dsparse_apply_adagrad_d_a_gpu_kernel.cc223 std::vector<TypeId> input_types(kSparseApplyAdagradDAInputsNum); in CheckDType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dsession_basic.h82 std::vector<InputType> input_types; member

12