Searched refs:OutputTypeAfterChange (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation.cc | 57 if (OutputTypeAfterChange(fusion, index) == BF16) { in DetermineFusionComputationPrecision()
182 if (OutputTypeAfterChange(while_hlo, index) == BF16) { in DetermineWhileComputationsPrecision()
250 if (OutputTypeAfterChange(fused_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
258 if (OutputTypeAfterChange(cond_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
264 if (OutputTypeAfterChange(body_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
285 if (OutputTypeAfterChange(use.instruction, use_output_index) == in AllUsersConsumeBF16()
294 if (OutputTypeAfterChange(use.instruction, use_output_index) == in AllUsersConsumeBF16()
299 if (OutputTypeAfterChange(use.instruction, use.operand_index) == in AllUsersConsumeBF16()
343 if (OutputTypeAfterChange(hlo, index) != F32) { in DetermineInstructionPrecision()
379 if (OutputTypeAfterChange(hlo, index) == F32 && in DetermineInstructionPrecision()
[all …]
|
| D | bfloat16_propagation.h | 171 PrimitiveType OutputTypeAfterChange(HloInstruction* hlo,
|