Home
last modified time | relevance | path

Searched refs:MaybeConvertLiteralToBfloat16 (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.cc286 client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument))); in ComputeAndTransfer()
310 client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument))); in ComputeAndCompareLiteralWithStatus()
371 client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument))); in ComputeAndCompareLiteralWithStatus()
632 Literal ClientLibraryTestBase::MaybeConvertLiteralToBfloat16( in MaybeConvertLiteralToBfloat16() function in xla::ClientLibraryTestBase
645 Literal param_literal = MaybeConvertLiteralToBfloat16(literal); in CreateParameterAndTransferLiteral()
Dclient_library_test_base.h418 Literal MaybeConvertLiteralToBfloat16(const Literal& literal);
Dreduce_window_test.cc1275 ComputeAndCompare(&b, {MaybeConvertLiteralToBfloat16(input_literal)}, in DoIt()