Searched refs:batch_component_shape (Results 1 – 3 of 3) sorted by relevance
288 TensorShape batch_component_shape({num_batch_elements}); in CopyBatch() local294 batch_component_shape.AddDim(0); in CopyBatch()296 batch_component_shape.AddDim(padded_shape.dim_size(dim)); in CopyBatch()316 batch_component_shape.dim_size(dim + 1)) { in CopyBatch()317 batch_component_shape.set_dim( in CopyBatch()323 batch_component_shape.dim_size(dim + 1)) { in CopyBatch()336 batch_component_shape); in CopyBatch()344 for (int i = 1; i < batch_component_shape.dims(); ++i) { in CopyBatch()345 component_shape.AddDim(batch_component_shape.dim_size(i)); in CopyBatch()
227 TensorShape batch_component_shape({num_batch_elements}); in GetNextInternal() local228 batch_component_shape.AppendShape(first_element.shape()); in GetNextInternal()230 batch_component_shape); in GetNextInternal()
751 TensorShape batch_component_shape({num_batch_elements}); in CopyBatch() local752 batch_component_shape.AppendShape(first_element_shape); in CopyBatch()754 batch_component_shape); in CopyBatch()