Searched refs:old_space_dim (Results 1 – 1 of 1) sorted by relevance
153 int64 old_batch_dim, int64 old_space_dim);827 const int64 old_space_dim = dim_map_val_op_0.second; in CanPropagate() local834 DimLookUp(permute_dims_first_operand, old_space_dim); in CanPropagate()964 const int64 old_space_dim = dim_map_val_op_0.second; in CanPropagate() local969 DimLookUp(permute_dims_first_operand, old_space_dim); in CanPropagate()978 const int64 stride = consumer->window().dimensions(old_space_dim).stride(); in CanPropagate()980 consumer->window().dimensions(old_space_dim).padding_high(); in CanPropagate()982 consumer->window().dimensions(old_space_dim).padding_low(); in CanPropagate()1005 const int64 old_space_dim = dim_map_val.second; in PropagateOnBroadcast() local1012 const int64 new_space_dim = DimLookUp(permute_dims, old_space_dim); in PropagateOnBroadcast()[all …]