Searched refs:shape_in_range (Results 1 – 1 of 1) sorted by relevance
389 auto shape_in_range = [&](const Shape& s) { in GetIndexTypeForKernel() local408 if (!shape_in_range(unnested_hlo->shape())) { in GetIndexTypeForKernel()413 return shape_in_range(operand->shape()); in GetIndexTypeForKernel()437 auto shape_in_range = [&](const Shape& s) { in GetIndexTypeForKernelFromMlir() local457 if (!shape_in_range(TypeToShape(result.getType()))) { in GetIndexTypeForKernelFromMlir()463 return shape_in_range(TypeToShape(operand.getType())); in GetIndexTypeForKernelFromMlir()