Searched refs:AddToOrRemoveFromBF16ChangeSet (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation.cc | 60 AddToOrRemoveFromBF16ChangeSet(root, index, BF16); in DetermineFusionComputationPrecision()
185 AddToOrRemoveFromBF16ChangeSet(body_root, index, BF16); in DetermineWhileComputationsPrecision()
220 AddToOrRemoveFromBF16ChangeSet(root, index, BF16); in DetermineConditionalComputationsPrecision()
444 AddToOrRemoveFromBF16ChangeSet(hlo, index, BF16); in DetermineInstructionPrecision()
490 AddToOrRemoveFromBF16ChangeSet(parameter, index, operand_type); in AdjustCalledComputationParameters()
535 AddToOrRemoveFromBF16ChangeSet(root, index, output_type); in AdjustCalledComputationRoot()
652 AddToOrRemoveFromBF16ChangeSet(hlo, index, type); in ResolveInconsistencyOfAliasingBuffersHelper()
1003 void BFloat16Propagation::AddToOrRemoveFromBF16ChangeSet( in AddToOrRemoveFromBF16ChangeSet() function in xla::BFloat16Propagation
|
| D | bfloat16_propagation.h | 187 void AddToOrRemoveFromBF16ChangeSet(HloInstruction* hlo,
|