Home
last modified time | relevance | path

Searched defs:type_tuple (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dnon_zero_ext.cc55 std::vector<TypePtr> type_tuple; in InferType() local
99 TypePtrList type_tuple; in InferType() local
Dminimum_grad.cc38 std::vector<TypePtr> type_tuple; in InferType() local
Dgeqrf.cc81 std::vector<TypePtr> type_tuple = {type, type}; in InferType() local
Dsplit_with_size.cc75 std::vector<TypePtr> type_tuple; in InferType() local
Dsplit.cc91 std::vector<TypePtr> type_tuple; in InferType() local
Dsplit_tensor.cc95 std::vector<TypePtr> type_tuple; in InferType() local
Dgrouped_matmul.cc265 std::vector<TypePtr> type_tuple; in InferType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/
Dfallback.cc85 auto type_tuple = type->cast_ptr<Tuple>(); in GetTypeElements() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dellipsis_to_slice.cc55 std::vector<TypePtr> type_tuple{kInt64, kInt64, kInt64}; in InferType() local
Dsort.cc77 std::vector<TypePtr> type_tuple; in SortInferType() local
Dnms_with_mask.cc105 std::vector<TypePtr> type_tuple; in NMSWithMaskInferType() local
Dsvd.cc103 std::vector<TypePtr> type_tuple = {type, type, type}; in SvdInferType() local
Dcompute_accidental_hits.cc84 std::vector<TypePtr> type_tuple; in ComputeAccidentalHitsInferType() local
Dsplit_v.cc123 std::vector<TypePtr> type_tuple; in SplitVInferType() local
Dunstack.cc57 std::vector<TypePtr> type_tuple; in UnstackInferType() local
Dsparse_split.cc120 std::vector<TypePtr> type_tuple; in SparseSplitInferType() local
Dapply_adam_with_amsgradv2.cc152 auto type_tuple = infer_type->cast_ptr<Tuple>(); in ApplyAdamWithAmsgradV2Infer() local
Ddensetodense_set_operation.cc173 std::vector<TypePtr> type_tuple; in DenseToDenseSetOperationInferType() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dbroadcast_grad_grad.cc78 std::vector<TypePtr> type_tuple{x_type, x_type, x_type}; in BroadcastGradGradInferType() local
Dlstm_grad_data.cc187 std::vector<TypePtr> type_tuple(kLstmOutputNum, x_dtype); in LstmGradDataInferType() local
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/
Dutils.cc283 auto type_tuple = type->cast_ptr<Tuple>(); in MakeAbstract() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_guard/
Dinfer.cc162 auto type_tuple = type->cast_ptr<mindspore::Tuple>(); in CreateTuple() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/
D_checkparam.py996 def check_axis_type(axis, type_int=True, type_tuple=True, type_list=True): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/
Darray_func.py637 def _check_axis_type(axis, type_int=True, type_tuple=True, type_list=True, ops_name="ops"): argument