Searched refs:shrink_axis (Results 1 – 2 of 2) sorted by relevance
117 const bool shrink_axis = op_context->params->shrink_axis_mask & (1 << idx); in ResizeOutputTensor() local118 if (shrink_axis) { in ResizeOutputTensor()125 if (!shrink_axis) { in ResizeOutputTensor()
121 const bool shrink_axis = shrink_axis_mask & (1 << axis); in StopForAxis() local128 if (shrink_axis) { in StopForAxis()