Home
last modified time | relevance | path

Searched refs:src_tensor (Results 1 – 25 of 80) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dpadding_test_util.cc30 TensorFloat32 src_tensor; in PaddingAppendWidthTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingAppendWidthTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in PaddingAppendWidthTest()
49 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in PaddingAppendWidthTest()
60 TensorFloat32 src_tensor; in PaddingPrependWidthTest() local
61 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingPrependWidthTest()
62 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in PaddingPrependWidthTest()
79 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in PaddingPrependWidthTest()
90 TensorFloat32 src_tensor; in PaddingAppendHeightTest() local
91 src_tensor.shape = BHWC(1, 2, 1, 2); in PaddingAppendHeightTest()
[all …]
H A Dmean_stddev_normalization_test_util.cc35 TensorFloat32 src_tensor; in MeanStddevNormSeparateBatchesTest() local
36 src_tensor.shape = BHWC(1, 1, 2, 4); in MeanStddevNormSeparateBatchesTest()
37 src_tensor.data = {mean - 2 * diff, mean - diff, mean + diff, in MeanStddevNormSeparateBatchesTest()
49 src_tensor.shape); in MeanStddevNormSeparateBatchesTest()
51 {src_tensor}, in MeanStddevNormSeparateBatchesTest()
70 op_def, env->GetGpuInfo(), src_tensor.shape, in MeanStddevNormSeparateBatchesTest()
73 env->ExecuteGPUOperation({src_tensor}, in MeanStddevNormSeparateBatchesTest()
86 TensorFloat32 src_tensor; in MeanStddevNormalizationAllBatchesTest() local
87 src_tensor.shape = BHWC(9, 1, 1, 4); in MeanStddevNormalizationAllBatchesTest()
88 src_tensor.data = { in MeanStddevNormalizationAllBatchesTest()
[all …]
H A Dsoftmax_test_util.cc31 TensorFloat32 src_tensor; in SoftmaxTest() local
32 src_tensor.shape = BHWC(1, 2, 1, 2); in SoftmaxTest()
33 src_tensor.data = {std::log(1.0f), std::log(2.0f), std::log(3.0f), in SoftmaxTest()
47 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in SoftmaxTest()
58 TensorFloat32 src_tensor; in SoftmaxBigNumberTest() local
59 src_tensor.shape = BHWC(1, 2, 1, 2); in SoftmaxBigNumberTest()
62 src_tensor.data.resize(4); in SoftmaxBigNumberTest()
63 src_tensor.data[0] = doubles[0]; in SoftmaxBigNumberTest()
64 src_tensor.data[1] = doubles[1]; in SoftmaxBigNumberTest()
65 src_tensor.data[2] = doubles[2]; in SoftmaxBigNumberTest()
[all …]
H A Dresize_test_util.cc30 TensorFloat32 src_tensor; in ResizeBilinearAlignedTest() local
31 src_tensor.shape = BHWC(1, 2, 3, 1); in ResizeBilinearAlignedTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f}; in ResizeBilinearAlignedTest()
50 src_tensor, std::make_unique<Resize>(std::move(operation)), in ResizeBilinearAlignedTest()
62 TensorFloat32 src_tensor; in ResizeBilinearNonAlignedTest() local
63 src_tensor.shape = BHWC(1, 2, 3, 1); in ResizeBilinearNonAlignedTest()
64 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f}; in ResizeBilinearNonAlignedTest()
82 src_tensor, std::make_unique<Resize>(std::move(operation)), in ResizeBilinearNonAlignedTest()
94 TensorFloat32 src_tensor; in ResizeBilinearWithoutHalfPixelTest() local
95 src_tensor.shape = BHWC(1, 2, 2, 1); in ResizeBilinearWithoutHalfPixelTest()
[all …]
H A Dtile_test_util.cc30 TensorFloat32 src_tensor; in TileChannelsTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 3); in TileChannelsTest()
32 src_tensor.data = {half(1.0f), half(2.0f), half(3.0f), in TileChannelsTest()
42 GPUOperation operation = CreateTile(op_def, src_tensor.shape.c); in TileChannelsTest()
44 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in TileChannelsTest()
57 TensorFloat32 src_tensor; in TileChannelsX4Test() local
58 src_tensor.shape = BHWC(1, 2, 1, 4); in TileChannelsX4Test()
59 src_tensor.data = {half(1.0f), half(2.0f), half(3.0f), half(7.0f), in TileChannelsX4Test()
69 GPUOperation operation = CreateTile(op_def, src_tensor.shape.c); in TileChannelsX4Test()
71 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in TileChannelsX4Test()
[all …]
H A Dwinograd_test_util.cc31 TensorFloat32 src_tensor; in Winograd4x4To36TileX6Test() local
32 src_tensor.shape = BHWC(1, 4, 4, 1); in Winograd4x4To36TileX6Test()
33 src_tensor.data.resize(16); in Winograd4x4To36TileX6Test()
35 src_tensor.data[i] = sin(i); in Winograd4x4To36TileX6Test()
51 const int index = src_tensor.shape.LinearIndex({0, y - 1, i - 1, 0}); in Winograd4x4To36TileX6Test()
52 sum += src_tensor.data[index] * b_t[x * 6 + i]; in Winograd4x4To36TileX6Test()
87 src_tensor, in Winograd4x4To36TileX6Test()
97 TensorFloat32 src_tensor; in Winograd36To4x4Tile4x1Test() local
98 src_tensor.shape = BHWC(1, 36, 1, 1); in Winograd36To4x4Tile4x1Test()
99 src_tensor.data.resize(36); in Winograd36To4x4Tile4x1Test()
[all …]
H A Dreduce_test_util.cc113 TensorFloat32 src_tensor; in MeanHWTest() local
114 src_tensor.shape = BHWC(1, 2, 2, 1); in MeanHWTest()
115 src_tensor.data = {1.0f, 2.0f, 3.0f, 4.0f}; in MeanHWTest()
128 CreateReduce(axis, src_tensor.shape, OperationType::MEAN, op_def, in MeanHWTest()
131 src_tensor, std::make_unique<Reduce>(std::move(operation)), in MeanHWTest()
140 TensorFloat32 src_tensor; in ReduceSumChannelsTest() local
141 src_tensor.shape = BHWC(1, 2, 1, 5); in ReduceSumChannelsTest()
142 src_tensor.data = {1.1, 2.1, 0.7, 0.3, 1.2, 3.1, 4.1, 0.0, 1.0, 4.4}; in ReduceSumChannelsTest()
155 CreateReduce(axis, src_tensor.shape, OperationType::REDUCE_SUM, in ReduceSumChannelsTest()
158 src_tensor, std::make_unique<Reduce>(std::move(operation)), in ReduceSumChannelsTest()
[all …]
H A Delementwise_test_util.cc30 TensorFloat32 src_tensor; in AbsTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 2); in AbsTest()
32 src_tensor.data = {half(0.0f), half(-1.0f), half(-0.05f), half(0.045f)}; in AbsTest()
45 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in AbsTest()
56 TensorFloat32 src_tensor; in CosTest() local
57 src_tensor.shape = BHWC(1, 2, 1, 2); in CosTest()
58 src_tensor.data = {0.0f, -1.0f, -0.05f, 0.045f}; in CosTest()
72 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in CosTest()
83 TensorFloat32 src_tensor; in CopyTest() local
84 src_tensor.shape = BHWC(1, 2, 1, 2); in CopyTest()
[all …]
H A Dconv_generic_test_util.cc30 TensorFloat32 src_tensor; in ConvGeneric1x1SimpleWeightsTest() local
31 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvGeneric1x1SimpleWeightsTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvGeneric1x1SimpleWeightsTest()
56 src_tensor, std::make_unique<ConvGeneric>(std::move(operation)), in ConvGeneric1x1SimpleWeightsTest()
67 TensorFloat32 src_tensor; in ConvGeneric1x1Test() local
68 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvGeneric1x1Test()
69 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvGeneric1x1Test()
93 src_tensor, std::make_unique<ConvGeneric>(std::move(operation)), in ConvGeneric1x1Test()
104 TensorFloat32 src_tensor; in ConvGenericSimpleWeightsTest() local
105 src_tensor.shape = BHWC(1, 2, 2, 2); in ConvGenericSimpleWeightsTest()
[all …]
H A Drelu_test_util.cc30 TensorFloat32 src_tensor; in ReLUNoClipNoAlphaTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 2); in ReLUNoClipNoAlphaTest()
32 src_tensor.data = {-0.5f, 0.8f, -0.6f, 3.2f}; in ReLUNoClipNoAlphaTest()
49 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in ReLUNoClipNoAlphaTest()
59 TensorFloat32 src_tensor; in ReLUClipTest() local
60 src_tensor.shape = BHWC(1, 2, 1, 2); in ReLUClipTest()
61 src_tensor.data = {-0.5f, 0.8f, -0.6f, 3.2f}; in ReLUClipTest()
78 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in ReLUClipTest()
88 TensorFloat32 src_tensor; in ReLUAlphaTest() local
89 src_tensor.shape = BHWC(1, 2, 1, 2); in ReLUAlphaTest()
[all …]
H A Dsplit_test_util.cc30 TensorFloat32 src_tensor; in SplitChannelsTest() local
31 src_tensor.shape = BHWC(1, 3, 2, 5); in SplitChannelsTest()
32 src_tensor.data = { in SplitChannelsTest()
54 {src_tensor}, std::make_unique<Split>(std::move(operation)), in SplitChannelsTest()
73 TensorFloat32 src_tensor; in SplitChannelsX4Test() local
74 src_tensor.shape = BHWC(1, 2, 2, 8); in SplitChannelsX4Test()
75 src_tensor.data = {half(0.1f), half(0.2f), half(0.3f), half(0.4), in SplitChannelsX4Test()
98 {src_tensor}, std::make_unique<Split>(std::move(operation)), in SplitChannelsX4Test()
118 TensorFloat32 src_tensor; in SplitWidthTest() local
119 src_tensor.shape = BHWC(1, 6, 5, 1); in SplitWidthTest()
[all …]
H A Dquantize_and_dequantize_test_util.cc29 TensorFloat32 src_tensor; in QuantAndDequant_Dim2Bits8Test() local
30 src_tensor.shape = BHWC(1, 3, 2, 1); in QuantAndDequant_Dim2Bits8Test()
31 src_tensor.data = {0.0f, 1.0f, 0.25f, 0.50f, 0.4444444f, 0.00001f}; in QuantAndDequant_Dim2Bits8Test()
54 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in QuantAndDequant_Dim2Bits8Test()
66 TensorFloat32 src_tensor; in QuantAndDequant_Dim3Bits8_NegativeRangeTest() local
67 src_tensor.shape = BHWC(1, 3, 1, 2); in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
68 src_tensor.data = {0.0f, -0.9f, 0.25f, 0.50f, 0.4444444f, -0.00001f}; in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
91 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
102 TensorFloat32 src_tensor; in QuantAndDequant_Dim3Bits16Test() local
103 src_tensor.shape = BHWC(1, 3, 1, 2); in QuantAndDequant_Dim3Bits16Test()
[all …]
H A Dspace_to_depth_test_util.cc31 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x1BlockSize2Test() local
32 src_tensor.shape = BHWC(1, 2, 2, 1); in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
33 src_tensor.data = {half(1.0f), half(2.0f), half(3.0f), half(4.0f)}; in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
45 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
57 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x2BlockSize2Test() local
58 src_tensor.shape = BHWC(1, 2, 2, 2); in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
59 src_tensor.data = {half(1.4f), half(2.3f), half(3.2f), half(4.1f), in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
72 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
85 TensorFloat32 src_tensor; in SpaceToDepthTensorShape1x2x2x3BlockSize2Test() local
86 src_tensor.shape = BHWC(1, 2, 2, 3); in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()
[all …]
H A Dfully_connected_test_util.cc30 TensorFloat32 src_tensor; in FullyConnectedTest() local
31 src_tensor.shape = BHWC(1, 1, 1, 4); in FullyConnectedTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in FullyConnectedTest()
53 src_tensor, std::make_unique<FullyConnected>(std::move(operation)), in FullyConnectedTest()
63 TensorFloat32 src_tensor; in FullyConnectedLargeTest() local
64 src_tensor.shape = BHWC(1, 1, 1, 8); in FullyConnectedLargeTest()
65 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in FullyConnectedLargeTest()
99 src_tensor, std::make_unique<FullyConnected>(std::move(operation)), in FullyConnectedLargeTest()
114 TensorFloat32 src_tensor; in FullyConnectedExtraLargeTest() local
115 src_tensor.shape = BHWC(1, 1, 1, kInputSize); in FullyConnectedExtraLargeTest()
[all …]
H A Dpooling_test_util.cc30 TensorFloat32 src_tensor; in AveragePoolingTest() local
31 src_tensor.shape = BHWC(1, 2, 2, 2); in AveragePoolingTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in AveragePoolingTest()
52 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in AveragePoolingTest()
61 TensorFloat32 src_tensor; in AveragePoolingNonEmptyPaddingTest() local
62 src_tensor.shape = BHWC(1, 2, 2, 1); in AveragePoolingNonEmptyPaddingTest()
63 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f}; in AveragePoolingNonEmptyPaddingTest()
83 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in AveragePoolingNonEmptyPaddingTest()
93 TensorFloat32 src_tensor; in MaxPoolingTest() local
94 src_tensor.shape = BHWC(1, 2, 2, 2); in MaxPoolingTest()
[all …]
H A Ddepthwise_conv_test_util.cc30 TensorFloat32 src_tensor; in DepthwiseConvSimpleWeightsTest() local
31 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConvSimpleWeightsTest()
32 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in DepthwiseConvSimpleWeightsTest()
56 src_tensor, std::make_unique<DepthwiseConv>(std::move(operation)), in DepthwiseConvSimpleWeightsTest()
67 TensorFloat32 src_tensor; in DepthwiseConvNoMultiplierTest() local
68 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConvNoMultiplierTest()
69 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in DepthwiseConvNoMultiplierTest()
93 src_tensor, std::make_unique<DepthwiseConv>(std::move(operation)), in DepthwiseConvNoMultiplierTest()
104 TensorFloat32 src_tensor; in DepthwiseConvMultiplier2Test() local
105 src_tensor.shape = BHWC(1, 2, 2, 2); in DepthwiseConvMultiplier2Test()
[all …]
H A Dconv_buffer_1x1_test_util.cc29 TensorFloat32 src_tensor; in ConvBuffer1x1SimpleWeightsTest() local
30 src_tensor.shape = BHWC(1, 1, 2, 4); in ConvBuffer1x1SimpleWeightsTest()
31 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvBuffer1x1SimpleWeightsTest()
54 CreateConvBuffer1x1(env->GetGpuInfo(), op_def, attr, &src_tensor.shape); in ConvBuffer1x1SimpleWeightsTest()
56 src_tensor, absl::make_unique<ConvBuffer1x1>(std::move(operation)), in ConvBuffer1x1SimpleWeightsTest()
65 TensorFloat32 src_tensor; in ConvBuffer1x1Test() local
66 src_tensor.shape = BHWC(1, 1, 2, 4); in ConvBuffer1x1Test()
67 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f}; in ConvBuffer1x1Test()
91 CreateConvBuffer1x1(env->GetGpuInfo(), op_def, attr, &src_tensor.shape); in ConvBuffer1x1Test()
93 src_tensor, absl::make_unique<ConvBuffer1x1>(std::move(operation)), in ConvBuffer1x1Test()
H A Dresampler_test_util.cc31 TensorFloat32 src_tensor; in ResamplerIdentityTest() local
32 src_tensor.shape = shape; in ResamplerIdentityTest()
33 src_tensor.data.resize(src_tensor.shape.DimensionsProduct()); in ResamplerIdentityTest()
34 for (int i = 0; i < src_tensor.data.size(); ++i) { in ResamplerIdentityTest()
35 src_tensor.data[i] = std::sin(i); in ResamplerIdentityTest()
59 {src_tensor, warp_tensor}, in ResamplerIdentityTest()
61 src_tensor.shape, &dst_tensor)); in ResamplerIdentityTest()
62 RETURN_IF_ERROR(PointWiseNear(src_tensor.data, dst_tensor.data, eps)); in ResamplerIdentityTest()
H A Dreshape_test_util.cc31 TensorFloat32 src_tensor; in ReshapeTest() local
32 src_tensor.shape = BHWC(1, 2, 1, 3); in ReshapeTest()
33 src_tensor.data = {half(0.5f), half(-1.1f), half(-2.2f), in ReshapeTest()
46 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in ReshapeTest()
57 TensorFloat32 src_tensor; in Reshapex4Test() local
58 src_tensor.shape = BHWC(1, 1, 1, 8); in Reshapex4Test()
59 src_tensor.data = {half(0.5f), half(-1.1f), half(-2.2f), half(3.1f), in Reshapex4Test()
72 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in Reshapex4Test()
H A Dprelu_test_util.cc30 TensorFloat32 src_tensor; in PReLUAlphaTest() local
31 src_tensor.shape = BHWC(1, 2, 1, 2); in PReLUAlphaTest()
32 src_tensor.data = {0.0f, -1.0f, -2.0f, 3.0f}; in PReLUAlphaTest()
51 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in PReLUAlphaTest()
61 TensorFloat32 src_tensor; in PReLUHWCAlphaTest() local
62 src_tensor.shape = BHWC(1, 2, 1, 2); in PReLUHWCAlphaTest()
63 src_tensor.data = {0.0f, -1.0f, -2.0f, 3.0f}; in PReLUHWCAlphaTest()
82 src_tensor, std::make_unique<GPUOperation>(std::move(operation)), in PReLUHWCAlphaTest()
/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
447 if (src_tensor.dims() == 0) { in Compute()
455 if (src_tensor.shape().num_elements() == 0) { in Compute()
471 src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute()
487 const T* src_data = src_tensor.flat<T>().data(); in Compute()
489 src.SetUsrMem(src_md, &src_tensor); in Compute()
510 dnn_shape_dst.SetTfLayout(src_tensor.dims(), in Compute()
511 TFShapeToMklDnnDims(src_tensor.shape()), in Compute()
519 tf_shape_dst = src_tensor.shape(); in Compute()
596 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute() local
[all …]
H A Dmkl_aggregate_ops.cc42 const Tensor& src_tensor = MklGetInput(ctx, src_index); in GetTensorShape() local
46 : src_tensor.shape(); in GetTensorShape()
95 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in ComputeScalar() local
96 T* src_i = const_cast<T*>(src_tensor.flat<T>().data()); in ComputeScalar()
137 const Tensor& src_tensor = MklGetInput(ctx, kSrc0Idx); in Compute() local
142 if (src_tensor.shape().num_elements() == 0) { in Compute()
144 output_tf_shape = src_tensor.shape(); in Compute()
150 if (src_tensor.dims() == 0) { in Compute()
188 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in Compute() local
195 if (src_tensor.dims() == 4) { in Compute()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
H A Dconv_test.mm78 TensorFloat32 src_tensor;
79 src_tensor.shape = src_shape;
80 src_tensor.data.resize(src_tensor.shape.DimensionsProduct());
81 for (int i = 0; i < src_tensor.data.size(); ++i) {
82 src_tensor.data[i] = sin(i);
98 env->ExecuteGPUOperation(src_tensor, std::move(op0_ptr), dst_shape, &output0));
113 env->ExecuteGPUOperation(src_tensor, std::move(op1_ptr), output1_shape, &output1));
133 TensorFloat32 src_tensor;
134 src_tensor.shape = BHWC(1, 1, 1, 8);
135 src_tensor.data = {0.0f, 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f};
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dwhile.cc66 const TfLiteTensor* src_tensor = in CopyTensorsShapeAndType() local
71 std::vector<int> dims(src_tensor->dims->data, in CopyTensorsShapeAndType()
72 src_tensor->dims->data + src_tensor->dims->size); in CopyTensorsShapeAndType()
77 TfLiteIntArrayCopy(src_tensor->dims))); in CopyTensorsShapeAndType()
79 dst_tensor->type = src_tensor->type; in CopyTensorsShapeAndType()
97 const TfLiteTensor* src_tensor = in CopyTensorsData() local
101 TfLiteTensorRealloc(src_tensor->bytes, dst_tensor); in CopyTensorsData()
103 TF_LITE_ENSURE_EQ(context, src_tensor->bytes, dst_tensor->bytes); in CopyTensorsData()
104 TfLiteTensorCopy(src_tensor, dst_tensor); in CopyTensorsData()
157 const TfLiteTensor* src_tensor = in ShallowCopyTensorsShapeTypeData() local
[all …]
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/utils/
H A DDependencyGraph.h113 for(auto src_tensor : inputs) variable
115 const auto src_ops = _adj_src_ops.find(src_tensor);
331 for(auto src_tensor : _adj_src_ops) in intermediate_tensors() local
333 if(!src_tensor.second.empty()) in intermediate_tensors()
335 const auto dst_tensor = _adj_dst_ops.find(src_tensor.first); in intermediate_tensors()
340 tensors.push_back(src_tensor.first); in intermediate_tensors()
395 for(TensorId src_tensor : src_tensors(op)) in src_ops() local
397 …ops.insert(ops.end(), std::begin(_adj_src_ops.at(src_tensor)), std::end(_adj_src_ops.at(src_tensor… in src_ops()
451 for(auto src_tensor : _adj_src_tensors.at(op)) in remove_operator() local
453 auto &dst_ops = _adj_dst_ops.at(src_tensor); in remove_operator()
[all …]

1234