| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_ops_benchmark_test.cc | 76 static Tensor MakeRandomTensor(const TensorShape& shape) { in MakeRandomTensor() function
90 ? MakeRandomTensor<T>({batch, height, width, in_depth}) in Conv2D()
91 : MakeRandomTensor<T>({batch, in_depth, height, width}); in Conv2D()
95 MakeRandomTensor<T>({filter_w, filter_h, in_depth, out_depth}); in Conv2D()
127 Tensor bias_t = MakeRandomTensor<T>({out_depth}); in Conv2DWithBias()
176 Tensor scale_t = MakeRandomTensor<T>({out_depth}); in Conv2DWithBatchNorm()
177 Tensor offset_t = MakeRandomTensor<T>({out_depth}); in Conv2DWithBatchNorm()
178 Tensor mean_t = MakeRandomTensor<T>({out_depth}); in Conv2DWithBatchNorm()
179 Tensor variance_t = MakeRandomTensor<T>({out_depth}); in Conv2DWithBatchNorm()
236 ? MakeRandomTensor<T>({batch, height, width, in_depth})
[all …]
|
| H A D | conv_grad_filter_ops_benchmark_test.cc | 33 static Tensor MakeRandomTensor(const TensorShape& shape) { in MakeRandomTensor() function
47 ? MakeRandomTensor<T>({batch, height, width, in_depth}) in Conv2DBackpropFilter()
48 : MakeRandomTensor<T>({batch, in_depth, height, width}); in Conv2DBackpropFilter()
50 MakeRandomTensor<T>({filter_h, filter_w, in_depth, out_depth}); in Conv2DBackpropFilter()
64 ? MakeRandomTensor<T>( in Conv2DBackpropFilter()
66 : MakeRandomTensor<T>( in Conv2DBackpropFilter()
|
| H A D | conv_grad_input_ops_benchmark_test.cc | 32 static Tensor MakeRandomTensor(const TensorShape& shape) { in MakeRandomTensor() function
46 ? MakeRandomTensor<T>({batch, height, width, in_depth}) in Conv2DBackpropInput()
47 : MakeRandomTensor<T>({batch, in_depth, height, width}); in Conv2DBackpropInput()
49 MakeRandomTensor<T>({filter_w, filter_h, in_depth, out_depth}); in Conv2DBackpropInput()
63 ? MakeRandomTensor<T>( in Conv2DBackpropInput()
65 : MakeRandomTensor<T>( in Conv2DBackpropInput()
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | BatchNormalizationTestImpl.cpp | 636 auto input = MakeRandomTensor<float>(inputTensorInfo, 21312); in CompareBatchNormTest()
638 auto mean = MakeRandomTensor<float>(tensorInfo, 123); in CompareBatchNormTest()
639 auto variance = MakeRandomTensor<float>(tensorInfo, 234, 0.0f); in CompareBatchNormTest()
640 auto beta = MakeRandomTensor<float>(tensorInfo, 123); in CompareBatchNormTest()
641 auto gamma = MakeRandomTensor<float>(tensorInfo, 345); in CompareBatchNormTest()
|
| H A D | MultiplicationTestImpl.cpp | 548 auto input0 = MakeRandomTensor<float>(inputTensorInfo0, 803506992); in CompareMultiplicationTest()
549 auto input1 = MakeRandomTensor<float>(inputTensorInfo1, 54902257); in CompareMultiplicationTest()
|
| H A D | AdditionTestImpl.cpp | 667 auto input1 = MakeRandomTensor<float>(inputTensorInfo1, 1232); in CompareAdditionTest()
668 auto input2 = MakeRandomTensor<float>(inputTensorInfo2, 456); in CompareAdditionTest()
|
| H A D | ActivationTestImpl.cpp | 260 …std::vector<float> input = MakeRandomTensor<float>(inputTensorInfo, 4605828, lowerBound - 5.0f, up… in BoundedReLuRandomInputTest()
363 std::vector<T> input = MakeRandomTensor<T>(inputTensorInfo, 7123561); in ConstantLinearActivationTestCommon()
1263 std::vector<T> input = MakeRandomTensor<T>(inputTensorInfo, 21453, minVal, 10.f); in CompareActivationTestImpl()
|
| H A D | Conv2dTestImpl.cpp | 1395 auto input = MakeRandomTensor<T>(inputTensorInfo, 124908); in CompareConvolution2dTestImpl()
1396 auto kernel = MakeRandomTensor<T>(kernelDesc, 891234); in CompareConvolution2dTestImpl()
1397 auto bias = MakeRandomTensor<T>(biasDesc, 1028); in CompareConvolution2dTestImpl()
3090 auto input = MakeRandomTensor<T>(inputTensorInfo, 124908, 0.0f, 255.0f); in CompareDepthwiseConvolution2dTestImpl()
3091 auto kernel = MakeRandomTensor<T>(kernelDesc, 891234, 0.0f, 255.0f); in CompareDepthwiseConvolution2dTestImpl()
3092 …auto bias = MakeRandomTensor<typename FullyConnectedBiasTypeForInputType<T>::Type>(biasDesc, 102… in CompareDepthwiseConvolution2dTestImpl()
|
| H A D | NormalizationTestImpl.cpp | 320 auto input = MakeRandomTensor<float>(inputTensorInfo, 111234); in CompareNormalizationTestImpl()
|
| H A D | SoftmaxTestImpl.cpp | 262 auto input = MakeRandomTensor<T>(inputTensorInfo, 0xF00D, 0.0f, 1.0f); in CompareSoftmaxTestImpl()
|
| H A D | Pooling3dTestImpl.cpp | 1155 std::vector<T> input = MakeRandomTensor<T>(inputTensorInfo, 81715); in ComparePooling3dTestCommon()
|
| H A D | Pooling2dTestImpl.cpp | 790 std::vector<T> input = MakeRandomTensor<T>(inputTensorInfo, 81715); in ComparePooling2dTestCommon()
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
| H A D | ActivationFixture.hpp | 28 input = MakeRandomTensor<float>(inputTensorInfo, 21453); in ActivationFixture()
53 input = MakeRandomTensor<float>(inputTensorInfo, 2342423, 0.0f, 1.0f); in PositiveActivationFixture()
|
| /aosp_15_r20/external/armnn/include/armnnTestUtils/ |
| H A D | TensorHelpers.hpp | 217 std::vector<T> MakeRandomTensor(const armnn::TensorInfo& tensorInfo, in MakeRandomTensor() function
|