Home
last modified time | relevance | path

Searched refs:GetOrCreateScalarAddComputation (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc81 HloComputation* GetOrCreateScalarAddComputation( in GetOrCreateScalarAddComputation() function in xla::__anon262de65b0111::BatchNormExpanderVisitor
241 GetOrCreateScalarAddComputation(ptype); in HandleBatchNormTraining()
525 GetOrCreateScalarAddComputation(ptype); in HandleBatchNormGrad()
Dalgebraic_simplifier.cc318 HloComputation* AddReduce_computation = GetOrCreateScalarAddComputation(); in AddReduce()
399 HloComputation* GetOrCreateScalarAddComputation() { in GetOrCreateScalarAddComputation() function in xla::__anondb6e78140111::AlgebraicSimplifierVisitor