Searched refs:batch_elements (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | padded_batch_dataset_op.cc | 199 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
207 batch_elements.reserve(dataset()->batch_size_); in GetNextInternal()
214 batch_elements.push_back(std::move(batch_element_tuple)); in GetNextInternal()
223 if (batch_elements.empty()) { in GetNextInternal()
229 batch_elements.size() < dataset()->batch_size_) { in GetNextInternal()
234 TF_RETURN_IF_ERROR(CopyBatch(ctx, batch_elements, out_tensors)); in GetNextInternal()
281 const std::vector<std::vector<Tensor>>& batch_elements, in CopyBatch() argument
283 const size_t num_tuple_components = batch_elements[0].size(); in CopyBatch()
284 const int64_t num_batch_elements = batch_elements.size(); in CopyBatch()
302 batch_elements[i][component_index].shape(); in CopyBatch()
[all …]
|
| D | batch_dataset_op.cc | 145 std::vector<std::vector<Tensor>> batch_elements; in Get() local
151 batch_elements.emplace_back(std::move(batch_element_tuple)); in Get()
153 TF_RETURN_IF_ERROR(CopyBatch(CopyBatchParams(ctx), batch_elements, in Get()
192 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
199 batch_elements.reserve(dataset()->reserve_size_); in GetNextInternal()
206 batch_elements.emplace_back(std::move(batch_element_tuple)); in GetNextInternal()
213 if (batch_elements.empty()) { in GetNextInternal()
219 batch_elements.size() < dataset()->batch_size_) { in GetNextInternal()
233 CopyBatchParams(ctx), batch_elements, dataset()->parallel_copy_, in GetNextInternal()
|
| D | parallel_batch_dataset_op.cc | 372 auto batch_elements = in CallBatching() local
374 batch_elements->reserve(dataset()->reserve_size_); in CallBatching()
388 batch_elements->emplace_back(std::move(batch_element_tuple)); in CallBatching()
396 if (batch_elements->empty()) { in CallBatching()
401 auto copy_elements_fn = [this, ctx, result, batch_elements]() { in CallBatching()
412 status = CopyBatch(CopyBatchParams(ctx.get()), *batch_elements, in CallBatching()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | sliding_window_dataset_op.cc | 172 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
175 batch_elements.reserve(window_size); in GetNextInternal()
200 batch_elements.emplace_back(buffer_[i]); in GetNextInternal()
222 const size_t num_tuple_components = batch_elements[0].size(); in GetNextInternal()
223 const int64_t num_batch_elements = batch_elements.size(); in GetNextInternal()
226 const Tensor& first_element = batch_elements[0][component_index]; in GetNextInternal()
235 if (batch_elements[i][component_index].shape() != in GetNextInternal()
242 batch_elements[i][component_index].shape().DebugString(), in GetNextInternal()
246 std::move(batch_elements[i][component_index]), &batch_component, in GetNextInternal()
|
| D | dense_to_sparse_batch_dataset_op.cc | 169 std::vector<Tensor> batch_elements; in GetNextInternal() local
171 batch_elements.reserve( in GetNextInternal()
201 batch_elements.push_back(std::move(batch_element_tuple[0])); in GetNextInternal()
232 if (batch_elements.empty()) { in GetNextInternal()
249 for (int64_t i = 0; i < batch_elements.size(); ++i) { in GetNextInternal()
250 const Tensor& t = batch_elements[i]; in GetNextInternal()
277 dense_shape_vec(0) = batch_elements.size(); in GetNextInternal()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | partial_batch_padding_handler.py | 80 batch_elements = []
82 batch_elements.append(_pad(batch_element))
83 return tuple(batch_elements)
|
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset_utils.cc | 740 const std::vector<std::vector<Tensor>>& batch_elements, in CopyBatch() argument
744 const size_t num_tuple_components = batch_elements.at(0).size(); in CopyBatch()
746 const int64_t num_batch_elements = batch_elements.size(); in CopyBatch()
749 const Tensor& first_element = batch_elements.at(0)[component_index]; in CopyBatch()
767 const Tensor& first_element = batch_elements.at(0)[component_index]; in CopyBatch()
771 auto copy_element_fn = [component_index, &batch_elements, &batch_component, in CopyBatch()
773 if (batch_elements.at(index)[component_index].shape() != in CopyBatch()
780 batch_elements.at(index)[component_index].shape().DebugString(), in CopyBatch()
784 std::move(batch_elements.at(index)[component_index]), in CopyBatch()
|
| D | dataset_utils.h | 309 const std::vector<std::vector<Tensor>>& batch_elements,
|