Searched refs:eq_computations (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 53 eq_computations) const override;
122 eq_computations) const override;
150 eq_computations) const override;
174 eq_computations) const override;
200 eq_computations) const override;
225 eq_computations) const override;
253 eq_computations) const { in IdenticalSlowPathIgnoringChannelIdValues()
271 eq_computations) const final;
294 eq_computations) const override;
382 eq_computations) const override;
[all …]
|
| D | hlo_instructions.cc | 89 eq_computations) const { in IdenticalSlowPath()
194 eq_computations) const { in IdenticalSlowPath()
234 eq_computations) const { in IdenticalSlowPath()
284 eq_computations) const { in IdenticalSlowPath()
360 eq_computations) const { in IdenticalSlowPath()
402 eq_computations) const { in IdenticalSlowPath()
451 eq_computations) const { in IdenticalSlowPath()
452 if (!IdenticalSlowPathIgnoringChannelIdValues(other, eq_computations)) { in IdenticalSlowPath()
485 eq_computations) const { in IdenticalSlowPathIgnoringChannelIdValues()
606 eq_computations) const { in IdenticalSlowPathIgnoringChannelIdValues()
[all …]
|
| D | hlo_cse.cc | 158 eq_computations = [](const HloComputation* lhs, in Run() local
162 return lhs->Identical(*rhs, eq_instructions, eq_computations, in Run()
|
| D | ar_crs_combiner.cc | 365 auto eq_computations = [](const HloComputation* a, const HloComputation* b) { in InstructionsComputeSameValue() local
374 return i1->Identical(*i2, eq_operands, eq_computations, in InstructionsComputeSameValue()
402 return i1->Identical(*i2, eq_instructions, eq_computations, in InstructionsComputeSameValue()
|
| D | hlo_instruction.h | 1142 eq_computations = std::equal_to<const HloComputation*>(),
1144 return IdenticalInternal(other, eq_operands, eq_computations,
1156 eq_computations = std::equal_to<const HloComputation*>(),
1158 return IdenticalInternal(other, eq_operands, eq_computations,
1997 eq_computations,
2031 eq_computations) const;
|
| D | hlo_instruction_test.cc | 913 auto eq_computations = [](const HloComputation* a, const HloComputation* b) { in StructuralEqual() local
919 instruction1.Identical(instruction1, eq_operand_shapes, eq_computations)); in StructuralEqual()
921 instruction2.Identical(instruction2, eq_operand_shapes, eq_computations)); in StructuralEqual()
924 instruction1.Identical(instruction2, eq_operand_shapes, eq_computations); in StructuralEqual()
927 eq_computations)); in StructuralEqual()
|
| D | conditional_code_motion.cc | 216 auto eq_computations = [](const HloComputation* a, const HloComputation* b) { in InstructionWithinBranchIdentical() local
234 *instruction, eq_operand, eq_computations, is_layout_sensitive); in InstructionWithinBranchIdentical()
243 *instruction, eq_operand, eq_computations, in InstructionWithinBranchIdentical()
|
| D | hlo_instruction.cc | 2016 eq_computations, in IdenticalInternal()
2051 other, eq_computations); in IdenticalInternal()
2054 return IdenticalSlowPath(other, eq_computations); in IdenticalInternal()
2115 eq_computations) const { in IdenticalSlowPath()
2184 return eq_computations(to_apply(), other.to_apply()); in IdenticalSlowPath()
2187 if (!eq_computations(branch_computation(j), in IdenticalSlowPath()
2194 return (eq_computations(while_body(), other.while_body()) && in IdenticalSlowPath()
2195 eq_computations(while_condition(), other.while_condition())); in IdenticalSlowPath()
|