Searched refs:scalar_rhs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduction_ops.cc | 39 const xla::XlaOp& scalar_rhs) override { in BuildReducer() argument
40 xla::Add(scalar_lhs, scalar_rhs); in BuildReducer()
58 const xla::XlaOp& scalar_rhs) override { in BuildReducer() argument
59 xla::Mul(scalar_lhs, scalar_rhs); in BuildReducer()
76 const xla::XlaOp& scalar_rhs) override { in BuildReducer() argument
77 xla::Min(scalar_lhs, scalar_rhs); in BuildReducer()
94 const xla::XlaOp& scalar_rhs) override { in BuildReducer() argument
95 xla::Max(scalar_lhs, scalar_rhs); in BuildReducer()
112 const xla::XlaOp& scalar_rhs) override { in BuildReducer() argument
113 xla::Add(scalar_lhs, scalar_rhs); in BuildReducer()
[all …]
|
| D | reduction_ops.h | 48 const xla::XlaOp& scalar_rhs) = 0;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 87 auto scalar_rhs = b.AddInstruction( in GetOrCreateScalarAddComputation() local
90 shape, HloOpcode::kAdd, scalar_lhs, scalar_rhs)); in GetOrCreateScalarAddComputation()
|
| D | algebraic_simplifier.cc | 408 auto scalar_rhs = b.AddInstruction( in GetOrCreateScalarAddComputation() local
411 shape, HloOpcode::kAdd, scalar_lhs, scalar_rhs)); in GetOrCreateScalarAddComputation()
|