| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | transpose_utils_test.cc | 24 RuntimeShape input_shape({9}); in TEST() local
42 RuntimeShape input_shape({9, 3}); in TEST() local
62 RuntimeShape input_shape({9, 1}); in TEST() local
81 RuntimeShape input_shape({4, 3, 8}); in TEST() local
103 RuntimeShape input_shape({4, 1, 8}); in TEST() local
125 RuntimeShape input_shape({4, 1, 1}); in TEST() local
145 RuntimeShape input_shape({1, 1, 1}); in TEST() local
165 RuntimeShape input_shape({9, 3, 2, 4}); in TEST() local
189 RuntimeShape input_shape({9, 3, 1, 4}); in TEST() local
212 RuntimeShape input_shape({1, 3, 1, 4}); in TEST() local
[all …]
|
| H A D | resize_nearest_neighbor_test.cc | 30 const RuntimeShape& input_shape, const std::vector<T>& input_data, in TestReferenceResizeNearestNeighbor()
49 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
60 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
72 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
84 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
95 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
106 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
118 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
130 RuntimeShape input_shape = {1, 3, 3, 1}; in TEST() local
141 RuntimeShape input_shape = {1, 3, 3, 1}; in TEST() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops_test.cc | 71 const TensorShape input_shape({4, 4}); in TEST() local
98 const TensorShape input_shape({4, 4}); in TEST() local
132 Status CreateSplitTensorGraph(const TensorShape& input_shape, in CreateSplitTensorGraph()
159 Status CreateSplitResourceGraph(const TensorShape& input_shape, in CreateSplitResourceGraph()
214 const TensorShape input_shape({1, 1, 1}); in TEST_P() local
232 const TensorShape input_shape({1, 1, 1}); in TEST_P() local
250 const TensorShape input_shape({2}); in TEST_P() local
268 const TensorShape input_shape({2, 2}); in TEST_P() local
285 const TensorShape input_shape({2, 2}); in TEST_P() local
303 const TensorShape input_shape({}); in TEST_P() local
[all …]
|
| H A D | topk_ops.cc | 44 const TensorShape& input_shape, in CreateKthOrderStatisticComputation()
106 const TensorShape& input_shape = ctx->InputShape(0); in Compile() local
149 const TensorShape& input_shape, int64_t k, bool masked_with_iota) { in CreateTopKUnique()
214 const TensorShape& input_shape = ctx->InputShape(0); in Compile() local
259 const TensorShape& input_shape) { in CreateMakeUnique()
315 const TensorShape& input_shape = ctx->InputShape(0); in Compile() local
343 const TensorShape& input_shape = ctx->InputShape(0); in Compile() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | WinogradConvolutionLayerFixture.h | 59 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
109 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_sha… in compute_target()
166 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
247 …void setup(TensorShape input_shape, WinogradInfo winograd_info, DataLayout data_layout, DataType d… in setup()
297 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd… in compute_target()
338 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape,… in compute_reference()
359 …void setup(TensorShape input_shape, Size2D output_tile, DataLayout data_layout, DataType data_type) in setup()
411 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd… in compute_target()
453 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape,… in compute_reference()
522 …TensorType compute_target(const TensorShape &input_shape, const WinogradInfo &winograd_info, DataT… in compute_target()
[all …]
|
| H A D | DirectConvolutionLayerFixture.h | 77 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
149 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
206 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
234 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
247 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
260 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
273 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
|
| H A D | DeconvolutionLayerFixture.h | 53 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
153 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape bi… in compute_target()
211 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
250 …void setup(TensorShape input_shape, unsigned int sx, unsigned int sy, unsigned int padx, unsigned … in setup()
271 …void setup(TensorShape input_shape, unsigned int sx, unsigned int sy, unsigned int pad_left, unsig… in setup()
292 …void setup(TensorShape input_shape, unsigned int sx, unsigned int sy, unsigned int padx, unsigned … in setup()
314 …void setup(TensorShape input_shape, unsigned int sx, unsigned int sy, unsigned int padx, unsigned … in setup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | propagate_fixed_sizes.cc | 35 void ComputeConvSizes(const Shape& input_shape, int output_depth, int kwidth, in ComputeConvSizes()
131 const auto& input_shape = input_array.shape(); in ProcessConvOperator() local
228 const auto& input_shape = input_array.shape(); in ProcessTransposeConvOperator() local
258 const auto& input_shape = input_array.shape(); in ProcessDepthwiseConvOperator() local
300 const auto& input_shape = input_array.shape(); in ProcessDepthToSpaceOperator() local
323 const auto& input_shape = input_array.shape(); in ProcessSpaceToDepthOperator() local
373 const auto& input_shape = input_array.shape(); in ProcessFullyConnectedOperator() local
409 const auto& input_shape = input_array.shape(); in ProcessTensorFlowReshapeOperator() local
563 const auto& input_shape = input_array.shape(); in ProcessTensorFlowReductionOperator() local
621 const Shape& input_shape = input_array.shape(); in ProcessSliceOperator() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | convolution_test.cc | 116 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 1, 2}); in RunTest() local
147 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4}); in RunTest() local
185 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4}); in RunTest() local
225 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4}); in RunTest() local
262 Shape input_shape = ShapeUtil::MakeShape(F32, input_dims); in XLA_TEST_F() local
327 Shape input_shape = ShapeUtil::MakeShapeWithType<T>(input_dims); in RunTest() local
384 Shape input_shape = ShapeUtil::MakeShapeWithType<T>(input_dims); in RunTest() local
448 Shape input_shape = ShapeUtil::MakeShapeWithType<T>(input_dims); in RunTest() local
516 Shape input_shape = ShapeUtil::MakeShapeWithType<T>(input_dims); in RunTest() local
578 Shape input_shape = ShapeUtil::MakeShapeWithType<T>(input_dims); in RunTest() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | dynamic_dimension_inference_test.cc | 112 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
135 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
158 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
187 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
210 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
242 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
275 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
495 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 3}); in TEST_F() local
534 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 3}); in TEST_F() local
573 auto input_shape = ShapeUtil::MakeShape(F32, {2, 3, 4, 5, 6}); in TEST_F() local
[all …]
|
| H A D | hlo_sharding_util_test.cc | 52 Shape input_shape = ShapeUtil::MakeShape(F32, {2, 3, 5}); in TEST() local
62 Shape input_shape = ShapeUtil::MakeShape(F32, {2, 3, 5}); in TEST() local
71 Shape input_shape = ShapeUtil::MakeShape(F32, {4, 5, 7}); in TEST() local
83 Shape input_shape = ShapeUtil::MakeShape(F32, {16, 7}); in TEST() local
95 Shape input_shape = ShapeUtil::MakeShape(F32, {16, 7}); in TEST() local
109 Shape input_shape = ShapeUtil::MakeShape(F32, {16, 4, 7}); in TEST() local
122 Shape input_shape = ShapeUtil::MakeShape(F32, {16, 7, 5, 3}); in TEST() local
135 Shape input_shape = ShapeUtil::MakeShape(F32, {3, 1, 5, 7}); in TEST() local
148 Shape input_shape = ShapeUtil::MakeShape(F32, {8, 16}); in TEST() local
168 Shape input_shape = ShapeUtil::MakeShape(F32, {1, 1, 1}); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/ |
| H A D | convolutional.py | 189 def build(self, input_shape): argument
294 def compute_output_shape(self, input_shape): argument
367 def _get_input_channel(self, input_shape): argument
967 def build(self, input_shape): argument
1051 def compute_output_shape(self, input_shape): argument
1238 def build(self, input_shape): argument
1343 def compute_output_shape(self, input_shape): argument
1548 def build(self, input_shape): argument
1651 def compute_output_shape(self, input_shape): argument
1815 def build(self, input_shape): argument
[all …]
|
| H A D | advanced_activations.py | 87 def compute_output_shape(self, input_shape): argument
145 def build(self, input_shape): argument
181 def compute_output_shape(self, input_shape): argument
225 def compute_output_shape(self, input_shape): argument
273 def compute_output_shape(self, input_shape): argument
358 def compute_output_shape(self, input_shape): argument
448 def compute_output_shape(self, input_shape): argument
|
| H A D | core.py | 127 def compute_output_shape(self, input_shape): argument
226 def compute_output_shape(self, input_shape): argument
434 def compute_output_shape(self, input_shape): argument
487 def _fix_unknown_dimension(self, input_shape, output_shape): argument
531 def compute_output_shape(self, input_shape): argument
599 def compute_output_shape(self, input_shape): argument
686 def compute_output_shape(self, input_shape): argument
737 def compute_output_shape(self, input_shape): argument
869 def compute_output_shape(self, input_shape): argument
1176 def build(self, input_shape): argument
[all …]
|
| H A D | merge.py | 86 def build(self, input_shape): argument
182 def compute_output_shape(self, input_shape): argument
280 def build(self, input_shape): argument
487 def build(self, input_shape): argument
522 def compute_output_shape(self, input_shape): argument
657 def build(self, input_shape): argument
704 def compute_output_shape(self, input_shape): argument
|
| /aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| H A D | onedevice_train_palm_detection_filter_inference_beta.cc | 121 inline std::tuple<int, int> GetConvOutputSizeVALID(const int32_t* input_shape, in GetConvOutputSizeVALID()
132 inline std::tuple<int, int> GetConvOutputSizeSAME(const int32_t* input_shape, in GetConvOutputSizeSAME()
143 const int32_t* __restrict input_shape, in GetReduceInnerSize()
185 void MatMul(const int32_t* __restrict input_shape, in MatMul()
217 void DepthwiseConv2dNative(const int32_t* __restrict input_shape, in DepthwiseConv2dNative()
308 void DepthwiseConv2dNativeVALID(const int32_t* __restrict input_shape, in DepthwiseConv2dNativeVALID()
323 void DepthwiseConv2dNativeSAME(const int32_t* __restrict input_shape, in DepthwiseConv2dNativeSAME()
337 void FullyConnected(const int32_t* __restrict input_shape, in FullyConnected()
373 void SpaceToBatchNDRank4(const int32_t* __restrict input_shape, in SpaceToBatchNDRank4()
444 void BatchToSpaceNDRank4(const int32_t* __restrict input_shape, in BatchToSpaceNDRank4()
[all …]
|
| H A D | onedevice_train_palm_detection_filter_inference.cc | 122 inline std::tuple<int, int> GetConvOutputSizeVALID(const int32_t* input_shape, in GetConvOutputSizeVALID()
133 inline std::tuple<int, int> GetConvOutputSizeSAME(const int32_t* input_shape, in GetConvOutputSizeSAME()
144 const int32_t* __restrict input_shape, in GetReduceInnerSize()
186 void MatMul(const int32_t* __restrict input_shape, in MatMul()
218 void DepthwiseConv2dNative(const int32_t* __restrict input_shape, in DepthwiseConv2dNative()
309 void DepthwiseConv2dNativeVALID(const int32_t* __restrict input_shape, in DepthwiseConv2dNativeVALID()
324 void DepthwiseConv2dNativeSAME(const int32_t* __restrict input_shape, in DepthwiseConv2dNativeSAME()
338 void FullyConnected(const int32_t* __restrict input_shape, in FullyConnected()
374 void SpaceToBatchNDRank4(const int32_t* __restrict input_shape, in SpaceToBatchNDRank4()
445 void BatchToSpaceNDRank4(const int32_t* __restrict input_shape, in BatchToSpaceNDRank4()
[all …]
|
| H A D | onedevice_train_palm_detection_filter_inference_v2.cc | 121 inline std::tuple<int, int> GetConvOutputSizeVALID(const int32_t* input_shape, in GetConvOutputSizeVALID()
132 inline std::tuple<int, int> GetConvOutputSizeSAME(const int32_t* input_shape, in GetConvOutputSizeSAME()
143 const int32_t* __restrict input_shape, in GetReduceInnerSize()
185 void MatMul(const int32_t* __restrict input_shape, in MatMul()
217 void DepthwiseConv2dNative(const int32_t* __restrict input_shape, in DepthwiseConv2dNative()
308 void DepthwiseConv2dNativeVALID(const int32_t* __restrict input_shape, in DepthwiseConv2dNativeVALID()
323 void DepthwiseConv2dNativeSAME(const int32_t* __restrict input_shape, in DepthwiseConv2dNativeSAME()
337 void FullyConnected(const int32_t* __restrict input_shape, in FullyConnected()
373 void SpaceToBatchNDRank4(const int32_t* __restrict input_shape, in SpaceToBatchNDRank4()
444 void BatchToSpaceNDRank4(const int32_t* __restrict input_shape, in BatchToSpaceNDRank4()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | reshape_test.cc | 38 const std::vector<int32_t> input_shape{ variable
59 const std::vector<int32_t> input_shape{ variable
80 const std::vector<int32_t> input_shape{ variable
101 const std::vector<int32_t> input_shape{ variable
122 const std::vector<int32_t> input_shape{{shape_rng(), shape_rng()}}; variable
142 const std::vector<int32_t> input_shape{{shape_rng(), shape_rng()}}; variable
212 const std::vector<int32_t> input_shape{ in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | shape_op.cc | 45 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
77 const TensorShape input_shape = ctx->InputShape("input"); in Compile() local
181 const TensorShape input_shape = ctx->InputShape(i); in Compile() local
218 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
234 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
261 const TensorShape input_shape = ctx->InputShape("input"); in Compile() local
311 StatusOr<xla::Shape> input_shape = ctx->builder()->GetShape(ctx->Input(0)); in Compile() local
416 auto input_shape = ctx->InputXlaShape(0).ValueOrDie(); in Compile() local
439 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/ |
| H A D | DirectConv2dFixture.h | 92 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
109 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
169 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
207 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape output_shape, TensorSha… in setup()
226 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
248 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
320 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
345 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| H A D | transpose_test.cc | 77 const RuntimeShape input_shape = ::tflite::GetTensorShape(shape); in ComputeExpectedTransposeResult() local
97 std::vector<int> input_shape = {3}; in TEST() local
110 std::vector<int> input_shape = {3}; in TEST() local
122 std::vector<int> input_shape = {3, 2}; in TEST() local
135 std::vector<int> input_shape = {3, 2}; in TEST() local
148 std::vector<int> input_shape = {2, 2, 3, 1}; in TEST() local
161 std::vector<int> input_shape = {2, 2, 3, 1}; in TEST() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/CL/ |
| H A D | ConvolutionLayer.cpp | 97 bool is_post_op_list_valid_in_gemmconv(const TensorShape &input_shape, const TensorShape &weights_s… in is_post_op_list_valid_in_gemmconv()
295 const auto input_shape = TensorShape(16U, 14U, 12U, 2U); in TEST_SUITE() local
322 const auto input_shape = TensorShape(14U, 12U, 16U, 2U); in TEST_CASE() local
343 const auto input_shape = TensorShape(16U, 14U, 12U, 2U); in TEST_CASE() local
364 const auto input_shape = TensorShape(16U, 14U, 12U, 2U); in TEST_CASE() local
385 const auto input_shape = TensorShape(16U, 14U, 12U, 2U); in TEST_CASE() local
408 const auto input_shape = TensorShape(16U, 14U, 12U, 2U); in TEST_SUITE() local
422 const auto input_shape = TensorShape(16U, 14U, 12U, 2U); in TEST_CASE() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | pad.h | 39 const RuntimeShape& input_shape, const T* input_data, in PadImpl()
119 const RuntimeShape& input_shape, const T* input_data, in Pad()
129 const RuntimeShape& input_shape, const T* input_data, in Pad()
140 const RuntimeShape& input_shape, const int32_t* input_data, in Pad()
149 const RuntimeShape& input_shape, const T* input_data, in PadImageStyle()
158 const RuntimeShape& input_shape, in PadImageStyle()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | optimized_ops.h | 269 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
305 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
368 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
748 const FullyConnectedParams& params, const RuntimeShape& input_shape, in ShuffledFullyConnected()
907 inline void Conv(const ConvParams& params, const RuntimeShape& input_shape, in Conv()
1003 const RuntimeShape& input_shape, in HybridConv()
1096 const RuntimeShape& input_shape, const int8_t* input_data, in HybridConvPerChannel()
1205 inline void Conv(const ConvParams& params, const RuntimeShape& input_shape, in Conv()
1319 const RuntimeShape input_shape = in DepthToSpace() local
1360 const RuntimeShape input_shape = in SpaceToDepth() local
[all …]
|