Searched refs:hlo_shape_in_range (Results 1 – 1 of 1) sorted by relevance
325 auto hlo_shape_in_range = [&](const HloInstruction* operand) -> bool { in GetIndexTypeForKernel() local330 if (!absl::c_all_of(unnested_hlo->operands(), hlo_shape_in_range)) { in GetIndexTypeForKernel()338 hlo_shape_in_range)) { in GetIndexTypeForKernel()