Searched refs:ConstantR3FromArray3D (Results 1 – 12 of 12) 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 | 242 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()
|
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 | 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 | array_elementwise_ops_test.cc | 2067 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()
|
D | triangular_solve_test.cc | 434 ConstantR3FromArray3D(&builder, avals), in XLA_TEST_F()
|
D | dot_operation_test.cc | 1752 auto rhs = ConstantR3FromArray3D(&builder, const_arr); in XLA_TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | math_test.cc | 425 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/ |
D | xla_builder.h | 1608 XlaOp ConstantR3FromArray3D(XlaBuilder* builder, 2624 XlaOp ConstantR3FromArray3D(XlaBuilder* builder, in ConstantR3FromArray3D() function
|