Searched defs:new_batch_size (Results 1 – 1 of 1) sorted by relevance
576 const int64 new_batch_size = activations->shape().dimensions(batch_dimension); in IncreaseSpatialSizeOnSpaceToBatchedShape() local1421 const int64 new_batch_size = new_shape.dimensions(new_batch_dim); in Propagate() local1658 const int64 new_batch_size = new_shape.dimensions(new_batch_dim); in SelectValidPortion() local2133 const int64 new_batch_size = new_producer->shape().dimensions(new_batch_dim); in PropagateOnConstant() local