Searched defs:operand_idx (Results 1 – 12 of 12) sorted by relevance
47 optional<int64> operand_idx = ProfitableToMakeDotOperandColumnMajor(*user); in ShouldMakeAllUsersColMajor() local58 optional<int64> operand_idx = in ShouldMakeOperandColumnMajor() local
30 for (int operand_idx = 0; operand_idx < instruction->operand_count(); in Run() local
219 HloInstruction* hlo, int64 operand_idx, PrimitiveType from, in InsertConvertBeforeOperand()
1835 for (int64 operand_idx = 0; operand_idx < hlo->operand_count(); in EmitElementalConcatenate() local2688 for (int64 operand_idx = 0; operand_idx < input_count; ++operand_idx) { in EmitElementalReduceWindow() local2698 for (int64 operand_idx = 0; operand_idx < accum_values.size(); in EmitElementalReduceWindow() local
1137 for (int64 operand_idx = 0; operand_idx < fusion->operand_count(); in TEST_F() local1178 for (int64 operand_idx = 0; operand_idx < fusion->operand_count(); in TEST_F() local
567 int64 operand_idx = parameter_idx / 2; in HandleSort() local
143 for (int operand_idx : llvm::seq<int>(0, op->getNumOperands())) { in VerifyControlOperandsAfterAllData() local295 for (int operand_idx : llvm::seq<int>(0, yield.getNumOperands())) { in Verify() local304 for (int operand_idx : llvm::seq<int>(0, island.getNumResults() - 1)) { in Verify() local
1110 unsigned operand_idx = 0; in matchAndRewrite() local1206 auto args = [&](int operand_idx) -> SmallVector<Value, 8> { in GetHoistParams()1216 auto is_all_tensors = [&](int operand_idx, int axis) -> bool { in GetHoistParams()1227 auto is_all_scalars = [&](int operand_idx) -> bool { in GetHoistParams()
505 int32_t operand_idx = operand_type_and_idx.index(); in Verify() local
1584 const auto should_partition_contracting_dim = [&](int64 operand_idx) { in PartitionBaseCase()
2013 for (uint64_t operand_idx = 0; operand_idx < num_elements; operand_idx++) { in fold() local
5099 for (int64 operand_idx = 0; operand_idx < context.operand_shapes.size(); in CheckAndEmitHloWithTile021() local