| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | tensor_array.cc | 35 …rray::Init(bool dynamic_size, bool identical_element_shapes, const std::vector<int> &element_shape, in Init()
51 void TensorArray::set_element_shape(const std::vector<int> &element_shape) { in set_element_shape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | _tensor_array.py | 56 def __init__(self, dtype, element_shape, dynamic_size=True, size=0, name="TA"): argument
147 def __init__(self, dtype, element_shape): argument
270 def __init__(self, dtype, element_shape, dynamic_size, size): argument
345 def __init__(self, dtype, element_shape): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/graph_syntax/ |
| D | test_expr_clear.py | 35 def __init__(self, dtype, element_shape): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_tensor_array.py | 24 def __init__(self, dtype, element_shape, is_dynamic_shape=True, size=0): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_tensor_array.py | 24 def __init__(self, dtype, element_shape, is_dynamic_shape=True, size=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/control/ |
| D | tensorlist_getitem_infer.c | 70 int element_shape[MAX_SHAPE_SIZE] = {0}; in TensorListGetItemInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/ |
| D | tensorlist_c_utils.c | 50 int TensorListMergeShape(int *element_shape, size_t *element_shape_size, const int *tmp, size_t tmp… in TensorListMergeShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/reinforcement/ |
| D | tensor_array.py | 61 def __init__(self, dtype, element_shape, dynamic_size=True, size=0, name="TA"): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/control/ |
| D | tensor_array.cc | 46 int *element_shape = this->ta_param_->element_shape_; in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/abstract/ |
| D | dshape.h | 255 …explicit DynamicSequenceShape(const BaseShapePtr &element_shape) : element_shape_(element_shape) {} in DynamicSequenceShape()
297 BaseShapePtr element_shape() { return element_shape_; } in element_shape() function
|
| D | abstract_value.cc | 1144 BaseShapePtr element_shape = element->GetShape(); in ElementsShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | unstack_ext.cc | 62 ShapeVector element_shape; in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/ |
| D | unstack_ext.cc | 34 ShapeVector element_shape; in InferOutputForRankKnown() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/ |
| D | fallback.cc | 112 auto element_shape = (*shape_seq)[it]; in GenerateAbstractSequence() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/ |
| D | tensorlist.h | 84 std::vector<int> element_shape() const { in element_shape() function
|
| D | tensorlist.cc | 29 TensorList::TensorList(std::vector<int> shape, std::vector<int> element_shape, Category category) in TensorList()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | add_tensor_array.cc | 177 std::vector<int> element_shape; in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | data_utils.cc | 749 TensorShape element_shape(std::vector<int64_t>(tensor_shape.begin() + 1, tensor_shape.end())); in BatchTensorToCVTensorVector() local
775 TensorShape element_shape(std::vector<int64_t>(tensor_shape.begin() + 1, tensor_shape.end())); in BatchTensorToTensorVector() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | kernel.cc | 233 const auto &element_shape = shapes[0]; in SetShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/ |
| D | convert_utils.cc | 655 const auto &element_shape = element_base_shape->cast<abstract::ShapePtr>(); in BaseShapeToShapeVector() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/legacy_optimizer/graph/ |
| D | infershape_pass.cc | 95 std::vector<int> element_shape; in ConvertTensorList() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | anf_runtime_algorithm.cc | 2280 const auto &element_shape = std::make_shared<abstract::Shape>(ShapeVector({})); in SequenceToTensor() local
|