Home
last modified time | relevance | path

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

/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.h174 static Literal CreateR3FromArray3D(const Array3D<NativeT>& values);
471 /* static */ Literal LiteralUtil::CreateR3FromArray3D( in CreateR3FromArray3D() function
498 return CreateR3FromArray3D(array); in CreateR3Projected()
Dliteral_test.cc238 auto literal = LiteralUtil::CreateR3FromArray3D(array_3d); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamic_ops_test.cc188 std::move(LiteralUtil::CreateR3FromArray3D(input_values_int) in RunR3()
192 std::move(LiteralUtil::CreateR3FromArray3D(expected_values_int) in RunR3()
476 std::move(LiteralUtil::CreateR3FromArray3D(input_values_int) in RunR3()
480 std::move(LiteralUtil::CreateR3FromArray3D(update_values_int) in RunR3()
484 std::move(LiteralUtil::CreateR3FromArray3D(expected_values_int) in RunR3()
563 Literal literal = LiteralUtil::CreateR3FromArray3D<NativeT>(values); in DumpArray()
Dconstants_test.cc116 ConstantLiteral(&builder, LiteralUtil::CreateR3FromArray3D<float>( in TEST_F()
132 ConstantLiteral(&builder, LiteralUtil::CreateR3FromArray3D<float>(array3d)); in TEST_F()
Dliteral_test_util.h191 EXPECT_TRUE(Equal(LiteralUtil::CreateR3FromArray3D(expected), actual)); in ExpectR3EqualArray3D()
249 EXPECT_TRUE(Near(LiteralUtil::CreateR3FromArray3D(expected), actual, error)); in ExpectR3NearArray3D()
Dconvolution_test.cc259 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
262 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
294 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in RunTest()
297 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in RunTest()
329 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
332 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
360 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
363 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
396 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in RunTest()
399 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in RunTest()
Dclient_library_test_base.h499 LiteralUtil::CreateR3FromArray3D<NativeT>(expected); in ComputeAndCompareR3()
516 LiteralUtil::CreateR3FromArray3D<NativeT>(expected); in ComputeAndCompareR3()
594 Literal literal = LiteralUtil::CreateR3FromArray3D(array_3d); in CreateR3Parameter()
Ddot_operation_test.cc624 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
630 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
661 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
700 ->TransferToServer(LiteralUtil::CreateR3FromArray3D<T>( in XLA_TYPED_TEST()
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()
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()
Dcopy_test.cc170 Literal literal = LiteralUtil::CreateR3FromArray3D(a); in TestCopyConstantLayout021()
Dbroadcast_test.cc283 LiteralUtil::CreateR3FromArray3D<float>(input_vals))); in TEST_F()
Dbatch_normalization_test.cc300 {LiteralUtil::CreateR3FromArray3D<float>(Array3D<float>(260, 2, 2, 1.0f)), in XLA_TEST_P()
332 {LiteralUtil::CreateR3FromArray3D<float>( in XLA_TEST_P()
Dreduce_test.cc528 Literal input_literal = LiteralUtil::CreateR3FromArray3D(input_data); in XLA_TEST_F()
818 auto input_literal = LiteralUtil::CreateR3FromArray3D(input_array); in XLA_TEST_P()
Dconcat_test.cc613 auto x_literal = LiteralUtil::CreateR3FromArray3D<float>(x3d); in XLA_TEST_F()
Darray_elementwise_ops_test.cc2190 LiteralUtil::CreateR3FromArray3D<float>(Array3D<float>(0, 7, 0)); in XLA_TEST_F()
2195 LiteralUtil::CreateR3FromArray3D<float>(Array3D<float>(0, 7, 0)); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc886 auto expected = LiteralUtil::CreateR3FromArray3D<float>(expected_array); in TEST_P()
895 auto lhs_literal = LiteralUtil::CreateR3FromArray3D<float>(lhs_array); in TEST_P()
900 auto rhs_literal = LiteralUtil::CreateR3FromArray3D<float>(rhs_array); in TEST_P()
935 auto expected = LiteralUtil::CreateR3FromArray3D<float>(expected_array); in TEST_P()