Searched defs:new_space_dim (Results 1 – 1 of 1) sorted by relevance
833 const int64 new_space_dim = in CanPropagate() local968 const int64 new_space_dim = in CanPropagate() local1012 const int64 new_space_dim = DimLookUp(permute_dims, old_space_dim); in PropagateOnBroadcast() local1184 const int64 new_space_dim = DimLookUp(permute_dims, old_space_dim); in SupportedOpForPropagation() local1370 const int64 new_space_dim = DimLookUp(permute_dims, old_space_dim); in Propagate() local1409 const int64 new_space_dim = DimLookUp(permute_dims, old_space_dim); in Propagate() local1651 HloInstruction* select_val, int64 new_batch_dim, int64 new_space_dim, in SelectValidPortion()2130 const int64 new_space_dim = DimLookUp(prod_transpose_dims, old_space_dim); in PropagateOnConstant() local2234 const int64 new_space_dim = DimLookUp(permute_dims, old_space_dim); in PropagateOnBackpropFilterConv() local