| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| reduce_precision_test.cc | 232 Literal a_literal = LiteralUtil::CreateR1<float>({input_values}); local
234 client_->TransferToServer(a_literal).ConsumeValueOrDie();
235 auto a = Parameter(&builder, 0, a_literal.shape(), "a");
255 Literal a_literal = LiteralUtil::CreateR1<float>({1.00001}); local
257 client_->TransferToServer(a_literal).ConsumeValueOrDie();
258 auto a = Parameter(&builder, 0, a_literal.shape(), "a");
284 Literal a_literal = LiteralUtil::CreateR1<float>({1.00001, 1.00001}); local
286 client_->TransferToServer(a_literal).ConsumeValueOrDie();
287 auto a = Parameter(&builder, 0, a_literal.shape(), "a");
310 Literal a_literal = LiteralUtil::CreateR1<float>({1.00001, 1.00001}) local
334 Literal a_literal = LiteralUtil::CreateR1<float>({1.00001}); local
359 Literal a_literal = LiteralUtil::CreateR1<float>({1.00001, 1.00001}); local
[all...] |
| scalar_computations_test.cc | 164 Literal a_literal = LiteralUtil::CreateR0<int64>(value); local
166 client_->TransferToServer(a_literal).ConsumeValueOrDie();
228 Literal a_literal = LiteralUtil::CreateR0<float>(2.1f); local
233 client_->TransferToServer(a_literal).ConsumeValueOrDie();
239 XlaOp a = Parameter(&builder, 0, a_literal.shape(), "a");
|
| concat_test.cc | 497 auto a_literal = LiteralUtil::CreateR1<float>({256.0}); local
498 auto a = Parameter(&builder, 0, a_literal.shape(), "x");
517 auto a_data = client_->TransferToServer(a_literal).ConsumeValueOrDie();
|
| array_elementwise_ops_test.cc | 322 Literal a_literal = LiteralUtil::CreateR1<float>({a_values}); local
324 client_->TransferToServer(a_literal).ConsumeValueOrDie();
326 auto a_param = Parameter(&builder, 0, a_literal.shape(), "a_param");
331 auto b_param = Parameter(&builder, 1, a_literal.shape(), "b_param");
355 auto a_literal = LiteralUtil::CreateR1<float>(values); local
356 auto a = Parameter(&builder, 0, a_literal.shape(), "x");
383 auto a_data = client_->TransferToServer(a_literal).ConsumeValueOrDie();
2896 Literal a_literal = LiteralUtil::CreateR4FromArray4DWithLayout( local
[all...] |
