Searched refs:scalar_lhs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduction_ops.cc | 38 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer() argument
40 xla::Add(scalar_lhs, scalar_rhs); in BuildReducer()
57 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer() argument
59 xla::Mul(scalar_lhs, scalar_rhs); in BuildReducer()
75 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer() argument
77 xla::Min(scalar_lhs, scalar_rhs); in BuildReducer()
93 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer() argument
95 xla::Max(scalar_lhs, scalar_rhs); in BuildReducer()
111 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer() argument
113 xla::Add(scalar_lhs, scalar_rhs); in BuildReducer()
[all …]
|
| D | reduction_ops.h | 47 const xla::XlaOp& scalar_lhs,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 85 auto scalar_lhs = b.AddInstruction( in GetOrCreateScalarAddComputation() local
90 shape, HloOpcode::kAdd, scalar_lhs, scalar_rhs)); in GetOrCreateScalarAddComputation()
|
| D | algebraic_simplifier.cc | 406 auto scalar_lhs = b.AddInstruction( in GetOrCreateScalarAddComputation() local
411 shape, HloOpcode::kAdd, scalar_lhs, scalar_rhs)); in GetOrCreateScalarAddComputation()
|