Searched refs:TensorQTable (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | batch_op.cc | 103 std::unique_ptr<TensorQTable> table = std::make_unique<TensorQTable>(); in operator ()()
116 table = std::make_unique<TensorQTable>(); in operator ()()
127 table = std::make_unique<TensorQTable>(); // this drops when drop == true in operator ()()
169 Status BatchOp::BatchRows(const std::unique_ptr<TensorQTable> *src, TensorRow *dest, dsize_t batch_… in BatchRows()
231 std::pair<std::unique_ptr<TensorQTable>, CBatchInfo> table_pair; in WorkerEntry()
248 Status BatchOp::MakeBatchedRow(std::pair<std::unique_ptr<TensorQTable>, CBatchInfo> table_pair, Ten… in MakeBatchedRow() argument
281 Status BatchOp::MapColumns(std::pair<std::unique_ptr<TensorQTable>, CBatchInfo> *table_pair) { in MapColumns() argument
284 std::unique_ptr<TensorQTable> in_q_table = std::move(table_pair->first); in MapColumns()
286 …auto out_q_table = std::make_unique<TensorQTable>(num_rows, TensorRow(column_name_id_map_.size(), … in MapColumns()
427 Status BatchOp::PadColumns(std::unique_ptr<TensorQTable> *table, const PadInfo &pad_info, in PadColumns()
[all …]
|
| D | batch_op.h | 201 …static Status BatchRows(const std::unique_ptr<TensorQTable> *src, TensorRow *dest, dsize_t batch_s…
207 static Status PadColumns(std::unique_ptr<TensorQTable> *table, const PadInfo &pad_info,
223 …Status MakeBatchedRow(std::pair<std::unique_ptr<TensorQTable>, CBatchInfo> table_pair, TensorRow *…
229 Status MapColumns(std::pair<std::unique_ptr<TensorQTable>, CBatchInfo> *table_pair);
278 …QueueList<std::pair<std::unique_ptr<TensorQTable>, CBatchInfo>> worker_queues_; // internal queue…
|
| D | shuffle_op.cc | 110 std::unique_ptr<TensorQTable> new_buffer_table; // A tensor table to be used for output. in operator ()()
139 new_buffer_table = std::make_unique<TensorQTable>(); in operator ()()
|
| D | bucket_batch_by_length_op.cc | 48 buckets_.push_back(std::make_unique<TensorQTable>()); in BucketBatchByLengthOp()
131 std::unique_ptr<TensorQTable> *bucket = &buckets_[bucket_index]; in PadAndBatchBucket()
|
| D | bucket_batch_by_length_op.h | 84 std::vector<std::unique_ptr<TensorQTable>> buckets_;
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | random_data_op.cc | 236 std::unique_ptr<TensorQTable> new_tensor_table = nullptr; in WorkerEntry()
246 new_tensor_table = std::make_unique<TensorQTable>(); in WorkerEntry()
|
| D | csv_op.h | 192 …Status LoadTensor(const std::string &line, std::unique_ptr<TensorQTable> *tensor_table, int64_t ro…
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/ |
| D | tensor_row.h | 32 using TensorQTable = std::deque<TensorRow>; // A different flavour of tensor table, this one has q… variable
|