| /external/tensorflow/tensorflow/compiler/xla/ |
| reference_util_test.cc | 167 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*result);
180 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*result);
221 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*actual);
234 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*actual);
|
| literal_util.h | 174 static Literal CreateR3FromArray3D(const Array3D<NativeT>& values);
471 /* static */ Literal LiteralUtil::CreateR3FromArray3D(
498 return CreateR3FromArray3D(array);
|
| literal_test.cc | 226 TEST_F(LiteralUtilTest, CreateR3FromArray3d) {
238 auto literal = LiteralUtil::CreateR3FromArray3D(array_3d);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| constants_test.cc | 116 ConstantLiteral(&builder, LiteralUtil::CreateR3FromArray3D<float>(
132 ConstantLiteral(&builder, LiteralUtil::CreateR3FromArray3D<float>(array3d));
|
| literal_test_util.h | 191 EXPECT_TRUE(Equal(LiteralUtil::CreateR3FromArray3D(expected), actual));
249 EXPECT_TRUE(Near(LiteralUtil::CreateR3FromArray3D(expected), actual, error));
|
| dynamic_ops_test.cc | 188 std::move(LiteralUtil::CreateR3FromArray3D(input_values_int)
192 std::move(LiteralUtil::CreateR3FromArray3D(expected_values_int)
476 std::move(LiteralUtil::CreateR3FromArray3D(input_values_int)
480 std::move(LiteralUtil::CreateR3FromArray3D(update_values_int)
484 std::move(LiteralUtil::CreateR3FromArray3D(expected_values_int)
563 Literal literal = LiteralUtil::CreateR3FromArray3D<NativeT>(values);
[all...] |
| convolution_test.cc | 259 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input))
262 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter))
294 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input))
297 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter))
329 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input))
332 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter))
360 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input))
363 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter))
396 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input))
399 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)
[all...] |
| broadcast_test.cc | 283 LiteralUtil::CreateR3FromArray3D<float>(input_vals)));
|
| copy_test.cc | 170 Literal literal = LiteralUtil::CreateR3FromArray3D(a);
|
| client_library_test_base.h | 499 LiteralUtil::CreateR3FromArray3D<NativeT>(expected);
516 LiteralUtil::CreateR3FromArray3D<NativeT>(expected);
594 Literal literal = LiteralUtil::CreateR3FromArray3D(array_3d);
|
| dot_operation_test.cc | 624 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>(
630 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>(
661 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>(
700 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>(
[all...] |
| batch_normalization_test.cc | 300 {LiteralUtil::CreateR3FromArray3D<float>(Array3D<float>(260, 2, 2, 1.0f)),
332 {LiteralUtil::CreateR3FromArray3D<float>(
[all...] |
| broadcast_simple_test.cc | 63 auto r3_data = LiteralUtil::CreateR3FromArray3D(*r3_array).Relayout(
368 auto expected = LiteralUtil::CreateR3FromArray3D(expected_array);
|
| map_test.cc | 395 LiteralUtil::CreateR3FromArray3D<int32>(Array3D<int32>(3, 0, 2));
400 LiteralUtil::CreateR3FromArray3D<int32>(Array3D<int32>(3, 0, 2));
|
| reduce_test.cc | 528 Literal input_literal = LiteralUtil::CreateR3FromArray3D(input_data);
818 auto input_literal = LiteralUtil::CreateR3FromArray3D(input_array);
[all...] |
| concat_test.cc | 613 auto x_literal = LiteralUtil::CreateR3FromArray3D<float>(x3d);
|
| array_elementwise_ops_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_evaluator_test.cc | [all...] |
