Searched refs:batch_component (Results 1 – 3 of 3) sorted by relevance
294 Tensor& batch_component = out_tensors->back(); in GetNextInternal() local296 &batch_component, dataset()->padding_values_[component_index])); in GetNextInternal()305 &batch_component, &component_shape](int index) { in GetNextInternal()310 batch_elements[index][component_index], &batch_component, in GetNextInternal()314 batch_elements[index][component_index], &batch_component, in GetNextInternal()
869 Tensor& batch_component = out_tensors->back(); in CopyBatch() local873 &batch_component](int index) { in CopyBatch()876 &batch_component, index)); in CopyBatch()
218 Tensor& batch_component = out_tensors->back(); in GetNextInternal() local233 std::move(batch_elements[i][component_index]), &batch_component, in GetNextInternal()