Searched refs:a_literal (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | reduce_precision_test.cc | 233 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 D | scalar_computations_test.cc | 167 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 D | array_elementwise_ops_test.cc | 199 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