Searched refs:a_3d (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | array_elementwise_ops_test.cc | 1960 Array3D<int32> a_3d({{{3, 9, -1}, {2, -10, 3}}, {{-2, 2, 8}, {12, 10, 4}}}); in XLA_TEST_F() local
1961 auto array = ConstantR3FromArray3D<int32>(&builder, a_3d); in XLA_TEST_F()
1971 Array3D<int32> a_3d(2, 0, 3); in XLA_TEST_F() local
1972 auto array = ConstantR3FromArray3D<int32>(&builder, a_3d); in XLA_TEST_F()
2698 Array3D<float> a_3d({{{1.0f, 2.0f}, {3.0f, 4.0f}, {5.0f, 6.0f}}, variable
2700 auto a = ConstantR3FromArray3D<float>(&builder, a_3d);
2718 Array3D<float> a_3d({ in XLA_TEST_F() local
2727 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
2742 Array3D<float> a_3d({ in XLA_TEST_F() local
2751 auto a = ConstantR3FromArray3D<float>(&builder, a_3d); in XLA_TEST_F()
[all …]
|