Home
last modified time | relevance | path

Searched refs:RngBitGeneratorExpander (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Drng_bit_generator_expander.cc51 bool RngBitGeneratorExpander::InstructionMatchesPattern( in InstructionMatchesPattern()
56 StatusOr<HloComputation*> RngBitGeneratorExpander::GetGeneratorComputation( in GetGeneratorComputation()
101 StatusOr<HloInstruction*> RngBitGeneratorExpander::ExpandInstruction( in ExpandInstruction()
Drng_bit_generator_expander.h29 class RngBitGeneratorExpander : public OpExpanderPass {
31 explicit RngBitGeneratorExpander(RandomAlgorithm default_algorithm) in RngBitGeneratorExpander() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc294 pipeline.AddPass<RngBitGeneratorExpander>(RandomAlgorithm::RNG_PHILOX); in RunHloPassesThroughLayoutAssn()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc167 pipeline.AddPass<RngBitGeneratorExpander>(RandomAlgorithm::RNG_PHILOX); in OptimizeHloModule()