Home
last modified time | relevance | path

Searched defs:src_tensor (Results 1 – 25 of 71) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dpadding_test_util.cc30 TensorFloat32 src_tensor; in PaddingAppendWidthTest() local
60 TensorFloat32 src_tensor; in PaddingPrependWidthTest() local
90 TensorFloat32 src_tensor; in PaddingAppendHeightTest() local
119 TensorFloat32 src_tensor; in PaddingPrependHeightTest() local
148 TensorFloat32 src_tensor; in PaddingAppendChannelsTest() local
177 TensorFloat32 src_tensor; in PaddingPrependChannelsTest() local
206 TensorFloat32 src_tensor; in PaddingPrependChannelsX4Test() local
235 TensorFloat32 src_tensor; in PaddingComplexTest() local
267 TensorFloat32 src_tensor; in PaddingReflectWidthTest() local
297 TensorFloat32 src_tensor; in PaddingReflectChannelsTest() local
H A Delementwise_test_util.cc30 TensorFloat32 src_tensor; in AbsTest() local
56 TensorFloat32 src_tensor; in CosTest() local
83 TensorFloat32 src_tensor; in CopyTest() local
107 TensorFloat32 src_tensor; in EluTest() local
135 TensorFloat32 src_tensor; in ExpTest() local
163 TensorFloat32 src_tensor; in FloorTest() local
189 TensorFloat32 src_tensor; in FloorDivTest() local
223 TensorFloat32 src_tensor; in FloorModTest() local
260 TensorFloat32 src_tensor; in HardSwishTest() local
287 TensorFloat32 src_tensor; in LogTest() local
[all …]
H A Dresize_test_util.cc30 TensorFloat32 src_tensor; in ResizeBilinearAlignedTest() local
62 TensorFloat32 src_tensor; in ResizeBilinearNonAlignedTest() local
94 TensorFloat32 src_tensor; in ResizeBilinearWithoutHalfPixelTest() local
126 TensorFloat32 src_tensor; in ResizeBilinearWithHalfPixelTest() local
158 TensorFloat32 src_tensor; in ResizeNearestTest() local
190 TensorFloat32 src_tensor; in ResizeNearestAlignCornersTest() local
222 TensorFloat32 src_tensor; in ResizeNearestHalfPixelCentersTest() local
H A Dsplit_test_util.cc30 TensorFloat32 src_tensor; in SplitChannelsTest() local
73 TensorFloat32 src_tensor; in SplitChannelsX4Test() local
118 TensorFloat32 src_tensor; in SplitWidthTest() local
161 TensorFloat32 src_tensor; in SplitHeightTest() local
203 TensorFloat32 src_tensor; in SplitBatchTest() local
245 Tensor5DFloat32 src_tensor; in SplitDepthTest() local
H A Dtile_test_util.cc30 TensorFloat32 src_tensor; in TileChannelsTest() local
57 TensorFloat32 src_tensor; in TileChannelsX4Test() local
85 TensorFloat32 src_tensor; in TileWidthTest() local
112 TensorFloat32 src_tensor; in TileHeightTest() local
139 TensorFloat32 src_tensor; in TileHWCTest() local
H A Dconv_generic_test_util.cc30 TensorFloat32 src_tensor; in ConvGeneric1x1SimpleWeightsTest() local
67 TensorFloat32 src_tensor; in ConvGeneric1x1Test() local
104 TensorFloat32 src_tensor; in ConvGenericSimpleWeightsTest() local
140 TensorFloat32 src_tensor; in ConvGenericTest() local
178 TensorFloat32 src_tensor; in ConvGenericGroupedTest() local
H A Dreduce_test_util.cc113 TensorFloat32 src_tensor; in MeanHWTest() local
140 TensorFloat32 src_tensor; in ReduceSumChannelsTest() local
171 TensorFloat32 src_tensor; in ReduceProductChannelsTest() local
202 TensorFloat32 src_tensor; in ReduceMaxChannelsTest() local
230 TensorFloat32 src_tensor; in ReduceMinChannelsTest() local
H A Dspace_to_depth_test_util.cc31 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x1BlockSize2Test() local
57 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x2BlockSize2Test() local
85 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x3BlockSize2Test() local
115 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x4x4x1BlockSize2Test() local
H A Drelu_test_util.cc30 TensorFloat32 src_tensor; in ReLUNoClipNoAlphaTest() local
59 TensorFloat32 src_tensor; in ReLUClipTest() local
88 TensorFloat32 src_tensor; in ReLUAlphaTest() local
117 TensorFloat32 src_tensor; in ReLUAlphaClipTest() local
H A Dsoftmax_test_util.cc31 TensorFloat32 src_tensor; in SoftmaxTest() local
58 TensorFloat32 src_tensor; in SoftmaxBigNumberTest() local
101 TensorFloat32 src_tensor; in Softmax1x1Test() local
127 TensorFloat32 src_tensor; in Softmax1x1BigNumberTest() local
H A Dfully_connected_test_util.cc30 TensorFloat32 src_tensor; in FullyConnectedTest() local
63 TensorFloat32 src_tensor; in FullyConnectedLargeTest() local
114 TensorFloat32 src_tensor; in FullyConnectedExtraLargeTest() local
170 TensorFloat32 src_tensor; in FullyConnectedInt8Test() local
H A Dquantize_and_dequantize_test_util.cc29 TensorFloat32 src_tensor; in QuantAndDequant_Dim2Bits8Test() local
66 TensorFloat32 src_tensor; in QuantAndDequant_Dim3Bits8_NegativeRangeTest() local
102 TensorFloat32 src_tensor; in QuantAndDequant_Dim3Bits16Test() local
139 TensorFloat32 src_tensor; in QuantAndDequant_Dim2Bits16_NegativeRangeTest() local
H A Dpooling_test_util.cc30 TensorFloat32 src_tensor; in AveragePoolingTest() local
61 TensorFloat32 src_tensor; in AveragePoolingNonEmptyPaddingTest() local
93 TensorFloat32 src_tensor; in MaxPoolingTest() local
124 TensorFloat32 src_tensor; in MaxPoolingIndicesTest() local
H A Dwinograd_test_util.cc31 TensorFloat32 src_tensor; in Winograd4x4To36TileX6Test() local
97 TensorFloat32 src_tensor; in Winograd36To4x4Tile4x1Test() local
166 TensorFloat32 src_tensor; in Winograd4x4To36Test() local
231 TensorFloat32 src_tensor; in Winograd36To4x4Test() local
H A Ddepthwise_conv_test_util.cc30 TensorFloat32 src_tensor; in DepthwiseConvSimpleWeightsTest() local
67 TensorFloat32 src_tensor; in DepthwiseConvNoMultiplierTest() local
104 TensorFloat32 src_tensor; in DepthwiseConvMultiplier2Test() local
H A Dmean_stddev_normalization_test_util.cc35 TensorFloat32 src_tensor; in MeanStddevNormSeparateBatchesTest() local
86 TensorFloat32 src_tensor; in MeanStddevNormalizationAllBatchesTest() local
156 TensorFloat32 src_tensor; in MeanStddevNormalizationLargeVectorTest() local
H A Dreshape_test_util.cc31 TensorFloat32 src_tensor; in ReshapeTest() local
57 TensorFloat32 src_tensor; in Reshapex4Test() local
H A Dconv_buffer_1x1_test_util.cc29 TensorFloat32 src_tensor; in ConvBuffer1x1SimpleWeightsTest() local
65 TensorFloat32 src_tensor; in ConvBuffer1x1Test() local
H A Dconv_constants_test_util.cc30 TensorFloat32 src_tensor; in ConvConstantsSimpleWeightsTest() local
66 TensorFloat32 src_tensor; in ConvConstantsTest() local
H A Dconvolution_transposed_3x3_thin_test_util.cc31 TensorFloat32 src_tensor; in ConvolutionTransposed3x3ThinSimpleWeightsTest() local
69 TensorFloat32 src_tensor; in ConvolutionTransposed3x3ThinTest() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_relu_op.cc444 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute() local
596 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute() local
776 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
808 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
843 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
879 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
924 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
970 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
1009 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
1041 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
[all …]
H A Dmkl_aggregate_ops.cc42 const Tensor& src_tensor = MklGetInput(ctx, src_index); in GetTensorShape() local
95 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in ComputeScalar() local
137 const Tensor& src_tensor = MklGetInput(ctx, kSrc0Idx); in Compute() local
188 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in Compute() local
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/utils/
H A DDependencyGraph.h113 for(auto src_tensor : inputs) variable
331 for(auto src_tensor : _adj_src_ops) in intermediate_tensors() local
395 for(TensorId src_tensor : src_tensors(op)) in src_ops() local
451 for(auto src_tensor : _adj_src_tensors.at(op)) in remove_operator() local
537 bool path_exists_from_tensor_to_op(TensorId src_tensor, OperatorId dst_op) const in path_exists_from_tensor_to_op()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dwhile.cc66 const TfLiteTensor* src_tensor = in CopyTensorsShapeAndType() local
97 const TfLiteTensor* src_tensor = in CopyTensorsData() local
157 const TfLiteTensor* src_tensor = in ShallowCopyTensorsShapeTypeData() local
179 const TfLiteTensor* src_tensor = in ShallowCopyTensorsShapeTypeData() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dgpu_model_test_util.cc87 TensorFloat32 src_tensor; in TestLinkingConvolutionAndCosOp() local
763 TensorFloat32 src_tensor; in TestLinkingConvolutionFirstTanh2InputDiff() local
881 TensorFloat32 src_tensor; in TestLinkingConvolutionSecondTanh2InputDiff() local
1016 TensorFloat32 src_tensor; in TestLinkingConvolutionFirstTanhSecondCos2InputDiff() local
1207 TensorFloat32 src_tensor; in TestLinkingComplex0() local

123