| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/customize/ |
| D | op_common.cc | 31 auto input_abs = input_tensor->ToAbstract(); in CopyCustomizeCall() local
94 auto input_abs = input_tensor->ToAbstract(); in ContiguousTensorOpProcess() local
117 std::vector<AbstractBasePtr> input_abs = {x_tensor->ToAbstract()}; in ClampTensorCustomizeCall() local
156 std::vector<AbstractBasePtr> input_abs = {x_tensor->ToAbstract()}; in ClampScalarCustomizeCall() local
|
| D | divmod.cc | 40 std::vector<AbstractBasePtr> input_abs{x_tensor->ToAbstract(), y_tensor->ToAbstract()}; in FloorDivCall() local
65 std::vector<AbstractBasePtr> input_abs{input_tensor->ToAbstract()}; in TruncCall() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | seqence_to_sequence_op_eliminate.h | 49 auto input_abs = real_node->abstract()->cast<abstract::AbstractListPtr>(); in operator() local
79 auto input_abs = real_node->abstract()->cast<abstract::AbstractTuplePtr>(); in operator() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_seq_to_seq.cc | 40 void compare_tuple_to_list(AbstractBasePtrList &input_abs, bool is_input_dyn_len) { in compare_tuple_to_list()
64 void compare_list_to_tuple(AbstractBasePtrList &input_abs, bool is_input_dyn_len) { in compare_list_to_tuple()
90 AbstractBasePtrList input_abs; in TEST_P() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/ |
| D | pyboost_utils.h | 87 const std::vector<AbstractBasePtr> &input_abs, in GetAddressInfoHelper()
97 … const std::vector<AbstractBasePtr> &input_abs, const T &... args) { in GetAddressInfo()
129 const abstract::AbstractBasePtr &input_abs, size_t index, in GetKernelTensor()
159 const abstract::AbstractBasePtr &input_abs, size_t index, in GetKernelTensor()
217 std::vector<AbstractBasePtr> input_abs; in SelectKernel() local
|
| D | op_runner.h | 58 const std::vector<AbstractBasePtr> &input_abs() const { return input_abs_; } in input_abs() function
59 void set_input_abs(const std::vector<AbstractBasePtr> &input_abs) { input_abs_ = input_abs; } in set_input_abs()
|
| D | pyboost_utils.cc | 246 … PyBoostUtils::InferByOpDef(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_ab… in InferByOpDef()
293 const abstract::AbstractBasePtr &input_abs, size_t index, in GetKernelTensor()
309 const abstract::AbstractBasePtr &input_abs, size_t index, in GetKernelTensor()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | copy.cc | 26 auto input_abs = ToAbstractNoValue(input_tensor); in CopyAscendCustomize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/ |
| D | max_min.cc | 32 …std::vector<AbstractBasePtr> input_abs{input_tensor->ToAbstract(), axis->ToAbstract(), keep_dims->… in MinOrMaxCPUCall() local
|
| D | prod_ext.cc | 29 const BoolImmPtr &keep_dims, const std::vector<AbstractBasePtr> &input_abs) { in ProdExtCPUCall()
|
| D | mean_ext.cc | 29 const BoolImmPtr &keep_dims, const std::vector<AbstractBasePtr> &input_abs) { in MeanExtCPUCall()
|
| D | sum_ext.cc | 30 const std::vector<AbstractBasePtr> &input_abs) { in SumExtCPUCall()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/pyboost/customize/ |
| D | max_min.cc | 33 …std::vector<AbstractBasePtr> input_abs{input_tensor->ToAbstract(), axis->ToAbstract(), keep_dims->… in MinOrMaxGPUCall() local
|
| D | prod_ext.cc | 42 …std::vector<AbstractBasePtr> input_abs{input_tensor->ToAbstract(), axis->ToAbstract(), keep_dims->… in ProdExtGPUCall() local
|
| D | mean_ext.cc | 42 …std::vector<AbstractBasePtr> input_abs{input_tensor->ToAbstract(), axis->ToAbstract(), keep_dims->… in MeanExtGPUCall() local
|
| D | sum_ext.cc | 43 …std::vector<AbstractBasePtr> input_abs{input_tensor->ToAbstract(), axis->ToAbstract(), keep_dims->… in SumExtGPUCall() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | sequence_count.cc | 62 auto input_abs = input_args[seq_index]; in InferValue() local
|
| D | sequence_index.cc | 65 auto input_abs = input_args[seq_index]; in InferValue() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/ |
| D | split_tensor.cc | 31 auto input_abs = input_args[kIndex0]; in InferAbstract() local
|
| D | split_with_size.cc | 31 auto input_abs = input_args[kIndex0]; in InferAbstract() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | split_with_size.cc | 25 std::vector<int64_t> CaculateSplitSize(const AbstractBasePtr &input_abs) { in CaculateSplitSize()
|
| D | split_tensor.cc | 24 int64_t CaculateSplitSections(const AbstractBasePtr &input_abs) { in CaculateSplitSections()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/ |
| D | bprop_meta_func_graph.cc | 36 FuncGraphPtr BpropMetaFuncGraph::GenerateFuncGraph(const abstract::AbstractBasePtrList &input_abs) { in GenerateFuncGraph()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/model/acl/ |
| D | acl_model_multi.cc | 174 auto input_abs = input_param->abstract(); in SetInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/symbol_engine/ |
| D | symbol_engine_impl.cc | 422 …SymbolEngineImpl::GeneralizeParamShape(const AnfNodePtr ¶m, const AbstractBasePtr &input_abs) { in GeneralizeParamShape()
501 …SymbolEngineImpl::GeneralizeParamValue(const AnfNodePtr ¶m, const AbstractBasePtr &input_abs) { in GeneralizeParamValue()
526 auto input_abs = inp->abstract(); in SetParamSymbols() local
|