Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbatchnorm_expander.cc72 HloComputation* GetOrCreateScalarAddComputation( in GetOrCreateScalarAddComputation() function in xla::__anon5ce86e510111::BatchNormExpanderVisitor
209 GetOrCreateScalarAddComputation(ptype); in HandleBatchNormTraining()
479 GetOrCreateScalarAddComputation(ptype); in HandleBatchNormGrad()
H A Dalgebraic_simplifier.h467 HloComputation* GetOrCreateScalarAddComputation(PrimitiveType type) { in GetOrCreateScalarAddComputation() function
H A Dalgebraic_simplifier.cc2116 HloComputation* AddReduce_computation = GetOrCreateScalarAddComputation(type); in AddReduce()