Home
last modified time | relevance | path

Searched defs:sequence_shape (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dsearchsorted.cc73 auto sequence_shape = sequence_shape_ptr->GetShapeVector(); in InferShape() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_searchsorted.cc33 std::vector<int64_t> sequence_shape; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dsearch_sorted_gpu_kernal.cc59 auto sequence_shape = inputs.at(kSearchSortedIndex0)->GetShapeVector(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsequence_slice_setitem.cc97 … auto sequence_shape = input_args[sequence_index]->GetShape()->cast<abstract::SequenceShapePtr>(); in InferShape() local
Dshape_calc.cc112 auto sequence_shape = base_shape->cast<abstract::SequenceShapePtr>(); in TryGetShapeArg() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/
Ddynamic_shape_helper.cc521 auto sequence_shape = base_shape->cast<abstract::SequenceShapePtr>(); in UpdateKernelTensorShape() local
536 auto sequence_shape = base_shape->cast<abstract::SequenceShapePtr>(); in UpdateKernelTensorShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Doutput_actor.cc78 const auto &sequence_shape = output_kernel_tensor->GetShape()->cast<abstract::SequenceShapePtr>(); in UpdateDynamicSequenceType() local
Ddata_prepare_actor.cc55 const auto &sequence_shape = tensor->base_shape_ptr()->cast<abstract::SequenceShapePtr>(); in IsEmptySequenceTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dkernel.cc165 const auto &sequence_shape = base_shape->cast<abstract::SequenceShapePtr>(); in GetShapeVectorByBaseShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/
Dbackend_base.cc1383 const auto &sequence_shape = tensor->base_shape_ptr()->cast<abstract::SequenceShapePtr>(); in IsEmptySequence() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_parallel_utils.cc641 auto sequence_shape = dyn_cast<abstract::SequenceShape>(shape); in ExtractNewShapeFromShape() local