Searched refs:shrink_axis (Results 1 – 4 of 4) sorted by relevance
103 const bool shrink_axis = op_context->params->shrink_axis_mask & (1 << idx); in CheckOutputSize() local104 if (shrink_axis) { in CheckOutputSize()111 if (!shrink_axis) { in CheckOutputSize()
121 const bool shrink_axis = op_context->params->shrink_axis_mask & (1 << idx); in ResizeOutputTensor() local122 if (shrink_axis) { in ResizeOutputTensor()129 if (!shrink_axis) { in ResizeOutputTensor()
135 const bool shrink_axis = shrink_axis_mask & (1 << axis); in StopForAxis() local142 if (shrink_axis) { in StopForAxis()
3090 // This shrink_axis mask is supported because it refers to a major dimension.3100 // This shrink_axis mask is unsupported because it does not refer to a major