Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc47 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.cc110 HloInstruction* hlo, int64 operand_idx, PrimitiveType to, in InsertConvertBeforeOperand()
Dhlo_instruction_test.cc1076 for (int64 operand_idx = 0; operand_idx < fusion->operand_count(); in TEST_F() local
1117 for (int64 operand_idx = 0; operand_idx < fusion->operand_count(); in TEST_F() local
Dhlo_verifier.cc362 int64 operand_idx = parameter_idx / 2; in HandleSort() local
Delemental_ir_emitter.cc1787 for (int64 operand_idx = 0; operand_idx < hlo->operand_count(); in EmitElementalConcatenate() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3380 for (int64 operand_idx = 0; operand_idx < hlo->operand_count(); in CheckAndEmitHloWithTile021() local