Searched defs:new_batch_dim (Results 1 – 1 of 1) sorted by relevance
488 int64 new_batch_dim, new_spatial_dim; in BringSpaceNextToBatch() local831 const int64 new_batch_dim = in CanPropagate() local865 const int64 new_batch_dim = in CanPropagate() local966 const int64 new_batch_dim = in CanPropagate() local1011 const int64 new_batch_dim = DimLookUp(permute_dims, old_batch_dim); in PropagateOnBroadcast() local1369 const int64 new_batch_dim = DimLookUp(permute_dims, old_batch_dim); in Propagate() local1408 const int64 new_batch_dim = DimLookUp(permute_dims, old_batch_dim); in Propagate() local1651 HloInstruction* select_val, int64 new_batch_dim, int64 new_space_dim, in SelectValidPortion()2129 const int64 new_batch_dim = DimLookUp(prod_transpose_dims, old_batch_dim); in PropagateOnConstant() local