Searched refs:batch_elements (Results 1 – 5 of 5) sorted by relevance
137 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local144 batch_elements.reserve(dataset()->batch_size_); in GetNextInternal()152 batch_elements.emplace_back(std::move(batch_element_tuple)); in GetNextInternal()159 if (batch_elements.empty()) { in GetNextInternal()165 batch_elements.size() < dataset()->batch_size_) { in GetNextInternal()177 const size_t num_tuple_components = batch_elements[0].size(); in GetNextInternal()178 const int64 num_batch_elements = batch_elements.size(); in GetNextInternal()181 const Tensor& first_element = batch_elements[0][component_index]; in GetNextInternal()195 if (batch_elements[i][component_index].shape() != in GetNextInternal()202 batch_elements[i][component_index].shape().DebugString(), in GetNextInternal()[all …]
227 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local235 batch_elements.reserve(dataset()->batch_size_); in GetNextInternal()242 batch_elements.push_back(std::move(batch_element_tuple)); in GetNextInternal()251 if (batch_elements.empty()) { in GetNextInternal()257 batch_elements.size() < dataset()->batch_size_) { in GetNextInternal()269 const size_t num_tuple_components = batch_elements[0].size(); in GetNextInternal()270 const int64 num_batch_elements = batch_elements.size(); in GetNextInternal()288 batch_elements[i][component_index].shape(); in GetNextInternal()301 if (batch_elements[i][component_index].shape().dim_size(dim) > in GetNextInternal()305 batch_elements[i][component_index].shape().dim_size(dim)); in GetNextInternal()[all …]
147 std::vector<std::vector<Tensor>> batch_elements; in GetNextInternal() local154 batch_elements.reserve(window_size); in GetNextInternal()178 batch_elements.emplace_back(buffer_[window_stride * i]); in GetNextInternal()200 const size_t num_tuple_components = batch_elements[0].size(); in GetNextInternal()201 const int64 num_batch_elements = batch_elements.size(); in GetNextInternal()204 const Tensor& first_element = batch_elements[0][component_index]; in GetNextInternal()213 if (batch_elements[i][component_index].shape() != in GetNextInternal()220 batch_elements[i][component_index].shape().DebugString(), in GetNextInternal()224 std::move(batch_elements[i][component_index]), &batch_component, in GetNextInternal()
161 std::vector<Tensor> batch_elements; in GetNextInternal() local163 batch_elements.reserve( in GetNextInternal()193 batch_elements.push_back(std::move(batch_element_tuple[0])); in GetNextInternal()224 if (batch_elements.empty()) { in GetNextInternal()241 for (int64 i = 0; i < batch_elements.size(); ++i) { in GetNextInternal()242 const Tensor& t = batch_elements[i]; in GetNextInternal()269 dense_shape_vec(0) = batch_elements.size(); in GetNextInternal()
84 batch_elements = []86 batch_elements.append(_pad(batch_element))87 return tuple(batch_elements)