Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.h554 literal = LiteralUtil::ConvertF32ToBF16(literal); in CreateR0Parameter()
568 literal = LiteralUtil::ConvertF32ToBF16(literal); in CreateR1Parameter()
582 literal = LiteralUtil::ConvertF32ToBF16(literal); in CreateR2Parameter()
596 literal = LiteralUtil::ConvertF32ToBF16(literal); in CreateR3Parameter()
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.cc629 expected = LiteralUtil::ConvertF32ToBF16(expected); in XLA_TEST_P()
782 auto expected = LiteralUtil::ConvertF32ToBF16(input_literal); in XLA_TEST_P()
Dreduce_window_test.cc1025 input_literal = LiteralUtil::ConvertF32ToBF16(input_literal); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc96 /* static */ Literal LiteralUtil::ConvertF32ToBF16( in ConvertF32ToBF16() function in xla::LiteralUtil
Dliteral_util.h262 static Literal ConvertF32ToBF16(const LiteralSlice& f32_literal);
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation_test.cc256 LiteralUtil::ConvertF32ToBF16(LiteralUtil::CreateFromArray(array_a)), in TEST_F()
259 LiteralUtil::ConvertF32ToBF16(LiteralUtil::CreateFromArray(array_b)), in TEST_F()