Home
last modified time | relevance | path

Searched defs:shape_tuple (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dgeqrf.cc43 std::vector<abstract::BaseShapePtr> shape_tuple; in InferShape() local
70 std::vector<abstract::BaseShapePtr> shape_tuple; in InferShape() local
Dscatter_nd.cc112 const ArrayValue<int64_t> &shape_tuple = shape_tuple_opt.value(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsparse_split.cc64 std::vector<abstract::BaseShapePtr> shape_tuple; in GetOutputShapes() local
100 …auto shape_tuple = GetOutputShapes(output_indices_vec, output_values_vec, shape_shape_vec, num_spl… in SparseSplitInferShape() local
161 …auto shape_tuple = GetOutputShapes(output_indices_vec, output_values_vec, shape_shape, num_splits); in InferShape() local
Dsvd.cc57 std::vector<abstract::BaseShapePtr> shape_tuple; in SvdInferShape() local
90 std::vector<abstract::BaseShapePtr> shape_tuple; in SvdInferShape() local
Dctc_loss_v2.cc88 auto shape_tuple = idx_shape_ptr->cast<abstract::TupleShapePtr>(); in CTCLossV2InferShape() local
95 auto shape_tuple = idx_shape_ptr->cast<abstract::TupleShapePtr>(); in CTCLossV2InferShape() local
Ddensetodense_set_operation.cc89 std::vector<abstract::BaseShapePtr> shape_tuple; in DenseToDenseSetOperationInferShape() local
155 std::vector<abstract::BaseShapePtr> shape_tuple; in DenseToDenseSetOperationFrontendInferShape() local
Dlog_uniform_candidate_sampler.cc73 std::vector<abstract::BaseShapePtr> shape_tuple; in InferShape() local
Ddynamic_broadcast_gradient_args.cc58 auto shape_tuple = idx_shape_ptr->cast<abstract::TupleShapePtr>(); in CheckInputsAndGetShape() local
Dsplit_v.cc103 std::vector<abstract::BaseShapePtr> shape_tuple; in SplitVInferShape() local
Dconcat_offset_v1.cc62 auto shape_tuple = idx_shape_ptr->cast<abstract::TupleShapePtr>(); in ConcatOffsetV1InferShape() local
Dunstack.cc67 std::vector<abstract::BaseShapePtr> shape_tuple; in UnstackInferShape() local
Dbuffer_append.cc48 auto shape_tuple = idx_shape_ptr->cast<abstract::TupleShapePtr>(); in GetSequnceIndexShape() local
Dapply_adam_with_amsgradv2.cc151 auto shape_tuple = infer_shape->cast_ptr<abstract::TupleShape>(); in ApplyAdamWithAmsgradV2Infer() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/pipeline/static_analysis/
Ddata_test.cc130 std::shared_ptr<TupleShape> shape_tuple = dyn_cast<TupleShape>(abstract_tup->BuildShape()); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dshape_reshape_fusion.cc197 VarPtr shape_tuple = std::make_shared<SeqVar>(); in DefinePattern() local
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/
Dutils.cc282 auto shape_tuple = base_shape->cast_ptr<TupleShape>(); in MakeAbstract() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dconvert_utils_py.cc745 ShapeVector ConvertShapeTupleToShapeVector(const ValueTuplePtr &shape_tuple) { in ConvertShapeTupleToShapeVector()
782 ValueTuplePtr shape_tuple = utils::cast<ValueTuplePtr>(ref); in ConvertToShapeVector() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_guard/
Dinfer.cc160 mindspore::abstract::SequenceShapePtr shape_tuple; in CreateTuple() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/
Dprim.cc605 py::tuple shape_tuple(len); in AbstractTupleToPython() local
965 auto shape_tuple = shape_obj.cast<py::tuple>(); in GetPyAbsItemOfTupleOut() local