Lines Matching refs:shrink_axis_mask
522 int32 shrink_axis_mask) {
539 {"shrink_axis_mask", shrink_axis_mask},
563 int32 shrink_axis_mask) {
582 {"shrink_axis_mask", shrink_axis_mask},
616 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0,
620 ellipsis_mask, new_axis_mask, shrink_axis_mask);
632 new_axis_mask, shrink_axis_mask);
643 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0,
647 ellipsis_mask, new_axis_mask, shrink_axis_mask);
659 new_axis_mask, shrink_axis_mask);
671 int begin_mask = 4, end_mask = 2, new_axis_mask = 8, shrink_axis_mask = 1,
675 ellipsis_mask, new_axis_mask, shrink_axis_mask);
687 new_axis_mask, shrink_axis_mask);
699 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 1,
703 ellipsis_mask, new_axis_mask, shrink_axis_mask);
715 new_axis_mask, shrink_axis_mask);
