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() 649 Parameter(builder, parameter_number, param_literal.shape(), name); in CreateParameterAndTransferLiteral()
|
D | array_elementwise_ops_test.cc | 1584 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()
|