Searched refs:batch_elements (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | padded_batch_dataset_op.cc | 197 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
205 batch_elements.reserve(dataset()->batch_size_); in GetNextInternal()
212 batch_elements.push_back(std::move(batch_element_tuple)); in GetNextInternal()
221 if (batch_elements.empty()) { in GetNextInternal()
227 batch_elements.size() < dataset()->batch_size_) { in GetNextInternal()
240 const size_t num_tuple_components = batch_elements[0].size(); in GetNextInternal()
241 const int64 num_batch_elements = batch_elements.size(); in GetNextInternal()
259 batch_elements[i][component_index].shape(); in GetNextInternal()
272 if (batch_elements[i][component_index].shape().dim_size(dim) > in GetNextInternal()
276 batch_elements[i][component_index].shape().dim_size(dim)); in GetNextInternal()
[all …]
|
| D | batch_dataset_op.cc | 162 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
169 batch_elements.reserve(dataset()->reserve_size_); in GetNextInternal()
176 batch_elements.emplace_back(std::move(batch_element_tuple)); in GetNextInternal()
183 if (batch_elements.empty()) { in GetNextInternal()
189 batch_elements.size() < dataset()->batch_size_) { in GetNextInternal()
203 ctx, out_tensors, &batch_elements)); in GetNextInternal()
|
| D | parallel_batch_dataset_op.cc | 332 auto batch_elements = in CallBatching() local
334 batch_elements->reserve(dataset()->reserve_size_); in CallBatching()
348 batch_elements->emplace_back(std::move(batch_element_tuple)); in CallBatching()
356 if (batch_elements->empty()) { in CallBatching()
362 auto copy_elements_fn = [this, ctx, result, batch_elements]() { in CallBatching()
367 &result->output, batch_elements.get()); in CallBatching()
|
| D | dataset_utils.cc | 849 std::vector<std::vector<Tensor>>* batch_elements) { in CopyBatch() argument
850 const size_t num_tuple_components = (*batch_elements)[0].size(); in CopyBatch()
852 const int64 num_batch_elements = batch_elements->size(); in CopyBatch()
855 const Tensor& first_element = (*batch_elements)[0][component_index]; in CopyBatch()
872 auto copy_element_fn = [component_index, &batch_elements, in CopyBatch()
875 std::move((*batch_elements)[index][component_index]), in CopyBatch()
887 if ((*batch_elements)[i][component_index].shape() != in CopyBatch()
894 (*batch_elements)[i][component_index].shape().DebugString(), "."); in CopyBatch()
|
| D | dataset_utils.h | 331 std::vector<std::vector<Tensor>>* batch_elements);
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | sliding_window_dataset_op.cc | 156 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local
163 batch_elements.reserve(window_size); in GetNextInternal()
187 batch_elements.emplace_back(buffer_[window_stride * i]); in GetNextInternal()
209 const size_t num_tuple_components = batch_elements[0].size(); in GetNextInternal()
210 const int64 num_batch_elements = batch_elements.size(); in GetNextInternal()
213 const Tensor& first_element = batch_elements[0][component_index]; in GetNextInternal()
222 if (batch_elements[i][component_index].shape() != in GetNextInternal()
229 batch_elements[i][component_index].shape().DebugString(), in GetNextInternal()
233 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 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 | 84 batch_elements = []
86 batch_elements.append(_pad(batch_element))
87 return tuple(batch_elements)
|