Home
last modified time | relevance | path

Searched refs:BHWC (Results 1 – 25 of 152) sorted by relevance

1234567

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dslice_test.cc34 TensorRef<BHWC> input; in TEST()
37 input.shape = BHWC(1, 1, 2, 2); in TEST()
39 TensorRef<BHWC> output; in TEST()
42 output.shape = BHWC(1, 1, 2, 2); in TEST()
45 attr.starts = BHWC(0, 0, 0, 0); in TEST()
46 attr.ends = BHWC(input.shape.b, 1, 2, 2); in TEST()
47 attr.strides = BHWC(1, 1, 1, 1); in TEST()
57 TensorRef<BHWC> input; in TEST()
60 input.shape = BHWC(1, 1, 2, 2); in TEST()
62 TensorRef<BHWC> output; in TEST()
[all …]
Dpad_test.cc36 const BHWC& output_shape, std::vector<float>&& expected) { in TestPadOperation()
37 TensorRef<BHWC> input; in TestPadOperation()
40 input.shape = BHWC(1, 1, 1, 1); in TestPadOperation()
42 TensorRef<BHWC> output; in TestPadOperation()
48 attr.prepended = BHWC(0, prepend.h, prepend.w, prepend.c); in TestPadOperation()
49 attr.appended = BHWC(0, append.h, append.w, append.c); in TestPadOperation()
58 void TestPrepending(const HWC& prepend, const BHWC& output_shape, in TestPrepending()
63 void TestAppending(const HWC& append, const BHWC& output_shape, in TestAppending()
72 /*output_shape=*/BHWC(1, 2, 1, 1), /*expected=*/{0, 1}); in TEST()
76 TestPrepending(/*prepend=*/HWC(0, 1, 0), /*output_shape=*/BHWC(1, 1, 2, 1), in TEST()
[all …]
Dresize_test.cc32 TensorRef<BHWC> input; in TEST()
35 input.shape = BHWC(1, 1, 1, 2); in TEST()
37 TensorRef<BHWC> output; in TEST()
40 output.shape = BHWC(1, 2, 2, 2); in TEST()
57 TensorRef<BHWC> input; in TEST()
60 input.shape = BHWC(1, 1, 2, 1); in TEST()
62 TensorRef<BHWC> output; in TEST()
65 output.shape = BHWC(1, 1, 4, 1); in TEST()
81 TensorRef<BHWC> input; in TEST()
84 input.shape = BHWC(1, 2, 2, 1); in TEST()
[all …]
Dmul_test.cc34 TensorRef<BHWC> input; in TEST()
37 input.shape = BHWC(1, 2, 2, 1); in TEST()
39 TensorRef<BHWC> output; in TEST()
42 output.shape = BHWC(1, 2, 2, 1); in TEST()
54 TensorRef<BHWC> input; in TEST()
57 input.shape = BHWC(1, 1, 2, 2); in TEST()
59 TensorRef<BHWC> output; in TEST()
62 output.shape = BHWC(1, 1, 2, 2); in TEST()
78 TensorRef<BHWC> input; in TEST()
81 input.shape = BHWC(1, 1, 2, 2); in TEST()
[all …]
Dspace_to_depth_test.cc34 const TensorRef<BHWC> input = { in TEST()
35 .type = DataType::FLOAT32, .shape = BHWC(1, 2, 2, 1), .ref = 0}; in TEST()
36 const TensorRef<BHWC> output = { in TEST()
37 .type = DataType::FLOAT32, .shape = BHWC(1, 1, 1, 4), .ref = 1}; in TEST()
48 const TensorRef<BHWC> input = { in TEST()
49 .type = DataType::FLOAT32, .shape = BHWC(1, 2, 2, 2), .ref = 0}; in TEST()
50 const TensorRef<BHWC> output = { in TEST()
51 .type = DataType::FLOAT32, .shape = BHWC(1, 1, 1, 8), .ref = 1}; in TEST()
64 const TensorRef<BHWC> input = { in TEST()
65 .type = DataType::FLOAT32, .shape = BHWC(1, 2, 2, 3), .ref = 0}; in TEST()
[all …]
Dsoftmax_test.cc36 TensorRef<BHWC> input; in TEST()
39 input.shape = BHWC(1, 2, 2, 1); in TEST()
41 TensorRef<BHWC> output; in TEST()
44 output.shape = BHWC(1, 2, 2, 1); in TEST()
58 TensorRef<BHWC> input; in TEST()
61 input.shape = BHWC(1, 2, 2, 1); in TEST()
63 TensorRef<BHWC> output; in TEST()
66 output.shape = BHWC(1, 2, 2, 1); in TEST()
78 TensorRef<BHWC> input; in TEST()
81 input.shape = BHWC(1, 2, 2, 1); in TEST()
[all …]
Dprelu_test.cc32 TensorRef<BHWC> input; in TEST()
35 input.shape = BHWC(1, 2, 2, 1); in TEST()
45 TensorRef<BHWC> output; in TEST()
48 output.shape = BHWC(1, 2, 2, 1); in TEST()
58 TensorRef<BHWC> input; in TEST()
61 input.shape = BHWC(1, 2, 2, 1); in TEST()
71 TensorRef<BHWC> output; in TEST()
74 output.shape = BHWC(1, 2, 2, 1); in TEST()
84 TensorRef<BHWC> input;
87 input.shape = BHWC(1, 2, 2, 1);
[all …]
Dreshape_test.cc34 TensorRef<BHWC> input;
37 input.shape = BHWC(1, 1, 2, 3);
39 TensorRef<BHWC> output;
42 output.shape = BHWC(1, 3, 2, 1);
56 TensorRef<BHWC> input;
59 input.shape = BHWC(1, 3, 1, 2);
61 TensorRef<BHWC> output;
64 output.shape = BHWC(1, 2, 1, 3);
78 TensorRef<BHWC> input;
81 input.shape = BHWC(1, 1, 1, 4);
[all …]
Dadd_test.cc34 TensorRef<BHWC> augend, addend, output; in TEST()
37 augend.shape = BHWC(1, 2, 2, 1); in TEST()
41 addend.shape = BHWC(1, 2, 2, 1); in TEST()
45 output.shape = BHWC(1, 2, 2, 1); in TEST()
60 TensorRef<BHWC> input, output; in TEST()
63 input.shape = BHWC(1, 3, 1, 2); in TEST()
67 output.shape = BHWC(1, 3, 1, 2); in TEST()
78 TensorRef<BHWC> input; in TEST()
81 input.shape = BHWC(1, 2, 2, 2); in TEST()
91 TensorRef<BHWC> output; in TEST()
[all …]
Dquantize_and_dequantize_test.cc33 TensorRef<BHWC> input; in TEST()
36 input.shape = BHWC(1, 3, 2, 1); in TEST()
48 TensorRef<BHWC> output; in TEST()
51 output.shape = BHWC(1, 3, 2, 1); in TEST()
64 TensorRef<BHWC> input; in TEST()
67 input.shape = BHWC(1, 3, 1, 2); in TEST()
79 TensorRef<BHWC> output; in TEST()
82 output.shape = BHWC(1, 3, 1, 2); in TEST()
95 TensorRef<BHWC> input; in TEST()
98 input.shape = BHWC(1, 3, 1, 2); in TEST()
[all …]
Dconcat_test.cc34 TensorRef<BHWC> input1, input2, output; in TEST()
37 input1.shape = BHWC(1, 2, 2, 1); in TEST()
41 input2.shape = BHWC(1, 2, 2, 1); in TEST()
45 output.shape = BHWC(1, 2, 2, 2); in TEST()
60 TensorRef<BHWC> input1, input2, output; in TEST()
63 input1.shape = BHWC(1, 1, 1, 4); in TEST()
67 input2.shape = BHWC(1, 1, 1, 4); in TEST()
71 output.shape = BHWC(1, 1, 1, 8); in TEST()
86 TensorRef<BHWC> input1, input2, output; in TEST()
89 input1.shape = BHWC(1, 1, 2, 1); in TEST()
[all …]
Delementwise_test.cc32 TensorRef<BHWC> GetTensorRef(int ref, const BHWC& shape) { in GetTensorRef()
33 TensorRef<BHWC> tensor_ref; in GetTensorRef()
42 const BHWC shape(1, 2, 2, 1); in TEST()
54 const BHWC shape(1, 2, 2, 1); in TEST()
66 const BHWC shape(1, 2, 2, 1); in TEST()
77 const BHWC shape(1, 1, 1, 7); in TEST()
92 const BHWC shape(1, 1, 1, 7); in TEST()
108 const BHWC shape(1, 1, 1, 7); in TEST()
122 const BHWC shape(1, 2, 2, 1); in TEST()
134 const BHWC shape(1, 2, 2, 1); in TEST()
[all …]
Dpooling_test.cc38 TensorRef<BHWC> input; in TEST()
41 input.shape = BHWC(1, 4, 4, 1); in TEST()
43 TensorRef<BHWC> output; in TEST()
46 output.shape = BHWC(1, 2, 2, 1); in TEST()
48 TensorRef<BHWC> indices; in TEST()
51 indices.shape = BHWC(1, 2, 2, 1); in TEST()
73 TensorRef<BHWC> input; in TEST()
76 input.shape = BHWC(1, 4, 4, 1); in TEST()
78 TensorRef<BHWC> output; in TEST()
81 output.shape = BHWC(1, 2, 2, 1); in TEST()
[all …]
Dtranspose_conv_test.cc34 TensorRef<BHWC> input; in TEST()
37 input.shape = BHWC(1, 2, 2, 1); in TEST()
57 TensorRef<BHWC> output; in TEST()
60 output.shape = BHWC(1, 3, 3, 2); in TEST()
73 TensorRef<BHWC> input; in TEST()
76 input.shape = BHWC(1, 3, 3, 1); in TEST()
96 TensorRef<BHWC> output; in TEST()
99 output.shape = BHWC(1, 6, 6, 1); in TEST()
114 TensorRef<BHWC> input; in TEST()
117 input.shape = BHWC(1, 2, 2, 1); in TEST()
[all …]
Dconv_test.cc34 TensorRef<BHWC> input; in TEST()
37 input.shape = BHWC(1, 2, 2, 1); in TEST()
57 TensorRef<BHWC> output; in TEST()
60 output.shape = BHWC(1, 2, 2, 2); in TEST()
72 TensorRef<BHWC> input; in TEST()
75 input.shape = BHWC(1, 3, 3, 1); in TEST()
95 TensorRef<BHWC> output; in TEST()
98 output.shape = BHWC(1, 1, 1, 1); in TEST()
109 TensorRef<BHWC> input; in TEST()
112 input.shape = BHWC(1, 2, 2, 1); in TEST()
[all …]
Dlstm_test.cc34 TensorRef<BHWC> input; in TEST()
37 input.shape = BHWC(1, 2, 2, 1); in TEST()
39 TensorRef<BHWC> prev_state; in TEST()
42 prev_state.shape = BHWC(1, 2, 2, 1); in TEST()
44 TensorRef<BHWC> output_state; in TEST()
47 output_state.shape = BHWC(1, 2, 2, 1); in TEST()
49 TensorRef<BHWC> output_activation; in TEST()
52 output_activation.shape = BHWC(1, 2, 2, 1); in TEST()
Ddepthwise_conv_test.cc34 TensorRef<BHWC> input; in TEST()
37 input.shape = BHWC(1, 1, 1, 2); in TEST()
58 TensorRef<BHWC> output; in TEST()
61 output.shape = BHWC(1, 1, 1, 4); in TEST()
72 TensorRef<BHWC> input; in TEST()
75 input.shape = BHWC(1, 3, 3, 1); in TEST()
96 TensorRef<BHWC> output; in TEST()
99 output.shape = BHWC(1, 2, 2, 2); in TEST()
111 TensorRef<BHWC> input; in TEST()
114 input.shape = BHWC(1, 3, 3, 1); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h205 BHWC CalculateOutputShape(const BHWC& input,
215 BHWC CalculateOutputShape(const BHWC& input, const Pooling2DAttributes& attr);
223 absl::Status CalculateOutputShape(const std::vector<BHWC>& input,
225 BHWC* output_shape);
235 Padding2D CalculateSamePadding(const BHWC& input,
245 Padding2D CalculateSamePadding(const BHWC& input,
273 BHWC CalculateOutputShape(const BHWC& input,
283 Padding2D CalculateSamePadding(const BHWC& input,
308 Padding2D CalculateSamePadding(const BHWC& input,
316 BHWC CalculateOutputShape(const BHWC& input,
[all …]
Dmemory_management_test.cc273 std::vector<TensorUsageRecord<BHWC>> usage_records{ in TEST()
274 {/*size=*/BHWC(1, 1, 2, 8), /*first=*/0, /*last=*/1}, in TEST()
275 {/*size=*/BHWC(1, 1, 2, 8), /*first=*/1, /*last=*/2}, in TEST()
276 {/*size=*/BHWC(1, 1, 1, 16), /*first=*/2, /*last=*/4}, in TEST()
277 {/*size=*/BHWC(1, 1, 2, 8), /*first=*/3, /*last=*/5}, in TEST()
278 {/*size=*/BHWC(1, 1, 8, 2), /*first=*/4, /*last=*/5}, in TEST()
279 {/*size=*/BHWC(1, 1, 2, 8), /*first=*/5, /*last=*/7}, in TEST()
280 {/*size=*/BHWC(1, 16, 1, 1), /*first=*/6, /*last=*/8}, in TEST()
281 {/*size=*/BHWC(16, 1, 1, 1), /*first=*/7, /*last=*/8}, in TEST()
282 {/*size=*/BHWC(1, 1, 1, 16), /*first=*/8, /*last=*/9}}; in TEST()
[all …]
Doperations.cc277 int32_t CalculateOutputWithoutStrides(const BHWC& input, in CalculateOutputWithoutStrides()
295 int32_t CalculateOutputWithoutStrides(const BHWC& input, in CalculateOutputWithoutStrides()
313 int32_t CalculateOutput(const BHWC& input, in CalculateOutput()
333 int32_t CalculateOutput(const BHWC& input, const AttrT& attr) { in CalculateOutput()
353 int32_t CalculateSamePadding(const BHWC& input, in CalculateSamePadding()
371 int32_t CalculateSamePadding(const BHWC& input, in CalculateSamePadding()
387 int32_t CalculateSamePadding(const BHWC& input, in CalculateSamePadding()
401 int32_t CalculateSamePadding(const BHWC& input, in CalculateSamePadding()
414 Padding2D MakeSamePadding(const BHWC& input, in MakeSamePadding()
441 Padding2D MakeSamePadding(const BHWC& input, const AttrT& attr) { in MakeSamePadding()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dpadding_test_util.cc30 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingAppendWidthTest()
34 attr.prepended = BHWC(0, 0, 0, 0); in PaddingAppendWidthTest()
35 attr.appended = BHWC(0, 0, 1, 0); in PaddingAppendWidthTest()
49 BHWC(1, 2, 2, 2), &dst_tensor)); in PaddingAppendWidthTest()
60 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingPrependWidthTest()
64 attr.prepended = BHWC(0, 0, 1, 0); in PaddingPrependWidthTest()
65 attr.appended = BHWC(0, 0, 0, 0); in PaddingPrependWidthTest()
79 BHWC(1, 2, 2, 2), &dst_tensor)); in PaddingPrependWidthTest()
90 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingAppendHeightTest()
94 attr.prepended = BHWC(0, 0, 0, 0); in PaddingAppendHeightTest()
[all …]
Delementwise_test_util.cc30 src_tensor.shape = BHWC(1, 2, 1, 2); in AbsTest()
45 BHWC(1, 2, 1, 2), &dst_tensor)); in AbsTest()
56 src_tensor.shape = BHWC(1, 2, 1, 2); in CosTest()
72 BHWC(1, 2, 1, 2), &dst_tensor)); in CosTest()
83 src_tensor.shape = BHWC(1, 2, 1, 2); in CopyTest()
98 BHWC(1, 2, 1, 2), &dst_tensor)); in CopyTest()
107 src_tensor.shape = BHWC(1, 1, 1, 7); in EluTest()
123 BHWC(1, 1, 1, 7), &dst_tensor)); in EluTest()
135 src_tensor.shape = BHWC(1, 1, 1, 7); in ExpTest()
151 BHWC(1, 1, 1, 7), &dst_tensor)); in ExpTest()
[all …]
Dmean_stddev_normalization_test_util.cc34 src_tensor.shape = BHWC(1, 1, 1, 4); in MeanStddevNormSeparateBatchesTest()
42 op_def.src_tensors.push_back({data_type, storage, Layout::BHWC}); in MeanStddevNormSeparateBatchesTest()
43 op_def.dst_tensors.push_back({data_type, storage, Layout::BHWC}); in MeanStddevNormSeparateBatchesTest()
50 BHWC(1, 1, 1, 4), &dst_tensor)); in MeanStddevNormSeparateBatchesTest()
70 src_tensor.shape = BHWC(9, 1, 1, 4); in MeanStddevNormalizationAllBatchesTest()
89 op_def.src_tensors.push_back({data_type, storage, Layout::BHWC}); in MeanStddevNormalizationAllBatchesTest()
90 op_def.dst_tensors.push_back({data_type, storage, Layout::BHWC}); in MeanStddevNormalizationAllBatchesTest()
97 BHWC(9, 1, 1, 4), &dst_tensor)); in MeanStddevNormalizationAllBatchesTest()
127 src_tensor.shape = BHWC(1, 1, 1, kVectorSize); in MeanStddevNormalizationLargeVectorTest()
144 op_def.src_tensors.push_back({data_type, storage, Layout::BHWC}); in MeanStddevNormalizationLargeVectorTest()
[all …]
Dlstm_test_util.cc28 src_tensor.shape = BHWC(1, 1, 1, 16); in LstmTest()
45 prev_state.shape = BHWC(1, 1, 1, 4); in LstmTest()
54 op_def.src_tensors.push_back({data_type, storage, Layout::BHWC}); in LstmTest()
55 op_def.src_tensors.push_back({data_type, storage, Layout::BHWC}); in LstmTest()
56 op_def.dst_tensors.push_back({data_type, storage, Layout::BHWC}); in LstmTest()
57 op_def.dst_tensors.push_back({data_type, storage, Layout::BHWC}); in LstmTest()
64 {BHWC(1, 1, 1, 4), BHWC(1, 1, 1, 4)}, {&new_state, &new_activ})); in LstmTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor_test.cc32 absl::Status TensorGenericTest(const BHWC& shape, in TensorGenericTest()
93 BHWC(1, 6, 7, 3), {data_type, storage_type, Layout::HWC}, env)); in TensorTests()
95 BHWC(1, 1, 4, 12), {data_type, storage_type, Layout::HWC}, env)); in TensorTests()
97 BHWC(1, 6, 1, 7), {data_type, storage_type, Layout::HWC}, env)); in TensorTests()
101 BHWC(2, 6, 7, 3), {data_type, storage_type, Layout::BHWC}, env)); in TensorTests()
103 BHWC(4, 1, 4, 12), {data_type, storage_type, Layout::BHWC}, env)); in TensorTests()
105 BHWC(7, 6, 1, 7), {data_type, storage_type, Layout::BHWC}, env)); in TensorTests()
107 BHWC(13, 7, 3, 3), {data_type, storage_type, Layout::BHWC}, env)); in TensorTests()
179 BHWC(1, 6, 14, 1), in TEST_F()
183 BHWC(1, 6, 14, 2), in TEST_F()
[all …]

1234567