Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.h610 literal = LiteralUtil::ConvertF32ToBF16(literal); in CreateR0Parameter()
624 literal = LiteralUtil::ConvertF32ToBF16(literal); in CreateR1Parameter()
638 literal = LiteralUtil::ConvertF32ToBF16(literal); in CreateR2Parameter()
652 literal = LiteralUtil::ConvertF32ToBF16(literal); in CreateR3Parameter()
666 literal = LiteralUtil::ConvertF32ToBF16(literal); in CreateR4Parameter()
680 literal = LiteralUtil::ConvertF32ToBF16(literal); in CreateParameter()
Dclient_library_test_base.cc327 converted_expected = LiteralUtil::ConvertF32ToBF16(expected); in ComputeAndCompareLiteralWithStatus()
384 converted_expected = LiteralUtil::ConvertF32ToBF16(expected); in ComputeAndCompareLiteralWithStatus()
604 ? LiteralUtil::ConvertF32ToBF16(literal) in CreateConstantFromLiteral()
635 return LiteralUtil::ConvertF32ToBF16(literal); in MaybeConvertLiteralToBfloat16()
Dreshape_test.cc662 expected = LiteralUtil::ConvertF32ToBF16(expected); in XLA_TEST_P()
821 auto expected = LiteralUtil::ConvertF32ToBF16(input_literal); in XLA_TEST_P()
Dreduce_window_test.cc1041 input_literal = LiteralUtil::ConvertF32ToBF16(input_literal); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h229 static Literal ConvertF32ToBF16(const LiteralSlice& f32_literal);
Dliteral_util.cc101 /* static */ Literal LiteralUtil::ConvertF32ToBF16( in ConvertF32ToBF16() function in xla::LiteralUtil
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation_test.cc257 LiteralUtil::ConvertF32ToBF16(LiteralUtil::CreateFromArray(array_a)), in TEST_F()
260 LiteralUtil::ConvertF32ToBF16(LiteralUtil::CreateFromArray(array_b)), in TEST_F()
Dalgebraic_simplifier.cc600 return HloInstruction::CreateConstant(LiteralUtil::ConvertF32ToBF16( in MakeScalarInstruction()