Searched refs:AddToOrRemoveFromBF16ChangeSet (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation.cc | 58 AddToOrRemoveFromBF16ChangeSet(root, index, BF16); in DetermineFusionComputationPrecision()
183 AddToOrRemoveFromBF16ChangeSet(body_root, index, BF16); in DetermineWhileComputationsPrecision()
381 AddToOrRemoveFromBF16ChangeSet(hlo, index, BF16); in DetermineInstructionPrecision()
427 AddToOrRemoveFromBF16ChangeSet(parameter, index, operand_type); in AdjustCalledComputationParameters()
466 AddToOrRemoveFromBF16ChangeSet(root, index, output_type); in AdjustCalledComputationRoot()
543 AddToOrRemoveFromBF16ChangeSet(hlo, index, type); in ResolveInconsistencyOfAliasingBuffersHelper()
849 void BFloat16Propagation::AddToOrRemoveFromBF16ChangeSet( in AddToOrRemoveFromBF16ChangeSet() function in xla::BFloat16Propagation
|
| D | bfloat16_propagation.h | 182 void AddToOrRemoveFromBF16ChangeSet(HloInstruction* hlo,
|