| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | resize_nearest_neighbor_test.cc | 32 const RuntimeShape& output_shape, in TestReferenceResizeNearestNeighbor()
52 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
63 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
75 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
87 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
98 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
109 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
121 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
133 RuntimeShape output_shape = {1, 2, 2, 1}; in TEST() local
144 RuntimeShape output_shape = {1, 2, 2, 1}; in TEST() local
[all …]
|
| H A D | transpose_utils_test.cc | 25 RuntimeShape output_shape({9}); in TEST() local
43 RuntimeShape output_shape({3, 9}); in TEST() local
63 RuntimeShape output_shape({1, 9}); in TEST() local
82 RuntimeShape output_shape({8, 4, 3}); in TEST() local
104 RuntimeShape output_shape({8, 4, 1}); in TEST() local
126 RuntimeShape output_shape({1, 4, 1}); in TEST() local
146 RuntimeShape output_shape({1, 1, 1}); in TEST() local
166 RuntimeShape output_shape({3, 9, 4, 2}); in TEST() local
190 RuntimeShape output_shape({3, 9, 4, 1}); in TEST() local
213 RuntimeShape output_shape({3, 1, 4, 1}); in TEST() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | reference_ops.h | 96 const RuntimeShape& output_shape, T* output_data) { in Relu()
108 const RuntimeShape& output_shape, T* output_data) { in Relu0To1()
122 const RuntimeShape& output_shape, T* output_data) { in Relu1()
135 const RuntimeShape& output_shape, float* output_data) { in Relu6()
150 const RuntimeShape& output_shape, T* output_data) { in ReluX()
168 const RuntimeShape& output_shape, T* output_data) { in ReluX()
191 const RuntimeShape& output_shape, in BroadcastMulFivefold()
241 const RuntimeShape& output_shape, int16* output_data) { in Mul()
260 const RuntimeShape& output_shape, uint8* output_data) { in Mul()
289 const RuntimeShape& output_shape, int16_t* output_data) { in Sub16()
[all …]
|
| H A D | sub.h | 37 const RuntimeShape& output_shape, in SubNonBroadcast()
53 const RuntimeShape& output_shape, in SubNonBroadcast()
74 const RuntimeShape& output_shape, in BroadcastSubSlow()
114 const RuntimeShape& output_shape, in BroadcastSubSlow()
154 const RuntimeShape& output_shape, int64_t* output_data) { in BroadcastSubSlow()
191 const RuntimeShape& output_shape, T* output_data) { in BroadcastSubSlow()
230 const RuntimeShape& output_shape, in BroadcastSub16POTSlow()
274 const RuntimeShape& output_shape, T* output_data) { in BroadcastQuantSubSlow()
356 const RuntimeShape& output_shape, uint8_t* output_data) { in Sub()
372 const RuntimeShape& output_shape, int8_t* output_data) { in Sub()
[all …]
|
| H A D | pad.h | 40 const P* pad_value_ptr, const RuntimeShape& output_shape, in PadImpl()
120 const P* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
130 const int32_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
141 const int32_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
151 const RuntimeShape& output_shape, T* output_data) { in PadImageStyle()
160 const RuntimeShape& output_shape, in PadImageStyle()
|
| H A D | mul.h | 51 const RuntimeShape& output_shape, T* output_data) { in Mul()
70 const RuntimeShape& output_shape, in Mul()
82 const RuntimeShape& output_shape, uint8_t* output_data) { in Mul()
96 const RuntimeShape& output_shape, in BroadcastMul4DSlow()
146 const RuntimeShape output_shape = in BroadcastMul4DSlow() local
190 const RuntimeShape output_shape = in BroadcastMul4DSlow() local
|
| H A D | add.h | 32 const RuntimeShape& output_shape, T* output_data) { in Add()
119 const RuntimeShape& output_shape, uint8_t* output_data) { in Add()
137 const RuntimeShape& output_shape, in AddGeneralParamScale()
202 const RuntimeShape& output_shape, T* output_data) { in BroadcastAdd4DSlow()
247 const RuntimeShape& output_shape, T* output_data) { in BroadcastAdd4DSlow()
309 const RuntimeShape& output_shape, in BroadcastAddFivefold()
|
| H A D | logistic.h | 31 const RuntimeShape& output_shape, float* output_data) { in Logistic()
62 const float* input_data, const RuntimeShape& output_shape, in Logistic()
70 const RuntimeShape& output_shape, int16_t* output_data) { in Logistic()
95 const RuntimeShape& output_shape, int8_t* output_data, in Logistic()
|
| H A D | tanh.h | 30 const RuntimeShape& output_shape, float* output_data) { in Tanh()
43 const float* input_data, const RuntimeShape& output_shape, in Tanh()
50 const int16_t* input_data, const RuntimeShape& output_shape, in Tanh()
84 const uint8_t* input_data, const RuntimeShape& output_shape, in Tanh()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | WinogradConvolutionLayerFixture.h | 59 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
109 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, const Pa… in compute_target()
249 …TensorShape output_shape = compute_winograd_input_transform_shape(TensorInfo(input_shape, 1, data_… in setup() local
297 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd… in compute_target()
338 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, cons… in compute_reference()
362 …TensorShape output_shape = compute_winograd_filter_transform_shape(TensorInfo(input_shape, 1, dat… in setup() local
411 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd… in compute_target()
453 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, cons… in compute_reference()
524 …TensorShape output_shape = compute_winograd_output_transform_shape(TensorInfo(input_shape, 1, data… in compute_target() local
569 …TensorShape output_shape = compute_winograd_output_transform_shape(TensorInfo(input_shape,… in compute_reference() local
|
| H A D | DeconvolutionLayerFixture.h | 53 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
153 …ape input_shape, TensorShape weights_shape, const TensorShape bias_shape, TensorShape output_shape, in compute_target()
211 …, const TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, in compute_reference()
260 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup() local
281 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup() local
302 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup() local
325 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup() local
|
| H A D | DirectConvolutionLayerFixture.h | 70 … const TensorShape output_shape = compute_deep_convolution_shape(input_info, weights_info, info); variable
77 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
149 …hape, TensorShape weights_shape, const TensorShape &bias_shape, TensorShape output_shape, const Pa… in compute_target()
206 …t TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, cons… in compute_reference()
260 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
273 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
|
| H A D | SpaceToBatchFixture.h | 43 …e input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, TensorShape output_shape, in setup()
63 … const TensorShape &block_shape_shape, const TensorShape &paddings_shape, TensorShape output_shape, in compute_target()
116 … const TensorShape &output_shape, DataType data_type, QuantizationInfo quantization_info) in compute_reference()
144 …e input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, TensorShape output_shape, in setup()
156 …e input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, TensorShape output_shape, in setup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | reshape_test.cc | 40 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
61 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
82 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
103 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
123 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
143 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
214 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | optimized_ops.h | 146 const RuntimeShape& output_shape, in BinaryBroadcastFiveFold()
272 const float* optional_bias_data, const RuntimeShape& output_shape, in FullyConnected()
308 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
371 const int32* bias_data_int32, const RuntimeShape& output_shape, in FullyConnected()
751 const int32* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected()
910 const float* bias_data, const RuntimeShape& output_shape, in Conv()
1009 int32_t* accum_scratch, const RuntimeShape& output_shape, in HybridConv()
1099 const RuntimeShape& output_shape, float* output_data, in HybridConvPerChannel()
1208 const int32* bias_data, const RuntimeShape& output_shape, in Conv()
1321 const RuntimeShape output_shape = in DepthToSpace() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | propagate_fixed_sizes.cc | 38 PaddingType padding_type, Shape* output_shape, in ComputeConvSizes()
157 const auto& output_shape = output_array.shape(); in ProcessConvOperator() local
458 auto& output_shape = *output_array.mutable_shape(); in ProcessTensorFlowReshapeOperator() local
650 Shape* output_shape = model->GetArray(output_name).mutable_shape(); in ProcessReorderAxesOperator() local
690 auto& output_shape = *output_array.mutable_shape(); in ProcessConcatenationOperator() local
754 Shape* output_shape = output_array.mutable_shape(); in ProcessRangeOperator() local
794 Shape output_shape = input_shape; in ProcessTensorFlowSplitOperator() local
888 Shape output_shape = input_shape; in ProcessTensorFlowSplitVOperator() local
967 std::vector<int32> output_shape = in ProcessResizeBilinearOperator() local
995 std::vector<int32> output_shape = in ProcessResizeNearestNeighborOperator() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_sharding_util_test.cc | 53 Shape output_shape = ShapeUtil::MakeShape(F32, {3, 5, 2}); in TEST() local
63 Shape output_shape = ShapeUtil::MakeShape(F32, {3, 5, 2}); in TEST() local
72 Shape output_shape = ShapeUtil::MakeShape(F32, {20, 7}); in TEST() local
84 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 4, 7}); in TEST() local
96 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 4, 7}); in TEST() local
110 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 16, 7}); in TEST() local
123 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 15, 2, 14}); in TEST() local
136 Shape output_shape = ShapeUtil::MakeShape(F32, {3, 5, 1, 7}); in TEST() local
149 Shape output_shape = ShapeUtil::MakeShape(F32, {8, 16, 1}); in TEST() local
169 Shape output_shape = ShapeUtil::MakeShape(F32, {}); in TEST() local
|
| H A D | dynamic_dimension_inference_test.cc | 361 auto output_shape = ShapeUtil::MakeShape(F32, {4, 2, 128, 128}); in TEST_F() local
400 auto output_shape = ShapeUtil::MakeShape(F32, {8, 64, 512}); in TEST_F() local
496 auto output_shape = ShapeUtil::MakeShape(F32, {3, 2, 1}); in TEST_F() local
535 auto output_shape = ShapeUtil::MakeShape(F32, {3, 1, 2}); in TEST_F() local
574 auto output_shape = ShapeUtil::MakeShape(F32, {6, 4, 1, 5, 2, 3}); in TEST_F() local
609 auto output_shape = ShapeUtil::MakeShape(F32, {1, 4, 5}); in TEST_F() local
634 auto output_shape = ShapeUtil::MakeShape(F32, {320, 4}); in TEST_F() local
660 auto output_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
712 auto output_shape = ShapeUtil::MakeShape(F32, {3, 2, 4}); in TEST_F() local
739 auto output_shape = ShapeUtil::MakeShape(F32, {2, 2, 2}); in TEST_F() local
[all …]
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/CL/ |
| H A D | ConvolutionLayer.cpp | 97 … TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &output_shape, Data… in is_post_op_list_valid_in_gemmconv()
298 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_SUITE() local
325 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_CASE() local
346 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_CASE() local
367 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_CASE() local
388 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_CASE() local
411 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_SUITE() local
425 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_CASE() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/heuristics/direct_conv/ |
| H A D | ClDirectConvDefaultConfigBifrost.cpp | 79 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_G71_f32() local
103 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_G71_f16() local
127 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_G71_u8() local
151 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_default_f32() local
175 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_default_f16() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/ |
| H A D | DirectConv2dFixture.h | 92 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, const Pa… in setup()
170 const TensorShape &output_shape, Conv2dAttributes conv2d_attr) in compute_reference()
207 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape output_shape, TensorSha… in setup()
241 …const TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_info… in setup() local
248 …hape, TensorShape weights_shape, const TensorShape &bias_shape, TensorShape output_shape, const Co… in compute_target()
320 …t TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, cons… in compute_reference()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | sparse_to_dense.cc | 40 TfLiteStatus Resize(TfLiteContext* context, const TfLiteTensor* output_shape, in Resize()
53 const TfLiteTensor* output_shape, in CheckDimensionsMatch()
129 const TfLiteTensor* output_shape, in ResizeOutputShape()
149 const TfLiteTensor* output_shape; in Prepare() local
205 const TfLiteTensor* output_shape; in SparseToDenseImpl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| H A D | fully_connected.h | 38 const RuntimeShape& output_shape, float* output_data) { in FullyConnectedSparseWeight()
84 const RuntimeShape& output_shape, int8_t* output_data, int thread_start, in FullyConnectedSparseWeight1x16Impl()
120 const RuntimeShape& output_shape, float* output_data, int thread_start, in FullyConnectedSparseWeight1x4Impl()
192 const RuntimeShape& output_shape; member
204 const RuntimeShape& output_shape, int8_t* output_data, in FullyConnectedSparseWeight1x16()
228 const RuntimeShape& output_shape, float* output_data, in FullyConnectedSparseWeight1x4()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | gather_scatter_handler.cc | 168 const HloGatherInstruction* gather, const Shape& output_shape, in PartitionGatherIndexPassthroughPartition()
271 const HloGatherInstruction* gather, Shape output_shape, in PartitionGatherPassthroughOperand()
375 const HloGatherInstruction* gather, Shape output_shape, in PartitionGatherTrivialIndexedOperandDimension()
486 const HloGatherInstruction* gather, Shape output_shape, in PartitionGatherIndexParallelDimensions()
639 PartitionedHlo& indices, const Shape& output_shape, in PartitionGather()
776 const Shape& output_shape, const HloSharding& output_sharding, in PartitionScatterPassthroughOperand()
815 const Shape& output_shape, const HloSharding& output_sharding, in PartitionScatterIndexPassthroughPartition()
954 const Shape& output_shape, const HloSharding& output_sharding, in PartitionScatterTrivialIndexedOperandDimension()
1010 const Shape& output_shape, const HloSharding& output_sharding, in PartitionScatter()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | image_resize_ops.cc | 43 xla::Shape output_shape = in GetOutputShape() local
55 const xla::Shape& output_shape) { in CompileGrad()
77 auto output_shape = GetOutputShape(ctx); in CompileForward() local
131 auto output_shape = in Compile() local
|