Home
last modified time | relevance | path

Searched defs:dst_tensor (Results 1 – 25 of 62) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Delementwise_test_util.cc41 TensorFloat32 dst_tensor; in AbsTest() local
68 TensorFloat32 dst_tensor; in CosTest() local
94 TensorFloat32 dst_tensor; in CopyTest() local
119 TensorFloat32 dst_tensor; in EluTest() local
147 TensorFloat32 dst_tensor; in ExpTest() local
175 TensorFloat32 dst_tensor; in FloorTest() local
205 TensorFloat32 dst_tensor; in FloorDivTest() local
239 TensorFloat32 dst_tensor; in FloorModTest() local
272 TensorFloat32 dst_tensor; in HardSwishTest() local
299 TensorFloat32 dst_tensor; in LogTest() local
[all …]
H A Dpadding_test_util.cc46 TensorFloat32 dst_tensor; in PaddingAppendWidthTest() local
76 TensorFloat32 dst_tensor; in PaddingPrependWidthTest() local
106 TensorFloat32 dst_tensor; in PaddingAppendHeightTest() local
135 TensorFloat32 dst_tensor; in PaddingPrependHeightTest() local
164 TensorFloat32 dst_tensor; in PaddingAppendChannelsTest() local
193 TensorFloat32 dst_tensor; in PaddingPrependChannelsTest() local
222 TensorFloat32 dst_tensor; in PaddingPrependChannelsX4Test() local
251 TensorFloat32 dst_tensor; in PaddingComplexTest() local
284 TensorFloat32 dst_tensor; in PaddingReflectWidthTest() local
314 TensorFloat32 dst_tensor; in PaddingReflectChannelsTest() local
H A Dresize_test_util.cc47 TensorFloat32 dst_tensor; in ResizeBilinearAlignedTest() local
79 TensorFloat32 dst_tensor; in ResizeBilinearNonAlignedTest() local
112 TensorFloat32 dst_tensor; in ResizeBilinearWithoutHalfPixelTest() local
144 TensorFloat32 dst_tensor; in ResizeBilinearWithHalfPixelTest() local
176 TensorFloat32 dst_tensor; in ResizeNearestTest() local
208 TensorFloat32 dst_tensor; in ResizeNearestAlignCornersTest() local
240 TensorFloat32 dst_tensor; in ResizeNearestHalfPixelCentersTest() local
H A Dreduce_test_util.cc54 tflite::gpu::Tensor<BHWC, T> dst_tensor; in ReduceSumChannelsIntTest() local
95 tflite::gpu::Tensor<BHWC, T> dst_tensor; in ReduceProductChannelsUIntTest() local
126 TensorFloat32 dst_tensor; in MeanHWTest() local
153 TensorFloat32 dst_tensor; in ReduceSumChannelsTest() local
184 TensorFloat32 dst_tensor; in ReduceProductChannelsTest() local
216 TensorFloat32 dst_tensor; in ReduceMaxChannelsTest() local
244 TensorFloat32 dst_tensor; in ReduceMinChannelsTest() local
H A Dselect_v2_test_util.cc115 bool broadcast_false, TensorFloat32& dst_tensor) { in RunSelectV2()
158 TensorFloat32 dst_tensor; in SelectV2TestTemplate() local
186 TensorFloat32 dst_tensor; in SelectV2BatchTestTemplate() local
215 TensorFloat32 dst_tensor; in SelectV2BroadcastFalseTestTemplate() local
245 TensorFloat32 dst_tensor; in SelectV2BroadcastTrueTestTemplate() local
276 TensorFloat32 dst_tensor; in SelectV2BroadcastBothTestTemplate() local
304 TensorFloat32 dst_tensor; in SelectV2ChannelsTestTemplate() local
333 TensorFloat32 dst_tensor; in SelectV2ChannelsBatchTestTemplate() local
364 TensorFloat32 dst_tensor; in SelectV2ChannelsBroadcastFalseTestTemplate() local
H A Dtile_test_util.cc41 TensorFloat32 dst_tensor; in TileChannelsTest() local
68 TensorFloat32 dst_tensor; in TileChannelsX4Test() local
96 TensorFloat32 dst_tensor; in TileWidthTest() local
123 TensorFloat32 dst_tensor; in TileHeightTest() local
151 TensorFloat32 dst_tensor; in TileHWCTest() local
H A Dconv_generic_test_util.cc52 TensorFloat32 dst_tensor; in ConvGeneric1x1SimpleWeightsTest() local
89 TensorFloat32 dst_tensor; in ConvGeneric1x1Test() local
126 TensorFloat32 dst_tensor; in ConvGenericSimpleWeightsTest() local
163 TensorFloat32 dst_tensor; in ConvGenericTest() local
204 TensorFloat32 dst_tensor; in ConvGenericGroupedTest() local
H A Dpooling_test_util.cc49 TensorFloat32 dst_tensor; in AveragePoolingTest() local
80 TensorFloat32 dst_tensor; in AveragePoolingNonEmptyPaddingTest() local
112 TensorFloat32 dst_tensor; in MaxPoolingTest() local
145 TensorFloat32 dst_tensor; in MaxPoolingIndicesTest() local
181 TensorFloat32 dst_tensor; in MaxPoolingIndicesTest() local
H A Dadd_test_util.cc56 tflite::gpu::Tensor<BHWC, T> dst_tensor; in AddTwoEqualIntTensorsTest() local
100 tflite::gpu::Tensor<BHWC, T> dst_tensor; in AddTwoEqualUintTensorsTest() local
133 TensorFloat32 dst_tensor; in AddTwoEqualTensorsTest() local
170 TensorFloat32 dst_tensor; in AddFirstTensorHasMoreChannelsThanSecondTest() local
201 TensorFloat32 dst_tensor; in AddFirstTensorHasLessChannelsThanSecond() local
H A Dspace_to_depth_test_util.cc42 TensorFloat32 dst_tensor; in SpaceToDepthTensorShape1x2x2x1BlockSize2Test() local
69 TensorFloat32 dst_tensor; in SpaceToDepthTensorShape1x2x2x2BlockSize2Test() local
98 TensorFloat32 dst_tensor; in SpaceToDepthTensorShape1x2x2x3BlockSize2Test() local
129 TensorFloat32 dst_tensor; in SpaceToDepthTensorShape1x4x4x1BlockSize2Test() local
H A Dconcat_test_util.cc49 TensorFloat32 dst_tensor; in ConcatWidthTest() local
82 TensorFloat32 dst_tensor; in ConcatHeightTest() local
117 TensorFloat32 dst_tensor; in ConcatChannelsTest() local
154 TensorFloat32 dst_tensor; in ConcatChannelsAlignedx4Test() local
H A Drelu_test_util.cc46 TensorFloat32 dst_tensor; in ReLUNoClipNoAlphaTest() local
75 TensorFloat32 dst_tensor; in ReLUClipTest() local
104 TensorFloat32 dst_tensor; in ReLUAlphaTest() local
133 TensorFloat32 dst_tensor; in ReLUAlphaClipTest() local
H A Dsoftmax_test_util.cc44 TensorFloat32 dst_tensor; in SoftmaxTest() local
84 TensorFloat32 dst_tensor; in SoftmaxBigNumberTest() local
114 TensorFloat32 dst_tensor; in Softmax1x1Test() local
153 TensorFloat32 dst_tensor; in Softmax1x1BigNumberTest() local
H A Dfully_connected_test_util.cc49 TensorFloat32 dst_tensor; in FullyConnectedTest() local
95 TensorFloat32 dst_tensor; in FullyConnectedLargeTest() local
156 TensorFloat32 dst_tensor; in FullyConnectedExtraLargeTest() local
191 TensorFloat32 dst_tensor; in FullyConnectedInt8Test() local
H A Dquantize_and_dequantize_test_util.cc51 TensorFloat32 dst_tensor; in QuantAndDequant_Dim2Bits8Test() local
88 TensorFloat32 dst_tensor; in QuantAndDequant_Dim3Bits8_NegativeRangeTest() local
124 TensorFloat32 dst_tensor; in QuantAndDequant_Dim3Bits16Test() local
161 TensorFloat32 dst_tensor; in QuantAndDequant_Dim2Bits16_NegativeRangeTest() local
H A Dwinograd_test_util.cc80 TensorFloat32 dst_tensor; in Winograd4x4To36TileX6Test() local
152 TensorFloat32 dst_tensor; in Winograd36To4x4Tile4x1Test() local
215 TensorFloat32 dst_tensor; in Winograd4x4To36Test() local
286 TensorFloat32 dst_tensor; in Winograd36To4x4Test() local
H A Dcast_test_util.cc52 tflite::gpu::Tensor<BHWC, DataType::INT32> dst_tensor; in CastTests() local
87 tflite::gpu::Tensor<BHWC, DataType::BOOL> dst_tensor; in CastToBoolTests() local
122 tflite::gpu::Tensor<BHWC, DataType::FLOAT32> dst_tensor; in CastFromBoolTests() local
H A Dtranspose_test_util.cc55 tflite::gpu::Tensor<BHWC, T> dst_tensor; in TransposeIntTest() local
97 tflite::gpu::Tensor<BHWC, T> dst_tensor; in TransposeUintTest() local
131 TensorFloat32 dst_tensor; in TransposeTest() local
H A Ddepthwise_conv_test_util.cc52 TensorFloat32 dst_tensor; in DepthwiseConvSimpleWeightsTest() local
89 TensorFloat32 dst_tensor; in DepthwiseConvNoMultiplierTest() local
127 TensorFloat32 dst_tensor; in DepthwiseConvMultiplier2Test() local
H A Dmean_stddev_normalization_test_util.cc47 TensorFloat32 dst_tensor; in MeanStddevNormSeparateBatchesTest() local
108 TensorFloat32 dst_tensor; in MeanStddevNormalizationAllBatchesTest() local
179 TensorFloat32 dst_tensor; in MeanStddevNormalizationLargeVectorTest() local
H A Dreshape_test_util.cc43 TensorFloat32 dst_tensor; in ReshapeTest() local
69 TensorFloat32 dst_tensor; in Reshapex4Test() local
H A Dconv_buffer_1x1_test_util.cc52 TensorFloat32 dst_tensor; in ConvBuffer1x1SimpleWeightsTest() local
89 TensorFloat32 dst_tensor; in ConvBuffer1x1Test() local
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/utils/
H A DDependencyGraph.h155 for(auto dst_tensor : outputs) variable
335 const auto dst_tensor = _adj_dst_ops.find(src_tensor.first); in intermediate_tensors() local
405 for(TensorId dst_tensor : _adj_dst_tensors.at(op)) in dst_ops() local
458 for(auto dst_tensor : _adj_dst_tensors.at(op)) in remove_operator() local
622 for(auto dst_tensor : dst_tensors) in build_operators_sequence_from_op() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dwhile.cc69 TfLiteTensor* dst_tensor = dst_subgraph->tensor(dst_tensor_indices[i]); in CopyTensorsShapeAndType() local
99 TfLiteTensor* dst_tensor = dst_subgraph->tensor(dst_tensor_indices[i]); in CopyTensorsData() local
159 TfLiteTensor* dst_tensor = dst_subgraph->tensor(dst_tensor_indices[i]); in ShallowCopyTensorsShapeTypeData() local
181 TfLiteTensor* dst_tensor = dst_subgraph->tensor(dst_tensor_indices[i]); in ShallowCopyTensorsShapeTypeData() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_relu_op.cc453 Tensor* dst_tensor = nullptr; in Compute() local
780 Tensor* dst_tensor = nullptr; in Compute_Scalar() local
847 Tensor* dst_tensor = nullptr; in Compute_Scalar() local
928 Tensor* dst_tensor = nullptr; in Compute_Scalar() local
1013 Tensor* dst_tensor = nullptr; in Compute_Scalar() local
1089 Tensor* dst_tensor = nullptr; in Compute_Scalar() local

123