Home
last modified time | relevance | path

Searched refs:Array4D (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_variants_test.cc57 const Array4D<float> input_array(1, 1, 1, 1, {2}); in XLA_TEST_F()
60 const Array4D<float> filter_array(1, 1, 1, 1, {3}); in XLA_TEST_F()
65 const Array4D<float> expected(1, 1, 1, 1, {6}); in XLA_TEST_F()
72 const Array4D<float> input_array(5, 1, 1, 1, {1, 2, 3, 4, 5}); in XLA_TEST_F()
75 const Array4D<float> filter_array(1, 1, 1, 1, {2}); in XLA_TEST_F()
80 const Array4D<float> expected(5, 1, 1, 1, {2, 4, 6, 8, 10}); in XLA_TEST_F()
87 Array4D<float> input_array(2, 1, 3, 4); in XLA_TEST_F()
91 const Array4D<float> filter_array(1, 1, 1, 1, {2.3}); in XLA_TEST_F()
96 Array4D<float> expected(2, 1, 3, 4); in XLA_TEST_F()
104 Array4D<float> input_array(1, 2, 1, 1, {10, 1}); in XLA_TEST_F()
[all …]
Dpad_test.cc134 Pad(AddParam(Array4D<float>(2, 0, 3, 2), &b), in XLA_TEST_P()
137 ComputeAndCompareR4<float>(&b, Array4D<float>(5, 2, 3, 2, 1.5f), {}, in XLA_TEST_P()
143 auto input = absl::make_unique<Array4D<float>>(1, 1, 3, 2); in TEST_P()
154 auto expected = absl::make_unique<Array4D<float>>(2, 3, 3, 2); in TEST_P()
169 Array4D<float> input(3, 2, 1, 1, {1, 2, 3, 4, 5, 6}); in TEST_P()
174 auto expected = absl::make_unique<Array4D<float>>(8, 5, 1, 1); in TEST_P()
209 Array4D<float> input_array(1, 1, 2, 3, {1, 2, 3, 4, 5, 6}); in TEST_P()
216 Array4D<float> expected_array(1, 1, 5, 8); in TEST_P()
251 Array4D<float> input_array(1, 25, 7, 7); in XLA_TEST_P()
262 Array4D<float> expected_array(1, 25, 17, 11); in XLA_TEST_P()
[all …]
Dselect_and_scatter_test.cc350 Array4D<float> o(4, 6, 15, 220); in TEST_F()
353 Array4D<float> e(4, 6, 15, 220); in TEST_F()
355 Array4D<float> s(2, 2, 15, 220); in TEST_F()
375 Array4D<float> o(4, 5, 17, 128); in TEST_F()
378 Array4D<float> e(4, 5, 17, 128); in TEST_F()
380 Array4D<float> s(2, 2, 17, 128); in TEST_F()
400 Array4D<float> o(4, 5, 1, 1); in TEST_F()
403 Array4D<float> e(4, 5, 1, 1); in TEST_F()
405 Array4D<float> s(2, 2, 1, 1); in TEST_F()
422 Array4D<float> o(4, 6, 4, 4); in TEST_F()
[all …]
Dreverse_test.cc119 Array4D<uint8> input({{ in XLA_TEST_F()
132 Array4D<uint8> expected({{ in XLA_TEST_F()
149 Array4D<float> input({ in TEST_F()
168 Array4D<float> expected({ in TEST_F()
Dreshape_test.cc312 LiteralUtil::CreateFromArray(Array4D<float>(2, 3, 0, 0)); in XLA_TEST_P()
319 auto input_literal = LiteralUtil::CreateFromArray(Array4D<float>(2, 3, 4, 0)); in XLA_TEST_P()
492 Array4D<float> t2x2x2x3(2, 2, 2, 3); in XLA_TEST_P()
511 Array4D<float> t(2, 1, 2, 2); in XLA_TEST_P()
582 Array4D<float>{ in XLA_TEST_P()
691 Array4D<float> input(2, 1, 1, 1); in XLA_TEST_P()
710 Array4D<float> input(2, 1, 4, 1); in XLA_TEST_P()
730 Array4D<float> input(5, 10, 2, 3); in XLA_TEST_P()
755 Array4D<float> input_array(2, 3, 5, 7); in XLA_TEST_P()
836 Array4D<float> input(bounds[0], bounds[1], bounds[2], bounds[3]); in XLA_TEST_P()
[all …]
Dbroadcast_test.cc165 Array4D<float> expected(2, 2, 3, 3); in TEST_F()
190 Array4D<float> expected(3, 3, 3, 1025); in TEST_F()
205 Array4D<float> r4_array(32, 64, 7, 7); in XLA_TEST_F()
238 Array4D<float> expected(64, 64, 3, 3); in TEST_F()
260 Array4D<float> expected(3, 3, 2, 2); in TEST_F()
272 Array4D<float> expected(2, 3, 4, 5); in TEST_F()
Dreduce_window_test.cc156 Array4D<float> input_array(1, 0, 2, 1); in XLA_TEST_P()
169 Array4D<float> input_array(1, 2, 2, 1); in TEST_P()
184 Array4D<float> input_array(1, 3, 3, 1); in TEST_P()
198 Array4D<float> input_array(3, 6, 7, 32); in TEST_P()
215 Array4D<float> input_array(4, 4, 6, 8); in TEST_P()
237 Array4D<float> input_array(9, 12, 4, 89); in TEST_P()
262 Array4D<float> input_array(15, 15, 4, 128); in TEST_P()
285 Array4D<float> input_array(19, 17, 8, 256); in TEST_P()
303 Array4D<float> input_array(1, 2, 2, 1); in XLA_TEST_P()
340 Array4D<float> input_array(13, 12, 8, 15); in TEST_P()
[all …]
Dbatch_normalization_test.cc87 Array4D<float> input_array_;
109 Array4D<float> expected(kSamples, kZ, kY, kX); in XLA_TEST_P()
126 Array4D<float> expected(kSamples, kZ, kY, kX); in XLA_TEST_P()
220 Array4D<float> expected(kSamples, kZ, kY, kX); in XLA_TEST_P()
347 ConstantR4FromArray4D<float>(&builder, Array4D<float>(2, 2, 2, 1, 0.0f)); in XLA_TEST_P()
453 Array4D<float> input_array(bounds[0], bounds[1], bounds[2], bounds[3]); in XLA_TEST_P()
502 Array4D<float> mean4D = in XLA_TEST_P()
554 Array4D<float> input_array(bounds[0], bounds[1], bounds[2], bounds[3]); in XLA_TEST_P()
603 Array4D<float> mean4D = in XLA_TEST_P()
629 Array4D<float> expected = normalized; in XLA_TEST_P()
[all …]
Dliteral_test_util.h87 static void ExpectR4EqualArray4D(const Array4D<NativeT>& expected,
140 static void ExpectR4NearArray4D(const Array4D<NativeT>& expected,
196 const Array4D<NativeT>& expected, const LiteralSlice& actual) { in ExpectR4EqualArray4D()
254 const Array4D<NativeT>& expected, const LiteralSlice& actual, in ExpectR4NearArray4D()
Dbinop_scaling_test.cc115 Array4D<int> lhs_array({ in TEST_F()
123 Array4D<int> expected({ in TEST_F()
Dconvolution_dimension_numbers_test.cc91 auto input_array = absl::make_unique<Array4D<float>>(2, 3, 5, 5); in XLA_TEST_F()
93 auto weight_array = absl::make_unique<Array4D<float>>(4, 3, 1, 1); in XLA_TEST_F()
Dconvolution_test.cc73 auto alhs = absl::make_unique<Array4D<T>>( in RunTest()
80 auto arhs = absl::make_unique<Array4D<T>>(kOutputActivationSizeZ, in RunTest()
123 Array4D<T> input_data(1, 1, 1, 2); in RunTest()
127 Array4D<T> filter_data(1, 1, 1, 2); in RunTest()
154 Array4D<T> input_data(1, 1, 4, 4); in RunTest()
161 Array4D<T> filter_data(1, 1, 2, 2); in RunTest()
188 Array4D<T> input_data(1, 1, 4, 4); in RunTest()
195 Array4D<T> filter_data(1, 1, 2, 2); in RunTest()
224 Array4D<T> input_data(1, 1, 4, 4); in RunTest()
229 Array4D<T> filter_data(1, 1, 3, 3); in RunTest()
[all …]
Dslice_test.cc159 Array4D<float> values(2, 2, 24, 256); in TEST_F()
180 Array4D<int> values(2, 3, 5, 7); in TEST_F()
190 Array4D<float> values(2, 4, 6, 8); in XLA_TEST_F()
482 Array4D<float> values(spec.input_dims[0], spec.input_dims[1], in Run()
Darray_elementwise_ops_test.cc1762 Array4D<float> values(2, 2, 2, 2); in XLA_TEST_F()
1772 Array4D<float> expected(2, 2, 2, 2, expected_vector); in XLA_TEST_F()
1782 Array4D<float> values(2, 2, 0, 2); in XLA_TEST_F()
1783 Array4D<float> expected(2, 2, 0, 2); in XLA_TEST_F()
2009 Array4D<float> expected( in XLA_TEST_F()
2019 Array4D<float> arg(2, 2, 0, 3); in XLA_TEST_F()
2023 Array4D<float> expected(2, 2, 0, 3); in XLA_TEST_F()
2833 std::unique_ptr<Array4D<float>> operand_a_4d(new Array4D<float>(2, 3, 4, 5));
2834 std::unique_ptr<Array4D<float>> operand_b_4d(new Array4D<float>(2, 3, 4, 5));
2835 std::unique_ptr<Array4D<float>> expected_4d(new Array4D<float>(2, 3, 4, 5));
[all …]
Dclient_library_test_base.h172 const Array4D<NativeT>& expected,
176 const Array4D<NativeT>& expected,
523 XlaBuilder* builder, const Array4D<NativeT>& expected, in ComputeAndCompareR4()
533 XlaBuilder* builder, const Array4D<NativeT>& expected, in ComputeAndCompareR4()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc91 Array4D<float>(1, 0, 1, 1), /*init=*/0, /*dims=*/{0, 1, 2}, in TEST_F()
115 auto input = absl::make_unique<Array4D<float>>(/*planes=*/2, /*depth=*/3, in TEST_F()
122 Array4D<float> expected(/*planes=*/2, /*depth=*/3, /*height=*/4, /*width=*/5); in TEST_F()
129 auto input = absl::make_unique<Array4D<float>>(/*planes=*/2, /*depth=*/3, in TEST_F()
139 Array4D<float> expected(/*planes=*/2, /*depth=*/3, /*height=*/4, /*width=*/5); in TEST_F()
188 Array4D<float> input(2, 3, 4, 5); in TEST_F()
201 Array4D<float> input(2, 3, 4, 5); in TEST_F()
241 Array4D<float> input(1, 1, 4, 4); in TEST_F()
250 Array4D<float> weights(1, 1, 2, 2); in TEST_F()
257 std::unique_ptr<Array4D<float>> actual = in TEST_F()
[all …]
Dreference_util.h71 static std::unique_ptr<Array4D<float>> ConvArray4D(
72 const Array4D<float>& lhs, const Array4D<float>& rhs,
77 static std::unique_ptr<Array4D<float>> ConvArray4DGeneralDimensions(
78 const Array4D<float>& lhs, const Array4D<float>& rhs,
84 static std::unique_ptr<Array4D<float>> ConvArray4DGeneralDimensionsDilated(
85 const Array4D<float>& lhs, const Array4D<float>& rhs,
108 static std::unique_ptr<Array4D<float>> SeparableConvArray4D(
109 const Array4D<float>& input, const Array4D<float>& depthwise_weights,
110 const Array4D<float>& pointwise_weights,
147 const Array4D<float>& array, float init, absl::Span<const int64> dims,
[all …]
Dreference_util.cc81 Array4D<float> a4dlhs(lhs.n1(), lhs.n2(), lhs.n3(), 1); in ConvArray3DGeneralDimensionsDilated()
86 Array4D<float> a4drhs(rhs.n1(), rhs.n2(), rhs.n3(), 1); in ConvArray3DGeneralDimensionsDilated()
96 std::unique_ptr<Array4D<float>> convr4 = ConvArray4DGeneralDimensionsDilated( in ConvArray3DGeneralDimensionsDilated()
109 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::ConvArray4D( in ConvArray4D()
110 const Array4D<float>& lhs, const Array4D<float>& rhs, in ConvArray4D()
117 /* static */ std::unique_ptr<Array4D<float>>
118 ReferenceUtil::SeparableConvArray4D(const Array4D<float>& input, in SeparableConvArray4D()
119 const Array4D<float>& depthwise_weights, in SeparableConvArray4D()
120 const Array4D<float>& pointwise_weights, in SeparableConvArray4D()
128 Array4D<float> weights(pointwise_weights.planes(), input.depth(), in SeparableConvArray4D()
[all …]
Darray4d_test.cc30 int64 Array4DLinearIndex(const Array4D<T>& arr, absl::Span<const int64> idx) { in Array4DLinearIndex()
37 Array4D<int> empty(2, 3, 4, 5); in TEST()
46 Array4D<int> fullof7(2, 3, 4, 5, 7); in TEST()
63 Array4D<int> arr(2, 3, 4, 5, filler); in TEST()
76 Array4D<int> arr = {{{{1}, {2}}, {{3}, {4}}, {{5}, {6}}, {{7}, {8}}}, in TEST()
99 Array4D<Eigen::half> arr = { in TEST()
129 Array4D<int> fullof7(2, 3, 4, 5, 7); in TEST()
139 Array4D<float> arr(2, 3, 4, 5); in TEST()
148 Array4D<float> array(1, 1, 128, 128); in TEST()
192 Array4D<float> actual(3, 2, 1, 1); in TEST()
Darray4d.h55 class Array4D : public Array<T> {
58 Array4D(int64 planes, int64 depth, int64 height, int64 width) in Array4D() function
62 Array4D(int64 planes, int64 depth, int64 height, int64 width, T value) in Array4D() function
71 Array4D(int64 planes, int64 depth, int64 height, int64 width, in Array4D() function
73 : Array4D(planes, depth, height, width) { in Array4D()
78 Array4D(std::initializer_list<std::initializer_list< in Array4D() function
91 Array4D(std::initializer_list<std::initializer_list< in Array4D() function
Dliteral_util.h179 static Literal CreateR4FromArray4D(const Array4D<NativeT>& values);
181 static Literal CreateR4FromArray4DWithLayout(const Array4D<NativeT>& values,
391 Array4D<NativeT> tmp(d0, d1, d2, d3); in CreateR4WithLayout()
510 Array4D<NativeT> array(dim0_size, dim1_size, dim2_size, dim3_size); in CreateR4Projected()
531 const Array4D<NativeT>& values) { in CreateR4FromArray4D()
537 const Array4D<NativeT>& values, const Layout& layout) { in CreateR4FromArray4DWithLayout()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization_test.cc63 LiteralUtil::CreateR4FromArray4D(Array4D<float>( in TEST_F()
67 LiteralUtil::CreateR4FromArray4D(Array4D<float>( in TEST_F()
126 LiteralUtil::CreateR4FromArray4D(Array4D<float>( in TEST_F()
130 LiteralUtil::CreateR4FromArray4D(Array4D<float>( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dinfeed_test.cc95 Array4D<float> array(80, 100, 8, 128); in TEST_F()
112 Array4D<float> array(40, 100, 8, 128); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc599 Array4D<float> input_array(3, 2, 1, 1, {1, 2, 3, 4, 5, 6}); in TEST_P()
617 auto expected_array = absl::make_unique<Array4D<float>>(8, 5, 1, 1); in TEST_P()
849 auto lhs_array = absl::make_unique<Array4D<float>>(2, 2, 3, 1); in TEST_P()
855 auto rhs_array = absl::make_unique<Array4D<float>>(2, 2, 3, 1); in TEST_P()
943 Array4D<float> lhs_array(1, 1, 4, 4); in TEST_P()
956 Array4D<float> rhs_array(1, 1, 2, 2); in TEST_P()
989 Array4D<float> expected_array(1, 1, 4, 4); in TEST_P()
1008 Array4D<float> input({ in TEST_P()
1018 Array4D<float> weight({{ in TEST_P()
1075 Array4D<float> expected_array({{{{2514, 2685}}}}); in TEST_P()
[all …]
Dhlo_sharding_test.cc264 HloSharding::Tile(Array4D<int64>({{{{0, 1}, {2, 3}}}})); in TEST_F()
278 check(HloSharding::Tile(Array4D<int64>({{{{0}, {1}}}}))); in TEST_F()
290 tuple_shape, {HloSharding::Tile(Array4D<int64>({{{{0}, {1}}}})), in TEST_F()
300 HloSharding::Tile(Array4D<int64>({{{{0}, {1}}}})), in TEST_F()

12