Searched defs:old_batch_size (Results 1 – 1 of 1) sorted by relevance
448 const int64_t old_batch_size = in IsConvSuitableForSpaceToBatch() local745 int64_t old_batch_size, absl::Span<const int64_t> spatial_dimensions) { in SplitAndTransposeMergedBatch()797 int64_t old_batch_size, absl::Span<const int64_t> spatial_dimensions, in ChangeSpatialSizeOnSpaceToBatchedShape()1796 const int64_t old_batch_size = in Propagate() local2043 const int64_t old_batch_size = old_shape.dimensions(old_batch_dim); in Propagate() local2298 const int64_t old_batch_size = old_shape.dimensions(old_batch_dim); in SelectValidPortion() local2379 const int64_t old_batch_size = old_instr->shape().dimensions(old_batch_dim); in BatchToSpace() local2567 const int64_t old_batch_size = in PropagateOnConv() local2815 int64_t activations_batch_dim, int64_t old_batch_size) { in TransposeAndMergeBatch()2859 const int64_t old_batch_size = in PerformSplitSpace() local[all …]