Searched refs:ConstantR3FromArray3D (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | transpose_test.cc | 79 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()
|
| D | concat_test.cc | 239 auto a = ConstantR3FromArray3D(&builder, Array3D<float>(3, 0, 2)); in XLA_TEST_F()
240 auto b = ConstantR3FromArray3D(&builder, Array3D<float>(3, 0, 1)); in XLA_TEST_F()
260 auto a = ConstantR3FromArray3D(&builder, a_array); in XLA_TEST_F()
261 auto b = ConstantR3FromArray3D(&builder, b_array); in XLA_TEST_F()
303 auto a = ConstantR3FromArray3D(&builder, a_array); in XLA_TEST_F()
304 auto b = ConstantR3FromArray3D(&builder, b_array); in XLA_TEST_F()
305 auto c = ConstantR3FromArray3D(&builder, c_array); in XLA_TEST_F()
|
| D | multidimensional_slice_test.cc | 49 auto original = ConstantR3FromArray3D<float>(&builder, array_3d); in XLA_TEST_F()
|
| D | log_test.cc | 33 auto x = ConstantR3FromArray3D<float>(&builder, Array3D<float>(3, 0, 0)); in XLA_TEST_F()
|
| D | slice_test.cc | 46 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()
|
| D | vector_ops_reduce_test.cc | 49 return ConstantR3FromArray3D<float>(&builder_, x3d); in BuildSampleConstantCube()
|
| D | array_elementwise_ops_test.cc | 1961 auto array = ConstantR3FromArray3D<int32>(&builder, a_3d); in XLA_TEST_F()
1972 auto array = ConstantR3FromArray3D<int32>(&builder, a_3d); in XLA_TEST_F()
2700 auto a = ConstantR3FromArray3D<float>(&builder, a_3d);
2704 auto b = ConstantR3FromArray3D<float>(&builder, b_3d);
2727 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2751 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2781 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2806 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2809 auto b = ConstantR3FromArray3D<float>(&builder, b_3d); in XLA_TEST_F()
|
| D | convolution_variants_test.cc | 1303 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()
|
| D | triangular_solve_test.cc | 424 ConstantR3FromArray3D(&builder, avals), in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 1094 XlaOp ConstantR3FromArray3D(XlaBuilder* builder,
1992 XlaOp ConstantR3FromArray3D(XlaBuilder* builder, in ConstantR3FromArray3D() function
|