Home
last modified time | relevance | path

Searched refs:CreateR2FromArray2DWithLayout (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dbef_executable_test.cc37 LiteralUtil::CreateR2FromArray2DWithLayout<float>( in XLA_TEST_F()
41 LiteralUtil::CreateR2FromArray2DWithLayout<float>( in XLA_TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dmatrix_ops_simple_test.cc201 client_->TransferToServer(LiteralUtil::CreateR2FromArray2DWithLayout<T>( in TestImpl()
205 client_->TransferToServer(LiteralUtil::CreateR2FromArray2DWithLayout<T>( in TestImpl()
H A Ddot_operation_test.cc324 Literal dot_lhs_lit = LiteralUtil::CreateR2FromArray2DWithLayout( in TestImpl()
335 LiteralUtil::CreateR2FromArray2DWithLayout(*dot_rhs_data, rhs_layout); in TestImpl()
345 addend_lit = LiteralUtil::CreateR2FromArray2DWithLayout( in TestImpl()
813 LiteralUtil::CreateR2FromArray2DWithLayout<T>( in XLA_TYPED_TEST()
820 LiteralUtil::CreateR2FromArray2DWithLayout<T>( in XLA_TYPED_TEST()
H A Dslice_test.cc407 auto literal = LiteralUtil::CreateR2FromArray2DWithLayout( in XLA_TEST_P()
H A Dreduce_window_test.cc1296 Literal input_literal = LiteralUtil::CreateR2FromArray2DWithLayout( in DoIt()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral_util.h140 static Literal CreateR2FromArray2DWithLayout(const Array2D<NativeT>& values,
412 /* static */ Literal LiteralUtil::CreateR2FromArray2DWithLayout( in CreateR2FromArray2DWithLayout() function