Home
last modified time | relevance | path

Searched refs:ConstantR3FromArray3D (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dtranspose_test.cc79 ConstantR3FromArray3D<int32>(&builder, Array3D<int32>(0, 2, 3)); in XLA_TEST_F()
88 ConstantR3FromArray3D<int32>(&builder, {{{1, 2, 3}, {4, 5, 6}}}); in TEST_F()
99 ConstantR3FromArray3D<int32>(&builder, {{{1, 2, 3}, {4, 5, 6}}}); in TEST_F()
110 ConstantR3FromArray3D<int32>(&builder, {{{1, 2, 3}, {4, 5, 6}}}); in TEST_F()
170 auto operand = ConstantR3FromArray3D(&builder, aoperand); in TestTransposeConstant021()
Dconcat_test.cc242 auto a = ConstantR3FromArray3D(&builder, Array3D<float>(3, 0, 2)); in XLA_TEST_F()
243 auto b = ConstantR3FromArray3D(&builder, Array3D<float>(3, 0, 1)); in XLA_TEST_F()
263 auto a = ConstantR3FromArray3D(&builder, a_array); in XLA_TEST_F()
264 auto b = ConstantR3FromArray3D(&builder, b_array); in XLA_TEST_F()
306 auto a = ConstantR3FromArray3D(&builder, a_array); in XLA_TEST_F()
307 auto b = ConstantR3FromArray3D(&builder, b_array); in XLA_TEST_F()
308 auto c = ConstantR3FromArray3D(&builder, c_array); in XLA_TEST_F()
Dmultidimensional_slice_test.cc49 auto original = ConstantR3FromArray3D<float>(&builder, array_3d); in XLA_TEST_F()
Dlog_test.cc33 auto x = ConstantR3FromArray3D<float>(&builder, Array3D<float>(3, 0, 0)); in XLA_TEST_F()
Dslice_test.cc46 auto original = ConstantR3FromArray3D<float>(&builder, values); in TEST_F()
59 auto original = ConstantR3FromArray3D<float>(&builder, values); in TEST_F()
72 auto original = ConstantR3FromArray3D<float>(&builder, values); in TEST_F()
Dvector_ops_reduce_test.cc49 return ConstantR3FromArray3D<float>(&builder_, x3d); in BuildSampleConstantCube()
Dconvolution_variants_test.cc1303 auto gradients = ConstantR3FromArray3D<float>( in XLA_TEST_F()
1306 ConstantR3FromArray3D<float>(&builder, Array3D<float>({{{1, 10, 100}}})); in XLA_TEST_F()
1318 ConstantR3FromArray3D<float>(&builder, Array3D<float>({{{1, 2, 3, 4}}})); in XLA_TEST_F()
1320 ConstantR3FromArray3D<float>(&builder, Array3D<float>({{{100, 10, 1}}})); in XLA_TEST_F()
Darray_elementwise_ops_test.cc2067 auto array = ConstantR3FromArray3D<int32>(&builder, a_3d); in XLA_TEST_F()
2078 auto array = ConstantR3FromArray3D<int32>(&builder, a_3d); in XLA_TEST_F()
2806 auto a = ConstantR3FromArray3D<float>(&builder, a_3d);
2810 auto b = ConstantR3FromArray3D<float>(&builder, b_3d);
2833 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2857 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2887 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2912 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2915 auto b = ConstantR3FromArray3D<float>(&builder, b_3d); in XLA_TEST_F()
Dtriangular_solve_test.cc434 ConstantR3FromArray3D(&builder, avals), in XLA_TEST_F()
Ddot_operation_test.cc1752 auto rhs = ConstantR3FromArray3D(&builder, const_arr); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmath_test.cc425 auto a = ConstantR3FromArray3D<float>( in XLA_TEST_F()
430 auto x = ConstantR3FromArray3D<float>( in XLA_TEST_F()
466 auto a = ConstantR3FromArray3D<half>( in XLA_TEST_F()
485 auto a = ConstantR3FromArray3D<float>( in XLA_TEST_F()
490 auto x = ConstantR3FromArray3D<float>( in XLA_TEST_F()
513 auto a = ConstantR3FromArray3D<half>( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h1608 XlaOp ConstantR3FromArray3D(XlaBuilder* builder,
2624 XlaOp ConstantR3FromArray3D(XlaBuilder* builder, in ConstantR3FromArray3D() function