Home
last modified time | relevance | path

Searched refs:Array4D (Results 1 – 25 of 40) 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.cc361 Array4D<float> o(4, 6, 15, 220); in TEST_F()
364 Array4D<float> e(4, 6, 15, 220); in TEST_F()
366 Array4D<float> s(2, 2, 15, 220); in TEST_F()
386 Array4D<float> o(4, 5, 17, 128); in TEST_F()
389 Array4D<float> e(4, 5, 17, 128); in TEST_F()
391 Array4D<float> s(2, 2, 17, 128); in TEST_F()
411 Array4D<float> o(4, 5, 1, 1); in TEST_F()
414 Array4D<float> e(4, 5, 1, 1); in TEST_F()
416 Array4D<float> s(2, 2, 1, 1); in TEST_F()
433 Array4D<float> o(4, 6, 4, 4); in TEST_F()
[all …]
Dreverse_test.cc120 Array4D<uint8> input({{ in XLA_TEST_F()
133 Array4D<uint8> expected({{ in XLA_TEST_F()
150 Array4D<float> input({ in TEST_F()
169 Array4D<float> expected({ in TEST_F()
Dbatch_normalization_test.cc88 Array4D<float> input_array_;
110 Array4D<float> expected(kSamples, kZ, kY, kX); in XLA_TEST_P()
127 Array4D<float> expected(kSamples, kZ, kY, kX); in XLA_TEST_P()
221 Array4D<float> expected(kSamples, kZ, kY, kX); in XLA_TEST_P()
258 Array4D<Eigen::half> input = {{{{1.f, 2.f}}, {{3.f, 4.f}}}, in XLA_TEST_P()
275 Array4D<Eigen::half> out = {{{{-1.6f, -2.0f}}, {{0.1f, 0.6f}}}, in XLA_TEST_P()
312 Array4D<Eigen::half> input = {{{{1.f}, {2.f}}, {{3.f}, {4.f}}}, in XLA_TEST_P()
329 Array4D<Eigen::half> out = {{{{-1.6f}, {-2.0f}}, {{0.1f}, {0.6f}}}, in XLA_TEST_P()
408 ConstantR4FromArray4D<float>(&builder, Array4D<float>(2, 2, 2, 1, 0.0f)); in XLA_TEST_P()
437 Array4D<Eigen::half>(2, 2, 2, 1, static_cast<Eigen::half>(0.0f))); 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()
Dreshape_test.cc330 LiteralUtil::CreateFromArray(Array4D<float>(2, 3, 0, 0)); in XLA_TEST_P()
337 auto input_literal = LiteralUtil::CreateFromArray(Array4D<float>(2, 3, 4, 0)); in XLA_TEST_P()
519 Array4D<float> t2x2x2x3(2, 2, 2, 3); in XLA_TEST_P()
539 Array4D<float> t(2, 1, 2, 2); in XLA_TEST_P()
614 Array4D<float>{ in XLA_TEST_P()
726 Array4D<float> input(2, 1, 1, 1); in XLA_TEST_P()
746 Array4D<float> input(2, 1, 4, 1); in XLA_TEST_P()
767 Array4D<float> input(5, 10, 2, 3); in XLA_TEST_P()
793 Array4D<float> input_array(2, 3, 5, 7); in XLA_TEST_P()
877 Array4D<float> input(bounds[0], bounds[1], bounds[2], bounds[3]); in XLA_TEST_P()
[all …]
Dreduce_window_test.cc165 Array4D<float> input_array(1, 0, 2, 1); in XLA_TEST_P()
178 Array4D<float> input_array(1, 2, 2, 1); in XLA_TEST_P()
193 Array4D<float> input_array(1, 3, 3, 1); in XLA_TEST_P()
207 Array4D<float> input_array(3, 6, 7, 32); in XLA_TEST_P()
224 Array4D<float> input_array(4, 4, 6, 8); in XLA_TEST_P()
246 Array4D<float> input_array(3, 3, 2, 1); in XLA_TEST_P()
260 Array4D<float> input_array(9, 12, 4, 89); in XLA_TEST_P()
285 Array4D<float> input_array(15, 15, 4, 128); in XLA_TEST_P()
308 Array4D<float> input_array(19, 17, 8, 256); in XLA_TEST_P()
326 Array4D<float> input_array(1, 2, 2, 1); 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()
493 Array4D<float> values(spec.input_dims[0], spec.input_dims[1], in Run()
Dclient_library_test_base.h174 const Array4D<NativeT>& expected,
178 const Array4D<NativeT>& expected,
393 const Array4D<NativeT>& array_4d, int64 parameter_number,
554 XlaBuilder* builder, const Array4D<NativeT>& expected, in ComputeAndCompareR4()
564 XlaBuilder* builder, const Array4D<NativeT>& expected, in ComputeAndCompareR4()
662 const Array4D<NativeT>& array_4d, int64 parameter_number, in CreateR4Parameter()
/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.cc66 Array4D<float> a4dlhs(lhs.n1(), lhs.n2(), lhs.n3(), 1); in ConvArray3DGeneralDimensionsDilated()
71 Array4D<float> a4drhs(rhs.n1(), rhs.n2(), rhs.n3(), 1); in ConvArray3DGeneralDimensionsDilated()
81 std::unique_ptr<Array4D<float>> convr4 = ConvArray4DGeneralDimensionsDilated( in ConvArray3DGeneralDimensionsDilated()
94 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::ConvArray4D( in ConvArray4D()
95 const Array4D<float>& lhs, const Array4D<float>& rhs, in ConvArray4D()
102 /* static */ std::unique_ptr<Array4D<float>>
103 ReferenceUtil::SeparableConvArray4D(const Array4D<float>& input, in SeparableConvArray4D()
104 const Array4D<float>& depthwise_weights, in SeparableConvArray4D()
105 const Array4D<float>& pointwise_weights, in SeparableConvArray4D()
113 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> {
57 Array4D() : Array<T>(std::vector<int64>{0, 0, 0, 0}) {} in Array4D() function
60 Array4D(int64 planes, int64 depth, int64 height, int64 width) in Array4D() function
64 Array4D(int64 planes, int64 depth, int64 height, int64 width, T value) in Array4D() function
73 Array4D(int64 planes, int64 depth, int64 height, int64 width, in Array4D() function
75 : Array4D(planes, depth, height, width) { in Array4D()
80 Array4D(std::initializer_list<std::initializer_list< in Array4D() function
93 Array4D(std::initializer_list<std::initializer_list< in Array4D() function
Dliteral_util.h142 static Literal CreateR4FromArray4D(const Array4D<NativeT>& values);
144 static Literal CreateR4FromArray4DWithLayout(const Array4D<NativeT>& values,
373 Array4D<NativeT> tmp(d0, d1, d2, d3); in CreateR4WithLayout()
477 Array4D<NativeT> array(dim0_size, dim1_size, dim2_size, dim3_size); in CreateR4Projected()
498 const Array4D<NativeT>& values) { in CreateR4FromArray4D()
504 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_sharding_util_test.cc30 HloSharding input = HloSharding::Tile(Array4D<int64>({{{{0, 1}}, {{2, 3}}}})); in TEST()
32 HloSharding::Tile(Array4D<int64>({{{{0}, {2}}}, {{{1}, {3}}}})); in TEST()
123 HloSharding::Tile(Array4D<int64>({{{{0}, {1}}}})); in TEST()
125 HloSharding::Tile(Array4D<int64>({{{{0}}, {{1}}}})); in TEST()
Dhlo_evaluator_test.cc676 Array4D<float> input_array(3, 2, 1, 1, {1, 2, 3, 4, 5, 6}); in TEST_P()
694 auto expected_array = absl::make_unique<Array4D<float>>(8, 5, 1, 1); in TEST_P()
926 auto lhs_array = absl::make_unique<Array4D<float>>(2, 2, 3, 1); in TEST_P()
932 auto rhs_array = absl::make_unique<Array4D<float>>(2, 2, 3, 1); in TEST_P()
1020 Array4D<float> lhs_array(1, 1, 4, 4); in TEST_P()
1033 Array4D<float> rhs_array(1, 1, 2, 2); in TEST_P()
1066 Array4D<float> expected_array(1, 1, 4, 4); in TEST_P()
1085 Array4D<float> input({ in TEST_P()
1095 Array4D<float> weight({{ in TEST_P()
1152 Array4D<float> expected_array({{{{2514, 2685}}}}); in TEST_P()
[all …]
Dhlo_sharding_test.cc381 HloSharding::Tile(Array4D<int64>({{{{0, 1}, {2, 3}}}})); in TEST_F()
399 check(HloSharding::Tile(Array4D<int64>({{{{0}, {1}}}}), GetParam())); in TEST_P()
412 {HloSharding::Tile(Array4D<int64>({{{{0}, {1}}}})), in TEST_P()
422 HloSharding::Tile(Array4D<int64>({{{{0}, {1}}}})), in TEST_P()

12