Home
last modified time | relevance | path

Searched defs:tuple_shape (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/
Dsequence_stack.h40 std::vector<int64_t> tuple_shape; variable
Dsequence_concat.h40 std::vector<int64_t> tuple_shape; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/sequence/
Dsequence_getitem_cpu_kernel.cc50 auto tuple_shape = inputs[0]->GetShapeVector(); in Resize() local
Dsequence_to_tensor_cpu_kernel.cc63 auto tuple_shape = inputs[0]->GetShapeVector(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/jit/
Djit_dfunctor.cc44 auto tuple_shape = shape_elem->cast<abstract::TupleShapePtr>(); in NewValue() local
94 auto tuple_shape = cnode_shape->cast<abstract::TupleShapePtr>(); in GenNewTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/expander/core/
Dnode.cc55 auto tuple_shape = shape_->cast<abstract::SequenceShapePtr>(); in shapes() local
Demitter.cc159 auto tuple_shape = TensorToTuple(shape); in Reshape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dconcat.cc161 auto tuple_shape = x_base_shape->cast<abstract::TupleShapePtr>(); in InferShape() local
Dstack_ext.cc84 auto tuple_shape = input_shape->cast<abstract::TupleShapePtr>(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/
Dcontrol_actor.cc585 auto tuple_shape = std::make_shared<abstract::TupleShape>(shape_list); in MergeDeviceAddress() local
677 auto tuple_shape = std::make_shared<abstract::TupleShape>(); in MergeEmptyAddressDeviceAddress() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dcallback_impl.cc51 auto tuple_shape = base_shape->cast<abstract::TupleShapePtr>(); in GetOutputInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dnormalize_tuple_index.cc265 auto tuple_shape = input_shape->cast<abstract::TupleShapePtr>()->shape(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dconvert_utils.cc647 const auto &tuple_shape = base_shape->cast<abstract::SequenceShapePtr>(); in BaseShapeToShapeVector() local
Danfalgo.cc746 const auto &tuple_shape = shape->cast<abstract::TupleShapePtr>(); in GetOutputShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_parallel_utils.cc595 auto tuple_shape = tuple_shape_ptr->shape(); in GetNodeShape() local
640 std::vector<ShapeBasePtr> tuple_shape; in ExtractNewShapeFromShape() local
692 auto tuple_shape = tuple_shape_ptr->shape(); in GetNodeNewShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dop_node.cc115 auto tuple_shape = baseshape->cast<abstract::TupleShapePtr>(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc1442 auto tuple_shape = base_shape->cast<abstract::TupleShapePtr>(); in GetAnfNodeOutputShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Danf_runtime_algorithm.cc1975 auto tuple_shape = base_shape->cast<abstract::TupleShapePtr>(); in GetOutputDetailShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/
Dabstract_value.cc1402 auto tuple_shape = dyn_cast_ptr<TupleShape>(shape); in set_shape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc1491 auto tuple_shape = dyn_cast<abstract::TupleShape>(shape); in SetValueInfoType() local