Home
last modified time | relevance | path

Searched refs:Array3D (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test_1d.cc213 Array3D<float> input({{{1, 2, 3, 4, 5}, {6, 7, 8, 9, 10}}}); in XLA_TEST_F()
214 Array3D<float> filter({{{10, 20}, {30, 40}}}); in XLA_TEST_F()
216 Array3D<float> expected({{{510, 610, 710, 810}}}); in XLA_TEST_F()
247 Array3D<T> input( in RunTest()
249 Array3D<T> filter({{{10.0f, 20.0f}, {30.0f, 40.0f}}}); in RunTest()
251 Array3D<T> expected({{{570.0f, 670.0f, 770.0f}}}); in RunTest()
283 Array3D<float> input({{{1, 2, 3, 4, 5}, {6, 7, 8, 9, 10}}}); in XLA_TEST_F()
284 Array3D<float> filter({{{10, 20}, {30, 40}}}); in XLA_TEST_F()
286 Array3D<float> expected({{{190, 320, 230, 380, 270, 440, 310, 500}}}); in XLA_TEST_F()
314 Array3D<float> input({{{1, 2, 3, 4, 5}, {6, 7, 8, 9, 10}}}); in XLA_TEST_F()
[all …]
Dtranspose_test.cc79 ConstantR3FromArray3D<int32_t>(&builder, Array3D<int32_t>(0, 2, 3)); in XLA_TEST_F()
82 ComputeAndCompareR3<int32_t>(&builder, Array3D<int32_t>(2, 3, 0), {}); in XLA_TEST_F()
91 Array3D<int32_t> expected({{{1}, {2}, {3}}, {{4}, {5}, {6}}}); in TEST_F()
102 Array3D<int32_t> expected({{{1}, {4}}, {{2}, {5}}, {{3}, {6}}}); in TEST_F()
113 Array3D<int32_t> expected({{{1, 2, 3}, {4, 5, 6}}}); in TEST_F()
158 Array3D<int32_t> aoperand(n1, n2, n3); in TestTransposeConstant021()
159 Array3D<int32_t> expected(n1, n3, n2); 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()
245 ComputeAndCompareR3<float>(&builder, Array3D<float>(3, 0, 3), {}, in XLA_TEST_F()
251 Array3D<float> a_array({ in XLA_TEST_F()
257 Array3D<float> b_array({ in XLA_TEST_F()
267 Array3D<float> expected({ in XLA_TEST_F()
288 Array3D<float> a_array({ in XLA_TEST_F()
294 Array3D<float> b_array({ in XLA_TEST_F()
300 Array3D<float> c_array({ in XLA_TEST_F()
311 Array3D<float> expected({ in XLA_TEST_F()
[all …]
Dmultidimensional_slice_test.cc47 Array3D<float> array_3d( in XLA_TEST_F()
52 Array3D<float> expected_3d({{{2.0f}}, {{6.0f}}}); in XLA_TEST_F()
Dbroadcast_simple_test.cc59 Array3D<float>* r3_array, float start, float end, int seed) { in MakeR3Data()
191 Array3D<float> expected(2, 2, 2);
209 Array3D<float> expected(2, 2, 2);
244 Array3D<bool> y_vals(2, 2, 1); in XLA_TEST_F()
255 Array3D<bool> expected(2, 2, 1); in XLA_TEST_F()
329 Array3D<float> r3_array(spec.output_bounds[0], spec.output_bounds[1], in XLA_TEST_P()
331 Array3D<float> r3_implicit_array(spec.input_bounds[0], spec.input_bounds[1], in XLA_TEST_P()
346 Array3D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1], in XLA_TEST_P()
382 Array3D<float> r1d = {{{1}}, {{2}}}; in XLA_TEST_F()
383 Array3D<float> r3d = {{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}}; in XLA_TEST_F()
Dlog_test.cc33 auto x = ConstantR3FromArray3D<float>(&builder, Array3D<float>(3, 0, 0)); in XLA_TEST_F()
36 ComputeAndCompareR3<float>(&builder, Array3D<float>(3, 0, 0), {}, in XLA_TEST_F()
Dcholesky_test.cc67 Array3D<float> a_vals({ in XLA_TEST_F()
85 Array3D<float> expected({ in XLA_TEST_F()
180 Array3D<float> a_vals({ in XLA_TEST_F()
199 Array3D<float> expected({ in XLA_TEST_F()
Dslice_test.cc41 Array3D<float> values(3, 3, 3); in TEST_F()
48 Array3D<float> expected{ in TEST_F()
54 Array3D<float> values(3, 3, 3); in TEST_F()
61 Array3D<float> expected{ in TEST_F()
67 Array3D<float> values(3, 3, 3); in TEST_F()
74 Array3D<float> expected{ in TEST_F()
Ddynamic_ops_test.cc183 void RunR3(const Array3D<int>& input_values_int, in RunR3()
186 const Array3D<int>& expected_values_int) { in RunR3()
471 void RunR3(const Array3D<int>& input_values_int, in RunR3()
472 const Array3D<int>& update_values_int, in RunR3()
474 const Array3D<int>& expected_values_int) { in RunR3()
516 Array3D<T> input_values(kSeq, kBatch, kDim); in RunR3Contiguous()
517 Array3D<T> update_values(size, kBatch, kDim); in RunR3Contiguous()
518 Array3D<T> expected_values(kSeq, kBatch, kDim); in RunR3Contiguous()
562 void DumpArray(const std::string& name, const Array3D<NativeT> values) { in DumpArray()
Dconstants_test.cc117 Array3D<float>(3, 0, 2))); in TEST_F()
119 ComputeAndCompareR3<float>(&builder, Array3D<float>(3, 0, 2), {}); in TEST_F()
124 Array3D<float> array3d({ in TEST_F()
Dliteral_test_util.h82 static void ExpectR3EqualArray3D(const Array3D<NativeT>& expected,
133 static void ExpectR3NearArray3D(const Array3D<NativeT>& expected,
189 const Array3D<NativeT>& expected, const LiteralSlice& actual) { in ExpectR3EqualArray3D()
246 const Array3D<NativeT>& expected, const LiteralSlice& actual, in ExpectR3NearArray3D()
Dclient_library_test_base.h162 const Array3D<NativeT>& expected,
166 const Array3D<NativeT>& expected,
377 const Array3D<NativeT>& array_3d, int64_t parameter_number,
527 XlaBuilder* builder, const Array3D<NativeT>& expected, in ComputeAndCompareR3()
537 XlaBuilder* builder, const Array3D<NativeT>& expected, in ComputeAndCompareR3()
645 const Array3D<NativeT>& array_3d, int64_t parameter_number, in CreateR3Parameter()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util_test.cc43 Array3D<int64_t>({{{0, 1, 2, 3}, {4, 5, 6, 7}}}), {OpSharding::MANUAL}); in TEST()
64 HloSharding sharding = HloSharding::Tile(Array3D<int64_t>({{{0}, {1}}})); in TEST()
74 HloSharding::Tile(Array3D<int64_t>({{{0}}, {{1}}})); in TEST()
87 HloSharding::Tile(Array3D<int64_t>({{{0}}, {{1}}})); in TEST()
112 HloSharding::Tile(Array3D<int64_t>({{{0}}, {{1}}})); in TEST()
114 HloSharding::Tile(Array3D<int64_t>({{{0}}, {{1}}})); in TEST()
152 HloSharding::Tile(Array3D<int64_t>({{{0}}, {{1}}})); in TEST()
161 HloSharding sharding = HloSharding::Tile(Array3D<int64_t>({{{0}, {1}}})); in TEST()
170 HloSharding sharding = HloSharding::Tile(Array3D<int64_t>({{{0}, {1}}})); in TEST()
192 HloSharding::Tile(Array3D<int64_t>({{{0, 1}, {2, 3}}, {{4, 5}, {6, 7}}})); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Darray3d.h35 class Array3D : public Array<T> {
37 Array3D() : Array<T>(std::vector<int64_t>{0, 0, 0}) {} in Array3D() function
40 Array3D(const int64_t n1, const int64_t n2, const int64_t n3) in Array3D() function
44 Array3D(const int64_t n1, const int64_t n2, const int64_t n3, const T value) in Array3D() function
54 Array3D(std::initializer_list<std::initializer_list<std::initializer_list<T>>> in Array3D() function
66 Array3D( in Array3D() function
Dreference_util_test.cc105 Array3D<float> input(2, 3, 4); in TEST_F()
171 Array3D<float> input(2, 3, 4); in TEST_F()
184 Array3D<float> input(2, 3, 4); in TEST_F()
224 Array3D<float> input = {{{1, 2, 3, 4}}}; in TEST_F()
225 Array3D<float> weights = {{{5, 6}}}; in TEST_F()
226 std::unique_ptr<Array3D<float>> actual = in TEST_F()
228 Array3D<float> expected = {{{17, 28, 39, 20}}}; in TEST_F()
237 Array3D<float> input = {{{1, 2, 3, 4}}}; in TEST_F()
238 Array3D<float> weights = {{{5, 6}}}; in TEST_F()
239 std::unique_ptr<Array3D<float>> actual = in TEST_F()
[all …]
Dreference_util.h91 static std::unique_ptr<Array3D<float>> ConvArray3D(const Array3D<float>& lhs,
92 const Array3D<float>& rhs,
97 static std::unique_ptr<Array3D<float>> ConvArray3DGeneralDimensionsDilated(
98 const Array3D<float>& lhs, const Array3D<float>& rhs,
156 const Array3D<float>& array, float init, absl::Span<const int64_t> dims,
173 static std::unique_ptr<Array3D<float>> MapArray3D(
174 const Array3D<float>& array,
179 static std::unique_ptr<Array3D<float>> MapArray3D(
180 const Array3D<float>& lhs, const Array3D<float>& rhs,
193 static std::unique_ptr<Array3D<float>> ReduceWindow3DAdd(
[all …]
Darray3d_test.cc27 Array3D<int> uninit(2, 3, 4); in TEST()
35 Array3D<int> fullof7(2, 3, 4, 7); in TEST()
51 Array3D<int> arr = {{{1, 2}, {3, 4}, {5, 6}, {7, 8}}, in TEST()
73 Array3D<Eigen::half> arr = { in TEST()
96 Array3D<int> fullof7(2, 3, 4, 7); in TEST()
Dreference_util.cc47 /* static */ std::unique_ptr<Array3D<float>> ReferenceUtil::ConvArray3D( in ConvArray3D()
48 const Array3D<float>& lhs, const Array3D<float>& rhs, int64_t kernel_stride, in ConvArray3D()
55 /*static*/ std::unique_ptr<Array3D<float>>
57 const Array3D<float>& lhs, const Array3D<float>& rhs, int64_t kernel_stride, in ConvArray3DGeneralDimensionsDilated()
85 auto convr3 = std::make_unique<Array3D<float>>( in ConvArray3DGeneralDimensionsDilated()
188 /* static */ std::unique_ptr<Array3D<float>> ReferenceUtil::ReduceWindow3DAdd( in ReduceWindow3DAdd()
189 const Array3D<float>& operand, float init, absl::Span<const int64_t> window, in ReduceWindow3DAdd()
201 auto result = std::make_unique<Array3D<float>>( in ReduceWindow3DAdd()
617 const Array3D<float>& array, float init, absl::Span<const int64_t> dims, in Reduce3DTo2D()
667 /* static */ std::unique_ptr<Array3D<float>> ReferenceUtil::MapArray3D( in MapArray3D()
[all …]
Dliteral_util.h143 static Literal CreateR3FromArray3D(const Array3D<NativeT>& values);
145 static Literal CreateR3FromArray3DWithLayout(const Array3D<NativeT>& values,
331 Array3D<NativeT> tmp(d0, d1, d2); in CreateR3WithLayout()
425 const Array3D<NativeT>& values, const Layout& layout) { in CreateR3FromArray3DWithLayout()
431 const Array3D<NativeT>& values) { in CreateR3FromArray3D()
443 Array3D<NativeT> array(dim0_size, dim1_size, dim2_size); in CreateR3Projected()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dtridiagonal_test.cc42 Array3D<float> upper_diagonal{{{34, 35, 999}}}; in XLA_TEST_P()
43 Array3D<float> main_diagonal{{{21, 22, 23}}}; in XLA_TEST_P()
45 Array3D<float> lower_diagonal{{{999, 10, 100}}}; in XLA_TEST_P()
46 Array3D<float> rhs{{{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}}}; in XLA_TEST_P()
117 Array3D<float> lower_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P()
118 Array3D<float> main_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P()
119 Array3D<float> upper_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P()
120 Array3D<float> rhs(batch_size, num_rhs, num_eqs); in XLA_TEST_P()
Dself_adjoint_eig_test.cc41 batch_3d_4x4_ = Array3D<float>{ in SetUp()
76 Array3D<float> GetUnitMatrix3D(const Array3D<float>& matrix) { in GetUnitMatrix3D()
77 Array3D<float> result(matrix.n1(), matrix.n2(), matrix.n3(), 0.0); in GetUnitMatrix3D()
86 Array3D<float> ExtractTriangularMatrix(const Array3D<float>& matrix, in ExtractTriangularMatrix()
88 Array3D<float> result(matrix); in ExtractTriangularMatrix()
105 Array3D<float> batch_3d_4x4_;
Dmatrix_test.cc64 Array3D<int32_t> input(2, 3, 4); in XLA_TEST_F()
70 Array3D<int32_t> expected({{{0, 0, 0, 0}, {4, 5, 0, 0}, {8, 9, 10, 0}}, in XLA_TEST_F()
175 Array3D<T> input(2, 3, 4); in TestMatrixDiagonal()
189 Array3D<T> input(2, 3, 4); in TestSetMatrixDiagonal()
227 std::map<int, Array3D<T>> k_and_expected = { in TestMatrixDiagonal4D()
259 Array3D<float> BatchedAValsFull() { in BatchedAValsFull()
Dsvd_test.cc44 batch_3d_4x5_ = Array3D<float>{ in SetUp()
64 Array3D<float> GetUnitMatrix3D(int32_t batch_dim, int32_t mat_dim) { in GetUnitMatrix3D()
65 Array3D<float> result(batch_dim, mat_dim, mat_dim, 0.0); in GetUnitMatrix3D()
116 Array3D<float> batch_3d_4x5_;
/external/snakeyaml/src/test/java/org/yaml/snakeyaml/array/
DPrimitiveArrayTest.java189 Array3D three = new Array3D(); in testMultiDimensional()
192 assertTrue(Arrays.deepEquals(three.getLongs(), ((Array3D) cycle(three)).getLongs())); in testMultiDimensional()
215 public static class Array3D { class in PrimitiveArrayTest
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dinfeed_test.cc122 Array3D<float> array(1024, 1024, 64); in TEST_F()
166 Array3D<float> array(1024, 1024, 64); in TEST_F()

12