Searched refs:param_literal (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | check_execution_arity_test.cc | 39 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()
|
| D | client_library_test_base.cc | 645 Literal param_literal = MaybeConvertLiteralToBfloat16(literal); in CreateParameterAndTransferLiteral() local
647 client_->TransferToServer(param_literal, device_handle) in CreateParameterAndTransferLiteral()
650 Parameter(builder, parameter_number, param_literal.shape(), name); in CreateParameterAndTransferLiteral()
|
| D | array_elementwise_ops_test.cc | 1480 Literal param_literal = LiteralUtil::CreateR1<float>(values); in XLA_TEST_F() local
1482 client_->TransferToServer(param_literal).ConsumeValueOrDie(); in XLA_TEST_F()
1485 auto param = Parameter(&b, 0, param_literal.shape(), "param"); in XLA_TEST_F()
|