Searched defs:element_shape (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | tensor_array.cc | 28 …rray::Init(bool dynamic_size, bool identical_element_shapes, const std::vector<int> &element_shape, in Init()
44 void TensorArray::set_element_shape(const std::vector<int> &element_shape) { in set_element_shape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/control/ |
| D | tensorlist_getitem_infer.c | 65 int element_shape[MAX_SHAPE_SIZE] = {0}; in TensorListGetItemInferShape() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/control/ |
| D | tensor_array.cc | 45 int *element_shape = this->ta_param_->element_shape_; in Init() local
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | tensorlist.h | 72 std::vector<int> &element_shape() { return element_shape_; } in element_shape() function
|
| D | tensorlist.cc | 26 TensorList::TensorList(std::vector<int> shape, std::vector<int> element_shape, Category category) in TensorList()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | add_tensor_array.cc | 167 std::vector<int> element_shape; in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | data_utils.cc | 695 TensorShape element_shape(std::vector<int64_t>(tensor_shape.begin() + 1, tensor_shape.end())); in BatchTensorToCVTensorVector() local
721 TensorShape element_shape(std::vector<int64_t>(tensor_shape.begin() + 1, tensor_shape.end())); in BatchTensorToTensorVector() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | common_infer.c | 49 int TensorListMergeShape(int *element_shape, size_t *element_shape_size, const int *tmp, size_t tmp… in TensorListMergeShape()
|
| /third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/ |
| D | infershape_pass.cc | 85 std::vector<int> element_shape; in ConvertTensorList() local
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | abstract_value.cc | 301 BaseShapePtr element_shape = ele->BuildShape(); in ElementsShape() local
|