Home
last modified time | relevance | path

Searched defs:TensorIterator (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/utils/
Dtensor_iterator.h25 explicit TensorIterator(const std::vector<int64_t> &shape) in TensorIterator() function
27 TensorIterator(const std::vector<int64_t> &shape, const std::vector<int64_t> &cur_coord) in TensorIterator() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/utils/
Dtensor_iterator.h25 explicit TensorIterator(const std::vector<int64_t> &shape) in TensorIterator() function
27 TensorIterator(const std::vector<int64_t> &shape, const std::vector<int64_t> &cur_coord) in TensorIterator() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/
Dtensor.h599 TensorIterator(const TensorIterator<T> &raw_iterator) { ptr_ = raw_iterator.ptr_; } in TensorIterator() function
692 TensorIterator(const TensorIterator<std::string_view, DUMMY> &raw_iterator) { in TensorIterator() function
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/dataset/
Dtensor_test.cc427 TEST_F(MindDataTestTensorDE, TensorIterator) { in TEST_F() argument