Home
last modified time | relevance | path

Searched refs:HWC (Results 1 – 25 of 79) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dpad_test.cc35 void TestPadOperation(const HWC& prepend, const HWC& append, in TestPadOperation()
58 void TestPrepending(const HWC& prepend, const BHWC& output_shape, in TestPrepending()
60 TestPadOperation(prepend, HWC(0, 0, 0), output_shape, std::move(expected)); in TestPrepending()
63 void TestAppending(const HWC& append, const BHWC& output_shape, in TestAppending()
65 TestPadOperation(HWC(0, 0, 0), append, output_shape, std::move(expected)); in TestAppending()
71 TestPrepending(/*prepend=*/HWC(1, 0, 0), in TEST()
76 TestPrepending(/*prepend=*/HWC(0, 1, 0), /*output_shape=*/BHWC(1, 1, 2, 1), in TEST()
81 TestPrepending(/*prepend=*/HWC(0, 0, 1), /*output_shape=*/BHWC(1, 1, 1, 2), in TEST()
86 TestPrepending(/*prepend=*/HWC(0, 0, 4), /*output_shape=*/BHWC(1, 1, 1, 5), in TEST()
91 TestPrepending(/*prepend=*/HWC(1, 1, 1), /*output_shape=*/BHWC(1, 2, 2, 2), in TEST()
[all …]
Dprelu_test.cc92 Tensor<HWC, DataType::FLOAT32> alpha;
93 alpha.shape = HWC(2, 2, 1);
118 Tensor<HWC, DataType::FLOAT32> alpha;
119 alpha.shape = HWC(2, 2, 1);
144 Tensor<HWC, DataType::FLOAT32> alpha;
145 alpha.shape = HWC(2, 1, 1);
170 Tensor<HWC, DataType::FLOAT32> alpha;
171 alpha.shape = HWC(2, 2, 2);
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Delementwise_test_util.cc38 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AbsTest()
39 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AbsTest()
65 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in CosTest()
66 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in CosTest()
91 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in CopyTest()
92 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in CopyTest()
116 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in EluTest()
117 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in EluTest()
144 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ExpTest()
145 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ExpTest()
[all …]
Dpadding_test_util.cc43 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendWidthTest()
44 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendWidthTest()
73 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependWidthTest()
74 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependWidthTest()
103 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendHeightTest()
104 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendHeightTest()
132 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependHeightTest()
133 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingPrependHeightTest()
161 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendChannelsTest()
162 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PaddingAppendChannelsTest()
[all …]
Dconcat_test_util.cc45 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatWidthTest()
46 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatWidthTest()
47 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatWidthTest()
78 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatHeightTest()
79 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatHeightTest()
80 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatHeightTest()
112 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatChannelsTest()
113 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatChannelsTest()
114 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatChannelsTest()
115 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConcatChannelsTest()
[all …]
Dadd_test_util.cc40 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddTwoEqualTensorsTest()
41 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddTwoEqualTensorsTest()
42 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AddTwoEqualTensorsTest()
70 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasMoreChannelsThanSecondTest()
71 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasMoreChannelsThanSecondTest()
72 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasMoreChannelsThanSecondTest()
101 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasLessChannelsThanSecond()
102 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasLessChannelsThanSecond()
103 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AddFirstTensorHasLessChannelsThanSecond()
Dresize_test_util.cc44 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearAlignedTest()
45 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearAlignedTest()
76 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearNonAlignedTest()
77 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearNonAlignedTest()
109 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearWithoutHalfPixelTest()
110 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearWithoutHalfPixelTest()
141 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearWithHalfPixelTest()
142 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeBilinearWithHalfPixelTest()
173 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeNearestTest()
174 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ResizeNearestTest()
[all …]
Dprelu_test_util.cc46 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PReLUAlphaTest()
47 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PReLUAlphaTest()
78 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PReLUAlphaClipTest()
79 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PReLUAlphaClipTest()
98 ::tflite::gpu::Tensor<HWC, DataType::FLOAT32> hwc_tensor; in PReLUHWCAlphaTest()
99 hwc_tensor.shape = HWC(2, 1, 2); in PReLUHWCAlphaTest()
110 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in PReLUHWCAlphaTest()
111 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in PReLUHWCAlphaTest()
Drelu_test_util.cc43 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUNoClipNoAlphaTest()
44 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUNoClipNoAlphaTest()
72 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUClipTest()
73 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUClipTest()
101 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUAlphaTest()
102 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUAlphaTest()
130 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUAlphaClipTest()
131 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReLUAlphaClipTest()
Dreduce_test_util.cc40 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in MeanHWTest()
41 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in MeanHWTest()
67 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceSumChannelsTest()
68 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceSumChannelsTest()
94 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceProductChannelsTest()
95 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceProductChannelsTest()
122 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceMaxChannelsTest()
123 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceMaxChannelsTest()
150 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceMinChannelsTest()
151 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReduceMinChannelsTest()
Dpooling_test_util.cc46 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AveragePoolingTest()
47 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AveragePoolingTest()
77 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in AveragePoolingNonEmptyPaddingTest()
78 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in AveragePoolingNonEmptyPaddingTest()
109 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in MaxPoolingTest()
110 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in MaxPoolingTest()
141 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in MaxPoolingIndicesTest()
142 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in MaxPoolingIndicesTest()
143 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in MaxPoolingIndicesTest()
Dquantize_and_dequantize_test_util.cc48 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim2Bits8Test()
49 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim2Bits8Test()
85 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
86 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
121 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim3Bits16Test()
122 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim3Bits16Test()
158 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim2Bits16_NegativeRangeTest()
159 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in QuantAndDequant_Dim2Bits16_NegativeRangeTest()
Dspace_to_depth_test_util.cc39 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
40 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
66 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
67 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
95 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()
96 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()
126 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x4x4x1BlockSize2Test()
127 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SpaceToDepthTensorShape1x4x4x1BlockSize2Test()
Dsoftmax_test_util.cc41 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SoftmaxTest()
42 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SoftmaxTest()
81 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in SoftmaxBigNumberTest()
82 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in SoftmaxBigNumberTest()
111 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in Softmax1x1Test()
112 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in Softmax1x1Test()
150 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in Softmax1x1BigNumberTest()
151 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in Softmax1x1BigNumberTest()
Dconv_powervr_test_util.cc49 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVR1x1SimpleWeightsTest()
50 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVR1x1SimpleWeightsTest()
86 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVR1x1Test()
87 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVR1x1Test()
123 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVRSimpleWeightsTest()
124 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVRSimpleWeightsTest()
160 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVRTest()
161 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConvPowerVRTest()
Dfully_connected_test_util.cc46 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in FullyConnectedTest()
47 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in FullyConnectedTest()
92 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in FullyConnectedLargeTest()
93 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in FullyConnectedLargeTest()
149 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in FullyConnectedExtraLargeTest()
150 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in FullyConnectedExtraLargeTest()
Ddepthwise_conv_test_util.cc49 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in DepthwiseConvSimpleWeightsTest()
50 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in DepthwiseConvSimpleWeightsTest()
86 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in DepthwiseConvNoMultiplierTest()
87 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in DepthwiseConvNoMultiplierTest()
124 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in DepthwiseConvMultiplier2Test()
125 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in DepthwiseConvMultiplier2Test()
Dreshape_test_util.cc40 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ReshapeTest()
41 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ReshapeTest()
66 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in Reshapex4Test()
67 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in Reshapex4Test()
Ddepthwise_conv_3x3_test_util.cc50 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in DepthwiseConv3x3SimpleWeightsTest()
51 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in DepthwiseConv3x3SimpleWeightsTest()
88 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in DepthwiseConv3x3Test()
89 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in DepthwiseConv3x3Test()
Dconvolution_transposed_test_util.cc50 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConvolutionTransposedSimpleWeightsTest()
51 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConvolutionTransposedSimpleWeightsTest()
90 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in ConvolutionTransposedTest()
91 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in ConvolutionTransposedTest()
Dmax_unpooling_test_util.cc48 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in MaxUnpoolingTest()
49 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in MaxUnpoolingTest()
50 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in MaxUnpoolingTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconvolution_transposed_thin_test.cc55 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()
56 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()
95 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()
96 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()
Dconvolution_transposed_3x3_thin_test.cc54 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()
55 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()
92 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()
93 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()
Dconv_buffer_1x1_test.cc55 {data_type, TensorStorageType::BUFFER, Layout::HWC}); in TEST_F()
57 {data_type, TensorStorageType::BUFFER, Layout::HWC}); in TEST_F()
92 {data_type, TensorStorageType::BUFFER, Layout::HWC}); in TEST_F()
94 {data_type, TensorStorageType::BUFFER, Layout::HWC}); in TEST_F()
Dconv_constants_test.cc55 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()
56 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()
92 op_def.src_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()
93 op_def.dst_tensors.push_back({data_type, storage, Layout::HWC}); in TEST_F()

1234