Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dcheck_execution_arity_test.cc39 auto param_literal = LiteralUtil::CreateR1<float>({1.1f, 2.2f}); in TEST_F() local
41 auto p0 = Parameter(&builder, 0, param_literal.shape(), "param0"); in TEST_F()
42 auto p1 = Parameter(&builder, 1, param_literal.shape(), "param1"); in TEST_F()
46 client_->TransferToServer(param_literal).ConsumeValueOrDie(); in TEST_F()
48 client_->TransferToServer(param_literal).ConsumeValueOrDie(); in TEST_F()
Dclient_library_test_base.cc645 Literal param_literal = MaybeConvertLiteralToBfloat16(literal); in CreateParameterAndTransferLiteral() local
647 client_->TransferToServer(param_literal, device_handle)); in CreateParameterAndTransferLiteral()
649 Parameter(builder, parameter_number, param_literal.shape(), name); in CreateParameterAndTransferLiteral()
Darray_elementwise_ops_test.cc1584 Literal param_literal = LiteralUtil::CreateR1<float>(values); in XLA_TEST_F() local
1586 client_->TransferToServer(param_literal).ConsumeValueOrDie(); in XLA_TEST_F()
1589 auto param = Parameter(&b, 0, param_literal.shape(), "param"); in XLA_TEST_F()