Home
last modified time | relevance | path

Searched refs:dst_tensor (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Delementwise_test_util.cc40 TensorFloat32 dst_tensor; in AbsTest() local
45 BHWC(1, 2, 1, 2), &dst_tensor)); in AbsTest()
48 dst_tensor.data, 0.0f)); in AbsTest()
67 TensorFloat32 dst_tensor; in CosTest() local
72 BHWC(1, 2, 1, 2), &dst_tensor)); in CosTest()
75 dst_tensor.data, eps)); in CosTest()
93 TensorFloat32 dst_tensor; in CopyTest() local
98 BHWC(1, 2, 1, 2), &dst_tensor)); in CopyTest()
99 RETURN_IF_ERROR(PointWiseNear(src_tensor.data, dst_tensor.data, 0.0f)); in CopyTest()
118 TensorFloat32 dst_tensor; in EluTest() local
[all …]
Dpadding_test_util.cc45 TensorFloat32 dst_tensor; in PaddingAppendWidthTest() local
49 BHWC(1, 2, 2, 2), &dst_tensor)); in PaddingAppendWidthTest()
52 dst_tensor.data, eps)); in PaddingAppendWidthTest()
75 TensorFloat32 dst_tensor; in PaddingPrependWidthTest() local
79 BHWC(1, 2, 2, 2), &dst_tensor)); in PaddingPrependWidthTest()
82 dst_tensor.data, eps)); in PaddingPrependWidthTest()
105 TensorFloat32 dst_tensor; in PaddingAppendHeightTest() local
109 BHWC(1, 3, 1, 2), &dst_tensor)); in PaddingAppendHeightTest()
111 dst_tensor.data, eps)); in PaddingAppendHeightTest()
134 TensorFloat32 dst_tensor; in PaddingPrependHeightTest() local
[all …]
Dresize_test_util.cc46 TensorFloat32 dst_tensor; in ResizeBilinearAlignedTest() local
50 BHWC(1, 4, 4, 1), &dst_tensor)); in ResizeBilinearAlignedTest()
54 dst_tensor.data, eps)); in ResizeBilinearAlignedTest()
78 TensorFloat32 dst_tensor; in ResizeBilinearNonAlignedTest() local
82 BHWC(1, 4, 4, 1), &dst_tensor)); in ResizeBilinearNonAlignedTest()
86 dst_tensor.data, eps)); in ResizeBilinearNonAlignedTest()
111 TensorFloat32 dst_tensor; in ResizeBilinearWithoutHalfPixelTest() local
115 BHWC(1, 3, 3, 1), &dst_tensor)); in ResizeBilinearWithoutHalfPixelTest()
118 dst_tensor.data, eps)); in ResizeBilinearWithoutHalfPixelTest()
143 TensorFloat32 dst_tensor; in ResizeBilinearWithHalfPixelTest() local
[all …]
Drelu_test_util.cc45 TensorFloat32 dst_tensor; in ReLUNoClipNoAlphaTest() local
49 BHWC(1, 2, 1, 2), &dst_tensor)); in ReLUNoClipNoAlphaTest()
51 PointWiseNear({0.0f, 0.8f, 0.0f, 3.2f}, dst_tensor.data, eps)); in ReLUNoClipNoAlphaTest()
74 TensorFloat32 dst_tensor; in ReLUClipTest() local
78 BHWC(1, 2, 1, 2), &dst_tensor)); in ReLUClipTest()
80 PointWiseNear({0.0f, 0.8f, 0.0f, 0.9f}, dst_tensor.data, eps)); in ReLUClipTest()
103 TensorFloat32 dst_tensor; in ReLUAlphaTest() local
107 BHWC(1, 2, 1, 2), &dst_tensor)); in ReLUAlphaTest()
109 PointWiseNear({-0.25f, 0.8f, -0.3f, 3.2f}, dst_tensor.data, eps)); in ReLUAlphaTest()
132 TensorFloat32 dst_tensor; in ReLUAlphaClipTest() local
[all …]
Dreduce_test_util.cc42 TensorFloat32 dst_tensor; in MeanHWTest() local
48 BHWC(1, 1, 1, 1), &dst_tensor)); in MeanHWTest()
49 RETURN_IF_ERROR(PointWiseNear({2.5f}, dst_tensor.data, eps)); in MeanHWTest()
69 TensorFloat32 dst_tensor; in ReduceSumChannelsTest() local
75 BHWC(1, 2, 1, 1), &dst_tensor)); in ReduceSumChannelsTest()
76 RETURN_IF_ERROR(PointWiseNear({5.4f, 12.6f}, dst_tensor.data, eps)); in ReduceSumChannelsTest()
96 TensorFloat32 dst_tensor; in ReduceProductChannelsTest() local
102 BHWC(1, 2, 1, 1), &dst_tensor)); in ReduceProductChannelsTest()
103 RETURN_IF_ERROR(PointWiseNear({2.2f, 12.4f}, dst_tensor.data, eps)); in ReduceProductChannelsTest()
124 TensorFloat32 dst_tensor; in ReduceMaxChannelsTest() local
[all …]
Dquantize_and_dequantize_test_util.cc50 TensorFloat32 dst_tensor; in QuantAndDequant_Dim2Bits8Test() local
54 BHWC(1, 3, 2, 1), &dst_tensor)); in QuantAndDequant_Dim2Bits8Test()
57 dst_tensor.data, eps)); in QuantAndDequant_Dim2Bits8Test()
87 TensorFloat32 dst_tensor; in QuantAndDequant_Dim3Bits8_NegativeRangeTest() local
91 BHWC(1, 3, 1, 2), &dst_tensor)); in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
94 dst_tensor.data, eps)); in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
123 TensorFloat32 dst_tensor; in QuantAndDequant_Dim3Bits16Test() local
127 BHWC(1, 3, 1, 2), &dst_tensor)); in QuantAndDequant_Dim3Bits16Test()
130 dst_tensor.data, eps)); in QuantAndDequant_Dim3Bits16Test()
160 TensorFloat32 dst_tensor; in QuantAndDequant_Dim2Bits16_NegativeRangeTest() local
[all …]
Dspace_to_depth_test_util.cc41 TensorFloat32 dst_tensor; in SpaceToDepthTensorShape1x2x2x1BlockSize2Test() local
45 BHWC(1, 1, 1, 4), &dst_tensor)); in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
48 dst_tensor.data, 0.0f)); in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
68 TensorFloat32 dst_tensor; in SpaceToDepthTensorShape1x2x2x2BlockSize2Test() local
72 BHWC(1, 1, 1, 8), &dst_tensor)); in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
76 dst_tensor.data, 0.0f)); in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
97 TensorFloat32 dst_tensor; in SpaceToDepthTensorShape1x2x2x3BlockSize2Test() local
101 BHWC(1, 1, 1, 12), &dst_tensor)); in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()
106 dst_tensor.data, 0.0f)); in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()
128 TensorFloat32 dst_tensor; in SpaceToDepthTensorShape1x4x4x1BlockSize2Test() local
[all …]
Dpooling_test_util.cc48 TensorFloat32 dst_tensor; in AveragePoolingTest() local
52 BHWC(1, 1, 1, 2), &dst_tensor)); in AveragePoolingTest()
53 RETURN_IF_ERROR(PointWiseNear({3.0f, 4.0f}, dst_tensor.data, eps)); in AveragePoolingTest()
79 TensorFloat32 dst_tensor; in AveragePoolingNonEmptyPaddingTest() local
83 BHWC(1, 2, 2, 1), &dst_tensor)); in AveragePoolingNonEmptyPaddingTest()
85 PointWiseNear({1.5f, 2.0f, 2.5f, 3.0f}, dst_tensor.data, eps)); in AveragePoolingNonEmptyPaddingTest()
111 TensorFloat32 dst_tensor; in MaxPoolingTest() local
115 BHWC(1, 1, 1, 2), &dst_tensor)); in MaxPoolingTest()
116 RETURN_IF_ERROR(PointWiseNear({8.0f, 7.0f}, dst_tensor.data, eps)); in MaxPoolingTest()
144 TensorFloat32 dst_tensor; in MaxPoolingIndicesTest() local
[all …]
Dsoftmax_test_util.cc43 TensorFloat32 dst_tensor; in SoftmaxTest() local
47 BHWC(1, 2, 1, 2), &dst_tensor)); in SoftmaxTest()
50 dst_tensor.data, eps)); in SoftmaxTest()
83 TensorFloat32 dst_tensor; in SoftmaxBigNumberTest() local
87 BHWC(1, 2, 1, 2), &dst_tensor)); in SoftmaxBigNumberTest()
93 dst_tensor.data, eps)); in SoftmaxBigNumberTest()
113 TensorFloat32 dst_tensor; in Softmax1x1Test() local
117 BHWC(1, 1, 1, 4), &dst_tensor)); in Softmax1x1Test()
119 PointWiseNear({0.1f, 0.2f, 0.3f, 0.4f}, dst_tensor.data, eps)); in Softmax1x1Test()
152 TensorFloat32 dst_tensor; in Softmax1x1BigNumberTest() local
[all …]
Dconv_powervr_test_util.cc51 TensorFloat32 dst_tensor; in ConvPowerVR1x1SimpleWeightsTest() local
56 BHWC(1, 2, 2, 2), &dst_tensor)); in ConvPowerVR1x1SimpleWeightsTest()
59 dst_tensor.data, eps)); in ConvPowerVR1x1SimpleWeightsTest()
88 TensorFloat32 dst_tensor; in ConvPowerVR1x1Test() local
93 BHWC(1, 2, 2, 2), &dst_tensor)); in ConvPowerVR1x1Test()
96 dst_tensor.data, eps)); in ConvPowerVR1x1Test()
125 TensorFloat32 dst_tensor; in ConvPowerVRSimpleWeightsTest() local
130 BHWC(1, 2, 2, 1), &dst_tensor)); in ConvPowerVRSimpleWeightsTest()
132 PointWiseNear({28.0f, 18.0f, 22.0f, 13.0f}, dst_tensor.data, eps)); in ConvPowerVRSimpleWeightsTest()
162 TensorFloat32 dst_tensor; in ConvPowerVRTest() local
[all …]
Dadd_test_util.cc43 TensorFloat32 dst_tensor; in AddTwoEqualTensorsTest() local
47 BHWC(1, 2, 1, 2), &dst_tensor)); in AddTwoEqualTensorsTest()
49 PointWiseNear({0.0f, 0.0f, -0.1f, 0.0f}, dst_tensor.data, eps)); in AddTwoEqualTensorsTest()
73 TensorFloat32 dst_tensor; in AddFirstTensorHasMoreChannelsThanSecondTest() local
77 BHWC(1, 2, 1, 6), &dst_tensor)); in AddFirstTensorHasMoreChannelsThanSecondTest()
80 dst_tensor.data, eps)); in AddFirstTensorHasMoreChannelsThanSecondTest()
104 TensorFloat32 dst_tensor; in AddFirstTensorHasLessChannelsThanSecond() local
108 BHWC(1, 2, 1, 6), &dst_tensor)); in AddFirstTensorHasLessChannelsThanSecond()
111 dst_tensor.data, eps)); in AddFirstTensorHasLessChannelsThanSecond()
Dprelu_test_util.cc48 TensorFloat32 dst_tensor; in PReLUAlphaTest() local
52 BHWC(1, 2, 1, 2), &dst_tensor)); in PReLUAlphaTest()
54 PointWiseNear({0.0f, 2.0f, -1.0f, 3.0f}, dst_tensor.data, eps)); in PReLUAlphaTest()
80 TensorFloat32 dst_tensor; in PReLUAlphaClipTest() local
84 BHWC(1, 2, 1, 2), &dst_tensor)); in PReLUAlphaClipTest()
86 PointWiseNear({0.0f, 2.0f, -1.0f, 0.7f}, dst_tensor.data, eps)); in PReLUAlphaClipTest()
112 TensorFloat32 dst_tensor; in PReLUHWCAlphaTest() local
116 BHWC(1, 2, 1, 2), &dst_tensor)); in PReLUHWCAlphaTest()
118 PointWiseNear({0.0f, 2.0f, -1.4f, 3.0f}, dst_tensor.data, eps)); in PReLUHWCAlphaTest()
Dconcat_test_util.cc48 TensorFloat32 dst_tensor; in ConcatWidthTest() local
52 BHWC(1, 2, 3, 2), &dst_tensor)); in ConcatWidthTest()
57 dst_tensor.data, 0.0f)); in ConcatWidthTest()
81 TensorFloat32 dst_tensor; in ConcatHeightTest() local
85 BHWC(1, 3, 1, 2), &dst_tensor)); in ConcatHeightTest()
88 dst_tensor.data, 0.0f)); in ConcatHeightTest()
116 TensorFloat32 dst_tensor; in ConcatChannelsTest() local
122 BHWC(1, 2, 1, 6), &dst_tensor)); in ConcatChannelsTest()
127 dst_tensor.data, 0.0f)); in ConcatChannelsTest()
153 TensorFloat32 dst_tensor; in ConcatChannelsAlignedx4Test() local
[all …]
Ddepthwise_conv_test_util.cc51 TensorFloat32 dst_tensor; in DepthwiseConvSimpleWeightsTest() local
56 BHWC(1, 2, 2, 2), &dst_tensor)); in DepthwiseConvSimpleWeightsTest()
59 dst_tensor.data, eps)); in DepthwiseConvSimpleWeightsTest()
88 TensorFloat32 dst_tensor; in DepthwiseConvNoMultiplierTest() local
93 BHWC(1, 2, 2, 2), &dst_tensor)); in DepthwiseConvNoMultiplierTest()
96 dst_tensor.data, eps)); in DepthwiseConvNoMultiplierTest()
126 TensorFloat32 dst_tensor; in DepthwiseConvMultiplier2Test() local
131 BHWC(1, 2, 2, 4), &dst_tensor)); in DepthwiseConvMultiplier2Test()
135 dst_tensor.data, eps)); in DepthwiseConvMultiplier2Test()
Dfully_connected_test_util.cc48 TensorFloat32 dst_tensor; in FullyConnectedTest() local
53 BHWC(1, 1, 1, 2), &dst_tensor)); in FullyConnectedTest()
54 RETURN_IF_ERROR(PointWiseNear({14.5f, 37.5f}, dst_tensor.data, eps)) in FullyConnectedTest()
94 TensorFloat32 dst_tensor; in FullyConnectedLargeTest() local
99 BHWC(1, 1, 1, 12), &dst_tensor)); in FullyConnectedLargeTest()
103 dst_tensor.data, eps)) in FullyConnectedLargeTest()
151 TensorFloat32 dst_tensor; in FullyConnectedExtraLargeTest() local
156 BHWC(1, 1, 1, kOutputSize), &dst_tensor)); in FullyConnectedExtraLargeTest()
157 RETURN_IF_ERROR(PointWiseNear(expected, dst_tensor.data, eps)) in FullyConnectedExtraLargeTest()
Dmean_stddev_normalization_test_util.cc44 TensorFloat32 dst_tensor; in MeanStddevNormSeparateBatchesTest() local
50 BHWC(1, 1, 1, 4), &dst_tensor)); in MeanStddevNormSeparateBatchesTest()
61 PointWiseNear(expected_output, dst_tensor.data, tolerance)); in MeanStddevNormSeparateBatchesTest()
91 TensorFloat32 dst_tensor; in MeanStddevNormalizationAllBatchesTest() local
97 BHWC(9, 1, 1, 4), &dst_tensor)); in MeanStddevNormalizationAllBatchesTest()
112 RETURN_IF_ERROR(PointWiseNear(expected_output, dst_tensor.data, eps)) in MeanStddevNormalizationAllBatchesTest()
146 TensorFloat32 dst_tensor; in MeanStddevNormalizationLargeVectorTest() local
152 BHWC(1, 1, 1, kVectorSize), &dst_tensor)); in MeanStddevNormalizationLargeVectorTest()
165 RETURN_IF_ERROR(PointWiseNear(expected_output, dst_tensor.data, eps)) in MeanStddevNormalizationLargeVectorTest()
Dreshape_test_util.cc42 TensorFloat32 dst_tensor; in ReshapeTest() local
46 BHWC(1, 3, 1, 2), &dst_tensor)); in ReshapeTest()
49 dst_tensor.data, 0.0)); in ReshapeTest()
68 TensorFloat32 dst_tensor; in Reshapex4Test() local
72 BHWC(1, 1, 2, 4), &dst_tensor)); in Reshapex4Test()
76 dst_tensor.data, 0.0f)); in Reshapex4Test()
Dconvolution_transposed_test_util.cc52 TensorFloat32 dst_tensor; in ConvolutionTransposedSimpleWeightsTest() local
58 BHWC(1, 4, 4, 2), &dst_tensor)); in ConvolutionTransposedSimpleWeightsTest()
64 dst_tensor.data, eps)); in ConvolutionTransposedSimpleWeightsTest()
92 TensorFloat32 dst_tensor; in ConvolutionTransposedTest() local
98 BHWC(1, 4, 4, 1), &dst_tensor)); in ConvolutionTransposedTest()
102 dst_tensor.data, eps)); in ConvolutionTransposedTest()
Ddepthwise_conv_3x3_test_util.cc52 TensorFloat32 dst_tensor; in DepthwiseConv3x3SimpleWeightsTest() local
57 BHWC(1, 2, 2, 2), &dst_tensor)); in DepthwiseConv3x3SimpleWeightsTest()
60 dst_tensor.data, eps)); in DepthwiseConv3x3SimpleWeightsTest()
90 TensorFloat32 dst_tensor; in DepthwiseConv3x3Test() local
95 BHWC(1, 2, 2, 2), &dst_tensor)); in DepthwiseConv3x3Test()
98 dst_tensor.data, eps)); in DepthwiseConv3x3Test()
Dwinograd_test_util.cc79 TensorFloat32 dst_tensor; in Winograd4x4To36TileX6Test() local
88 BHWC(1, 36, 1, 1), &dst_tensor)); in Winograd4x4To36TileX6Test()
89 RETURN_IF_ERROR(PointWiseNear(dst_ref.data, dst_tensor.data, eps)); in Winograd4x4To36TileX6Test()
151 TensorFloat32 dst_tensor; in Winograd36To4x4Tile4x1Test() local
157 BHWC(1, 4, 4, 1), &dst_tensor)); in Winograd36To4x4Tile4x1Test()
158 RETURN_IF_ERROR(PointWiseNear(dst_ref.data, dst_tensor.data, eps)); in Winograd36To4x4Tile4x1Test()
215 TensorFloat32 dst_tensor; in Winograd4x4To36Test() local
222 BHWC(1, 36, 1, 1), &dst_tensor)); in Winograd4x4To36Test()
223 RETURN_IF_ERROR(PointWiseNear(dst_ref.data, dst_tensor.data, eps)); in Winograd4x4To36Test()
286 TensorFloat32 dst_tensor; in Winograd36To4x4Test() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconvolution_transposed_3x3_thin_test.cc56 TensorFloat32 dst_tensor; in TEST_F() local
63 BHWC(1, 4, 4, 1), &dst_tensor)); in TEST_F()
64 EXPECT_THAT(dst_tensor.data, in TEST_F()
94 TensorFloat32 dst_tensor; in TEST_F() local
101 BHWC(1, 4, 4, 1), &dst_tensor)); in TEST_F()
103 dst_tensor.data, in TEST_F()
Dconvolution_transposed_thin_test.cc57 TensorFloat32 dst_tensor; in TEST_F() local
63 BHWC(1, 4, 4, 2), &dst_tensor)); in TEST_F()
65 dst_tensor.data, in TEST_F()
97 TensorFloat32 dst_tensor; in TEST_F() local
103 BHWC(1, 4, 4, 1), &dst_tensor)); in TEST_F()
105 dst_tensor.data, in TEST_F()
Dconv_buffer_1x1_test.cc58 TensorFloat32 dst_tensor; in TEST_F() local
64 BHWC(1, 2, 1, 2), &dst_tensor)); in TEST_F()
65 EXPECT_THAT(dst_tensor.data, in TEST_F()
95 TensorFloat32 dst_tensor; in TEST_F() local
101 BHWC(1, 2, 1, 4), &dst_tensor)); in TEST_F()
102 EXPECT_THAT(dst_tensor.data, in TEST_F()
Dconv_constants_test.cc57 TensorFloat32 dst_tensor; in TEST_F() local
63 BHWC(1, 2, 2, 1), &dst_tensor)); in TEST_F()
64 EXPECT_THAT(dst_tensor.data, in TEST_F()
94 TensorFloat32 dst_tensor; in TEST_F() local
100 BHWC(1, 2, 2, 2), &dst_tensor)); in TEST_F()
101 EXPECT_THAT(dst_tensor.data, in TEST_F()
/external/tensorflow/tensorflow/lite/kernels/
Dwhile.cc56 TfLiteTensor* dst_tensor = dst_subgraph->tensor(dst_tensor_indices[i]); in CopyTensorsShapeAndType() local
63 context, context->ResizeTensor(context, dst_tensor, in CopyTensorsShapeAndType()
66 dst_tensor->type = src_tensor->type; in CopyTensorsShapeAndType()
83 TfLiteTensor* dst_tensor = dst_subgraph->tensor(dst_tensor_indices[i]); in CopyTensorsData() local
84 if (IsDynamicTensor(dst_tensor)) { in CopyTensorsData()
85 TfLiteTensorRealloc(src_tensor->bytes, dst_tensor); in CopyTensorsData()
87 TF_LITE_ENSURE_EQ(context, src_tensor->bytes, dst_tensor->bytes); in CopyTensorsData()
88 memcpy(dst_tensor->data.raw, src_tensor->data.raw, src_tensor->bytes); in CopyTensorsData()

12