Searched refs:a_literal (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dreduce_precision_test.cc233 std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({input_values}); local
235 client_->TransferToServer(*a_literal).ConsumeValueOrDie();
236 auto a = builder.Parameter(0, a_literal->shape(), "a");
257 std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({1.00001}); local
259 client_->TransferToServer(*a_literal).ConsumeValueOrDie();
260 auto a = builder.Parameter(0, a_literal->shape(), "a");
286 std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({1.00001}); local
288 client_->TransferToServer(*a_literal).ConsumeValueOrDie();
289 auto a = builder.Parameter(0, a_literal->shape(), "a");
312 std::unique_ptr<Literal> a_literal local
336 std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({1.00001}); local
361 std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({1.00001}); local
[all...]
H A Dscalar_computations_test.cc167 std::unique_ptr<Literal> a_literal = Literal::CreateR0<int64>(value); local
169 client_->TransferToServer(*a_literal).ConsumeValueOrDie();
233 std::unique_ptr<Literal> a_literal = Literal::CreateR0<float>(2.1f); local
238 client_->TransferToServer(*a_literal).ConsumeValueOrDie();
244 ComputationDataHandle a = builder.Parameter(0, a_literal->shape(), "a");
H A Darray_elementwise_ops_test.cc199 std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({a_values}); local
201 client_->TransferToServer(*a_literal).ConsumeValueOrDie();
203 auto a_param = builder.Parameter(0, a_literal->shape(), "a_param");
208 auto b_constant = builder.Parameter(1, a_literal->shape(), "b_param");
2658 std::unique_ptr<Literal> a_literal = Literal::CreateR4FromArray4DWithLayout( local
2660 auto a = builder.ConstantLiteral(*a_literal);

Completed in 170 milliseconds