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_t>(&builder, Array3D<int32_t>(0, 2, 3)); in XLA_TEST_F()
88 ConstantR3FromArray3D<int32_t>(&builder, {{{1, 2, 3}, {4, 5, 6}}}); in TEST_F()
99 ConstantR3FromArray3D<int32_t>(&builder, {{{1, 2, 3}, {4, 5, 6}}}); in TEST_F()
110 ConstantR3FromArray3D<int32_t>(&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.cc45 auto original = ConstantR3FromArray3D<float>(&builder, values); in TEST_F()
58 auto original = ConstantR3FromArray3D<float>(&builder, values); in TEST_F()
71 auto original = ConstantR3FromArray3D<float>(&builder, values); in TEST_F()
Dvector_ops_reduce_test.cc49 return ConstantR3FromArray3D<float>(&builder_, x3d); in BuildSampleConstantCube()
Dconvolution_variants_test.cc1302 auto gradients = ConstantR3FromArray3D<float>( in XLA_TEST_F()
1305 ConstantR3FromArray3D<float>(&builder, Array3D<float>({{{1, 10, 100}}})); in XLA_TEST_F()
1317 ConstantR3FromArray3D<float>(&builder, Array3D<float>({{{1, 2, 3, 4}}})); in XLA_TEST_F()
1319 ConstantR3FromArray3D<float>(&builder, Array3D<float>({{{100, 10, 1}}})); in XLA_TEST_F()
Darray_elementwise_ops_test.cc2096 auto array = ConstantR3FromArray3D<int32_t>(&builder, a_3d); in XLA_TEST_F()
2107 auto array = ConstantR3FromArray3D<int32_t>(&builder, a_3d); in XLA_TEST_F()
2885 auto a = ConstantR3FromArray3D<float>(&builder, a_3d);
2889 auto b = ConstantR3FromArray3D<float>(&builder, b_3d);
2912 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2936 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2966 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2991 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2994 auto b = ConstantR3FromArray3D<float>(&builder, b_3d); in XLA_TEST_F()
Dtriangular_solve_test.cc427 ConstantR3FromArray3D(&builder, avals), in XLA_TEST_F()
Ddot_operation_test.cc1753 auto rhs = ConstantR3FromArray3D(&builder, const_arr); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmath_test.cc434 auto a = ConstantR3FromArray3D<float>( in XLA_TEST_F()
439 auto x = ConstantR3FromArray3D<float>( in XLA_TEST_F()
475 auto a = ConstantR3FromArray3D<half>( in XLA_TEST_F()
494 auto a = ConstantR3FromArray3D<float>( in XLA_TEST_F()
499 auto x = ConstantR3FromArray3D<float>( in XLA_TEST_F()
522 auto a = ConstantR3FromArray3D<half>( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h1706 XlaOp ConstantR3FromArray3D(XlaBuilder* builder,
2782 XlaOp ConstantR3FromArray3D(XlaBuilder* builder, in ConstantR3FromArray3D() function