Home
last modified time | relevance | path

Searched refs:MaxFiniteValue (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconstants_test.cc116 MaxFiniteValue(&builder, S32); in XLA_TEST_F()
128 MaxFiniteValue(&builder, F32); in XLA_TEST_F()
Dconstants.h152 XlaOp MaxFiniteValue(XlaBuilder* builder, PrimitiveType type);
Dconstants.cc109 XlaOp MaxFiniteValue(XlaBuilder* builder, PrimitiveType type) { in MaxFiniteValue() function
Dmath.cc934 XlaOp underflow = Lt(ax, -Log(MaxFiniteValue(&b, type))); in Igamma()
981 XlaOp underflow = Lt(ax, -Log(MaxFiniteValue(&b, type))); in IgammaGradA()
1028 XlaOp underflow = Lt(ax, -Log(MaxFiniteValue(&b, type))); in RandomGammaGrad()
1072 XlaOp underflow = Lt(ax, -Log(MaxFiniteValue(&b, type))); in Igammac()
1205 auto sqrt_max_value = Sqrt(MaxFiniteValue(b, shape.element_type())); in Acosh()
1254 auto sqrt_max_value = Sqrt(MaxFiniteValue(b, shape.element_type())); in Asinh()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dquantize_and_dequantize_op.cc147 xla::MaxFiniteValue(b, xla_type)); in Compile()
150 xla::MaxFiniteValue(b, xla_type)); in Compile()
Dsegment_reduction_ops.cc164 return xla::MaxFiniteValue(builder, type_); in InitialValue()