Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgemm_rewrite_test.cc438 …reduce.1 = f32[] reduce(%multiply.1, %constant_1), dimensions={0}, to_apply=%scalar_add_computation
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc517 HloComputation*& scalar_add_computation = scalar_add_computations_[type]; in GetOrCreateScalarAddComputation() local
518 if (scalar_add_computation) { in GetOrCreateScalarAddComputation()
519 return scalar_add_computation; in GetOrCreateScalarAddComputation()
531 scalar_add_computation = in GetOrCreateScalarAddComputation()
533 return scalar_add_computation; in GetOrCreateScalarAddComputation()