Searched refs:shape2D (Results 1 – 1 of 1) sorted by relevance
585 TensorShape shape2D = input->tensor_shape(); variable590 shape2D.collapse_from(axis);593 shape2D.collapse(shape2D.num_dimensions() - 1);598 shape2D.collapse(shape2D.num_dimensions());605 shape2D.shift_right(1);608 return shape2D;