Home
last modified time | relevance | path

Searched refs:NanValue (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconstants.h155 XlaOp NanValue(XlaBuilder* builder, PrimitiveType type);
Dconstants_test.cc160 NanValue(&builder, F32); in XLA_TEST_F()
Dconstants.cc126 XlaOp NanValue(XlaBuilder* builder, PrimitiveType type) { in NanValue() function
Dmath.cc1845 result = Select(result_is_nan, NanValue(&builder, element_type), result); in RegularizedIncompleteBeta()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_rewriter.cc116 LiteralUtil::NanValue(a_shape.element_type())); in CreateCholesky()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc129 xla::NanValue(b, ctx->output_xla_type(kVarianceOutputIndex)), in CompileImpl()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h117 static StatusOr<Literal> NanValue(PrimitiveType primitive_type);
Dliteral_util.cc293 /* static */ StatusOr<Literal> LiteralUtil::NanValue( in NanValue() function in xla::LiteralUtil