Home
last modified time | relevance | path

Searched refs:operand_idx (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc47 optional<int64> operand_idx = ProfitableToMakeDotOperandColumnMajor(*user); in ShouldMakeAllUsersColMajor() local
48 if (!operand_idx || user->operand(*operand_idx) != instruction || in ShouldMakeAllUsersColMajor()
58 optional<int64> operand_idx = in ShouldMakeOperandColumnMajor() local
60 if (!operand_idx) { in ShouldMakeOperandColumnMajor()
64 const HloInstruction* operand = instruction.operand(*operand_idx); in ShouldMakeOperandColumnMajor()
77 return it->second ? operand_idx : nullopt; in ShouldMakeOperandColumnMajor()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization.cc65 Status InsertConvertBeforeOperand(HloInstruction* hlo, int64 operand_idx,
110 HloInstruction* hlo, int64 operand_idx, PrimitiveType to, in InsertConvertBeforeOperand() argument
112 auto operand = hlo->mutable_operand(operand_idx); in InsertConvertBeforeOperand()
115 TF_RETURN_IF_ERROR(hlo->ReplaceOperandWith(operand_idx, convert)); in InsertConvertBeforeOperand()
Dhlo_instruction_test.cc1076 for (int64 operand_idx = 0; operand_idx < fusion->operand_count(); in TEST_F() local
1077 ++operand_idx) { in TEST_F()
1078 const HloInstruction* operand = fusion->operand(operand_idx); in TEST_F()
1080 EXPECT_FALSE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()
1082 EXPECT_TRUE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()
1117 for (int64 operand_idx = 0; operand_idx < fusion->operand_count(); in TEST_F() local
1118 ++operand_idx) { in TEST_F()
1119 if (fusion->operand(operand_idx) == y) { in TEST_F()
1120 EXPECT_FALSE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()
1122 EXPECT_TRUE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()
Dhlo_instructions.cc930 const absl::optional<int64>& operand_idx) const { in IsElementwiseImpl()
1040 const absl::optional<int64>& operand_idx) const { in IsElementwiseImpl()
1179 const absl::optional<int64>& operand_idx) const { in IsElementwiseImpl()
1180 if (!operand_idx.has_value()) { in IsElementwiseImpl()
1192 worklist.push_back(fused_parameter(operand_idx.value())); in IsElementwiseImpl()
1193 visited.insert(fused_parameter(operand_idx.value())); in IsElementwiseImpl()
1620 const absl::optional<int64>& operand_idx) const { in IsElementwiseImpl()
Dhlo_instructions.h593 const absl::optional<int64>& operand_idx) const override;
670 const absl::optional<int64>& operand_idx) const override;
816 const absl::optional<int64>& operand_idx) const override;
846 const absl::optional<int64>& operand_idx) const override;
Dhlo_instruction.h1235 bool IsElementwiseOnOperand(int64 operand_idx) const;
1687 const absl::optional<int64>& operand_idx) const;
Delemental_ir_emitter.cc1787 for (int64 operand_idx = 0; operand_idx < hlo->operand_count(); in EmitElementalConcatenate() local
1788 ++operand_idx) { in EmitElementalConcatenate()
1789 const HloInstruction* operand = hlo->operand(operand_idx); in EmitElementalConcatenate()
1791 exit_block, StrCat("concat_index_not_from_operand", operand_idx), b_); in EmitElementalConcatenate()
Dhlo_instruction.cc2118 const absl::optional<int64>& operand_idx) const { in IsElementwiseImpl()
2175 return operand_idx.has_value() && operand_idx.value() == 0; in IsElementwiseImpl()
2857 bool HloInstruction::IsElementwiseOnOperand(int64 operand_idx) const { in IsElementwiseOnOperand()
2858 return IsElementwiseImpl(operand_idx); in IsElementwiseOnOperand()
Dhlo_verifier.cc362 int64 operand_idx = parameter_idx / 2; in HandleSort() local
364 sort->operand(operand_idx)->shape().element_type(), {}); in HandleSort()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3380 for (int64 operand_idx = 0; operand_idx < hlo->operand_count(); in CheckAndEmitHloWithTile021() local
3381 ++operand_idx) { in CheckAndEmitHloWithTile021()
3382 HloInstruction* operand = hlo->mutable_operand(operand_idx); in CheckAndEmitHloWithTile021()
3397 params_012.push_back(operand_idx); in CheckAndEmitHloWithTile021()