Searched defs:new_batch_dim (Results 1 – 1 of 1) sorted by relevance
658 int64_t new_batch_dim, new_spatial_dim; in BringSpaceNextToBatch() local1209 const int64_t new_batch_dim = in CanPropagate() local1250 const int64_t new_batch_dim = in CanPropagate() local1373 const int64_t new_batch_dim = in CanPropagate() local1421 const int64_t new_batch_dim = DimLookUp(permute_dims, old_batch_dim); in PropagateOnBroadcast() local1924 const int64_t new_batch_dim = DimLookUp(permute_dims, old_batch_dim); in Propagate() local1967 int64_t new_batch_dim, new_space_dim, new_feature_dim; in Propagate() local2017 const int64_t new_batch_dim = DimLookUp(permute_dims, old_batch_dim); in Propagate() local2288 HloInstruction* select_val, int64_t new_batch_dim, in SelectValidPortion()2976 const int64_t new_batch_dim = DimLookUp(prod_transpose_dims, old_batch_dim); in PropagateOnConstant() local