Searched refs:ReduceAll (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | quantize_and_dequantize_op.cc | 76 min_range = ReduceAll(input, xla::MaxValue(b, xla_type), *fmin); in Compile() 77 max_range = ReduceAll(input, xla::MinValue(b, xla_type), *fmax); in Compile()
|
D | fake_quantize_ops.cc | 313 xla::XlaOp reduce1 = xla::ReduceAll( in Compile() 322 xla::XlaOp reduce2 = xla::ReduceAll( in Compile()
|
/external/tensorflow/tensorflow/core/api_def/base_api/ |
D | api_def_All.pbtxt | 7 name: "ReduceAll"
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | svd_test.cc | 101 return ReduceAll(Abs(m1 - m2), ConstantR0WithType(builder, F32, 0), in GetAverageAbsoluteError()
|
D | self_adjoint_eig_test.cc | 121 return ReduceAll(Abs(m1 - m2), ConstantR0WithType(builder, F32, 0), in GetAverageAbsoluteError()
|
D | self_adjoint_eig.cc | 233 auto tol_cond = ReduceAll(Lt(tol, norms.off_diagonal_norm), in WhileLoopFn()
|
D | svd.cc | 602 auto tolerance_cond = ReduceAll( in WhileLoopFn()
|
D | math.cc | 1446 ReduceAll(tolerance_comparison, ConstantR0<bool>(body_builder, false), in LentzThompsonBarnettAlgorithm()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.h | 495 XlaOp ReduceAll(XlaOp operand, XlaOp init_value, 908 friend XlaOp ReduceAll(XlaOp operand, XlaOp init_value, 1631 XlaOp ReduceAll(XlaOp operand, XlaOp init_value,
|
D | xla_builder.cc | 1950 XlaOp XlaBuilder::ReduceAll(XlaOp operand, XlaOp init_value, in ReduceAll() function in xla::XlaBuilder 3274 XlaOp ReduceAll(const XlaOp operand, const XlaOp init_value, in ReduceAll() function 3276 return operand.builder()->ReduceAll(operand, init_value, computation); in ReduceAll()
|
/external/tensorflow/tensorflow/compiler/xrt/tests/ |
D | raw_api_test.cc | 264 xla::ReduceAll(icmp, xla::Zero(&builder, xla::S32), in IsEqualComputation()
|