Searched defs:old_batch_size (Results 1 – 1 of 1) sorted by relevance
325 const int64 old_batch_size = in IsConvSuitableForSpaceToBatch() local345 int64 activations_batch_dim, int64 old_batch_size, int64 low_padding, in HaloDuplicateWithSlice()570 HloInstruction* activations, int64 batch_dimension, int64 old_batch_size, in IncreaseSpatialSizeOnSpaceToBatchedShape()1259 const int64 old_batch_size = in Propagate() local1660 const int64 old_batch_size = old_shape.dimensions(old_batch_dim); in SelectValidPortion() local1717 const int64 old_batch_size = old_instr->shape().dimensions(old_batch_dim); in BatchToSpace() local1877 const int64 old_batch_size = in PropagateOnConv() local2041 const int64 old_batch_size = in SplitSpaceHelper() local2132 const int64 old_batch_size = producer->shape().dimensions(old_batch_dim); in PropagateOnConstant() local2169 const int64 old_batch_size = in PropagateOnBackpropFilterConv() local[all …]