Lines Matching refs:shrink_axis_mask
38 const int32 new_axis_mask, shrink_axis_mask;
66 int32 shrink_axis_mask;
82 dense->shrink_axis_mask = 0;
134 if (sparse.shrink_axis_mask & (1 << i)) {
136 dense->shrink_axis_mask |= (1 << full_index);
151 int32 new_axis_mask, int32 shrink_axis_mask,
203 shrink_axis_mask};
259 bool shrink_i = (dense_spec.shrink_axis_mask & (1 << i));
374 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape,
382 end_mask_spec, ellipsis_mask, new_axis_mask, shrink_axis_mask,
