Home
last modified time | relevance | path

Searched refs:CreateR3FromArray3D (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test_1d.cc223 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
226 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
258 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in RunTest()
261 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in RunTest()
293 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
296 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
324 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
327 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
360 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in RunTest()
363 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in RunTest()
Dconstants_test.cc116 ConstantLiteral(&builder, LiteralUtil::CreateR3FromArray3D<float>( in TEST_F()
132 ConstantLiteral(&builder, LiteralUtil::CreateR3FromArray3D<float>(array3d)); in TEST_F()
Ddynamic_ops_test.cc189 std::move(LiteralUtil::CreateR3FromArray3D(input_values_int) in RunR3()
193 std::move(LiteralUtil::CreateR3FromArray3D(expected_values_int) in RunR3()
477 std::move(LiteralUtil::CreateR3FromArray3D(input_values_int) in RunR3()
481 std::move(LiteralUtil::CreateR3FromArray3D(update_values_int) in RunR3()
485 std::move(LiteralUtil::CreateR3FromArray3D(expected_values_int) in RunR3()
564 Literal literal = LiteralUtil::CreateR3FromArray3D<NativeT>(values); in DumpArray()
Dliteral_test_util.h191 EXPECT_TRUE(Equal(LiteralUtil::CreateR3FromArray3D(expected), actual)); in ExpectR3EqualArray3D()
249 EXPECT_TRUE(Near(LiteralUtil::CreateR3FromArray3D(expected), actual, error)); in ExpectR3NearArray3D()
Ddot_operation_test.cc668 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
674 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
703 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
741 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
1693 AddParam(LiteralUtil::CreateR3FromArray3D<float>(input_arr), &builder); in XLA_TEST_F()
1710 AddParam(LiteralUtil::CreateR3FromArray3D<float>(input_arr), &builder); in XLA_TEST_F()
1748 AddParam(LiteralUtil::CreateR3FromArray3D<float>(input_arr), &builder); in XLA_TEST_F()
1843 auto input_literal = LiteralUtil::CreateR3FromArray3D<float>(input_arr); in DOT_ReorderContracting()
Dclient_library_test_base.h530 LiteralUtil::CreateR3FromArray3D<NativeT>(expected); in ComputeAndCompareR3()
547 LiteralUtil::CreateR3FromArray3D<NativeT>(expected); in ComputeAndCompareR3()
650 Literal literal = LiteralUtil::CreateR3FromArray3D(array_3d); in CreateR3Parameter()
Dmap_test.cc395 LiteralUtil::CreateR3FromArray3D<int32>(Array3D<int32>(3, 0, 2)); in XLA_TEST_F()
400 LiteralUtil::CreateR3FromArray3D<int32>(Array3D<int32>(3, 0, 2)); in XLA_TEST_F()
Dcopy_test.cc170 Literal literal = LiteralUtil::CreateR3FromArray3D(a); in TestCopyConstantLayout021()
Dbroadcast_simple_test.cc63 auto r3_data = LiteralUtil::CreateR3FromArray3D(*r3_array).Relayout( in MakeR3Data()
368 auto expected = LiteralUtil::CreateR3FromArray3D(expected_array); in XLA_TEST_P()
Dbroadcast_test.cc283 LiteralUtil::CreateR3FromArray3D<float>(input_vals))); in TEST_F()
Dbatch_normalization_test.cc361 {LiteralUtil::CreateR3FromArray3D<float>(Array3D<float>(260, 2, 2, 1.0f)), in XLA_TEST_P()
393 {LiteralUtil::CreateR3FromArray3D<float>( in XLA_TEST_P()
Dreduce_test.cc529 Literal input_literal = LiteralUtil::CreateR3FromArray3D(input_data); in XLA_TEST_F()
819 auto input_literal = LiteralUtil::CreateR3FromArray3D(input_array); in XLA_TEST_P()
Dconcat_test.cc859 auto x_literal = LiteralUtil::CreateR3FromArray3D<float>(x3d); in XLA_TEST_F()
Darray_elementwise_ops_test.cc2296 LiteralUtil::CreateR3FromArray3D<float>(Array3D<float>(0, 7, 0)); in XLA_TEST_F()
2301 LiteralUtil::CreateR3FromArray3D<float>(Array3D<float>(0, 7, 0)); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc167 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*result); in TEST_F()
180 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*result); in TEST_F()
221 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*actual); in TEST_F()
234 auto actual_literal = LiteralUtil::CreateR3FromArray3D(*actual); in TEST_F()
Dliteral_util.h137 static Literal CreateR3FromArray3D(const Array3D<NativeT>& values);
438 /* static */ Literal LiteralUtil::CreateR3FromArray3D( in CreateR3FromArray3D() function
465 return CreateR3FromArray3D(array); in CreateR3Projected()
Dliteral_test.cc248 auto literal = LiteralUtil::CreateR3FromArray3D(array_3d); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc963 auto expected = LiteralUtil::CreateR3FromArray3D<float>(expected_array); in TEST_P()
972 auto lhs_literal = LiteralUtil::CreateR3FromArray3D<float>(lhs_array); in TEST_P()
977 auto rhs_literal = LiteralUtil::CreateR3FromArray3D<float>(rhs_array); in TEST_P()
1012 auto expected = LiteralUtil::CreateR3FromArray3D<float>(expected_array); in TEST_P()