Home
last modified time | relevance | path

Searched defs:operand_idx (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc46 optional<int64> operand_idx = ProfitableToMakeDotOperandColumnMajor(*user); in ShouldMakeAllUsersColMajor() local
58 optional<int64> operand_idx = in ShouldMakeOperandColumnMajor() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization.cc104 HloInstruction* hlo, int64 operand_idx, PrimitiveType to, in InsertConvertBeforeOperand()
Dhlo_instruction_test.cc1021 for (int64 operand_idx = 0; operand_idx < fusion->operand_count(); in TEST_F() local
1064 for (int64 operand_idx = 0; operand_idx < fusion->operand_count(); in TEST_F() local
Delemental_ir_emitter.cc1434 for (int64 operand_idx = 0; operand_idx < hlo->operand_count(); in MakeElementGenerator() local
Dhlo_instruction.cc1521 for (int64 operand_idx = 0; in CloneFusionWithNewOperands() local