Home
last modified time | relevance | path

Searched refs:ReduceAll (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dquantize_and_dequantize_op.cc76 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()
Dfake_quantize_ops.cc313 xla::XlaOp reduce1 = xla::ReduceAll( in Compile()
322 xla::XlaOp reduce2 = xla::ReduceAll( in Compile()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_All.pbtxt7 name: "ReduceAll"
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsvd_test.cc101 return ReduceAll(Abs(m1 - m2), ConstantR0WithType(builder, F32, 0), in GetAverageAbsoluteError()
Dself_adjoint_eig_test.cc121 return ReduceAll(Abs(m1 - m2), ConstantR0WithType(builder, F32, 0), in GetAverageAbsoluteError()
Dself_adjoint_eig.cc233 auto tol_cond = ReduceAll(Lt(tol, norms.off_diagonal_norm), in WhileLoopFn()
Dsvd.cc602 auto tolerance_cond = ReduceAll( in WhileLoopFn()
Dmath.cc1446 ReduceAll(tolerance_comparison, ConstantR0<bool>(body_builder, false), in LentzThompsonBarnettAlgorithm()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h495 XlaOp ReduceAll(XlaOp operand, XlaOp init_value,
908 friend XlaOp ReduceAll(XlaOp operand, XlaOp init_value,
1631 XlaOp ReduceAll(XlaOp operand, XlaOp init_value,
Dxla_builder.cc1950 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/
Draw_api_test.cc264 xla::ReduceAll(icmp, xla::Zero(&builder, xla::S32), in IsEqualComputation()