Searched refs:hlo_shape_in_range (Results 1 – 1 of 1) sorted by relevance
412 auto hlo_shape_in_range = [&](const HloInstruction* operand) -> bool { in GetIndexTypeForKernel() local417 if (!absl::c_all_of(unnested_hlo->operands(), hlo_shape_in_range)) { in GetIndexTypeForKernel()425 hlo_shape_in_range)) { in GetIndexTypeForKernel()462 auto hlo_shape_in_range = [&](mlir::Value operand) -> bool { in GetIndexTypeForKernelFromMlir() local467 if (!absl::c_all_of(op->getOperands(), hlo_shape_in_range)) { in GetIndexTypeForKernelFromMlir()475 if (!hlo_shape_in_range(result)) { in GetIndexTypeForKernelFromMlir()