Home
last modified time | relevance | path

Searched defs:shape1 (Results 1 – 25 of 73) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dunsigned_quantized_concatenation_test.cc38 const std::vector<int32_t> shape1({shape_rng()}); variable
63 const std::vector<int32_t> shape1({shape_rng(), shape_rng()}); variable
87 const std::vector<int32_t> shape1({shape_rng(), shape_rng(), shape_rng()}); variable
111 const std::vector<int32_t> shape1( variable
133 const std::vector<int32_t> shape1({shape_rng()}); variable
159 const std::vector<int32_t> shape1({shape_rng(), shape_rng()}); variable
184 const std::vector<int32_t> shape1({shape_rng(), shape_rng(), shape_rng()}); variable
209 const std::vector<int32_t> shape1( variable
232 const std::vector<int32_t> shape1({shape_rng()}); variable
259 const std::vector<int32_t> shape1({shape_rng(), shape_rng()}); variable
[all …]
H A Dconcatenation_test.cc38 const std::vector<int32_t> shape1({shape_rng()}); variable
63 const std::vector<int32_t> shape1({shape_rng(), shape_rng()}); variable
87 const std::vector<int32_t> shape1({shape_rng(), shape_rng(), shape_rng()}); variable
111 const std::vector<int32_t> shape1( variable
133 const std::vector<int32_t> shape1({shape_rng()}); variable
159 const std::vector<int32_t> shape1({shape_rng(), shape_rng()}); variable
184 const std::vector<int32_t> shape1({shape_rng(), shape_rng(), shape_rng()}); variable
209 const std::vector<int32_t> shape1( variable
232 const std::vector<int32_t> shape1({shape_rng()}); variable
259 const std::vector<int32_t> shape1({shape_rng(), shape_rng()}); variable
[all …]
H A Dsigned_quantized_concatenation_test.cc38 const std::vector<int32_t> shape1({shape_rng()}); variable
63 const std::vector<int32_t> shape1({shape_rng(), shape_rng()}); variable
87 const std::vector<int32_t> shape1({shape_rng(), shape_rng(), shape_rng()}); variable
111 const std::vector<int32_t> shape1( variable
133 const std::vector<int32_t> shape1({shape_rng()}); variable
159 const std::vector<int32_t> shape1({shape_rng(), shape_rng()}); variable
184 const std::vector<int32_t> shape1({shape_rng(), shape_rng(), shape_rng()}); variable
209 const std::vector<int32_t> shape1( variable
232 const std::vector<int32_t> shape1({shape_rng()}); variable
259 const std::vector<int32_t> shape1({shape_rng(), shape_rng()}); variable
[all …]
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DSubtractionTestImpl.cpp25 const unsigned int shape1[] = { 1, 1, 2, 2 }; in SubtractionUint8Test() local
55 const unsigned int shape1[] = { 1, 1, 1, 1 }; in SubtractionBroadcast1ElementUint8Test() local
87 const unsigned int shape1[] = { 1, 1, 2, 1 }; in SubtractionBroadcastUint8Test() local
113 const unsigned int shape1[] = { 1, 1, 2, 2 }; in SubtractionTest() local
137 const unsigned int shape1[] = { 1, 1, 1, 1 }; in SubtractionBroadcast1ElementTest() local
163 const unsigned int shape1[] = { 1, 1, 1, 2 }; in SubtractionBroadcastTest() local
191 const unsigned int shape1[] = { 1, 1, 2, 2 }; in SubtractionFloat16Test() local
217 const unsigned int shape1[] = { 1, 1, 1, 1 }; in SubtractionBroadcast1ElementFloat16Test() local
245 const unsigned int shape1[] = { 1, 1, 1, 2 }; in SubtractionBroadcastFloat16Test() local
300 const unsigned int shape1[] = { 1, 1, 1, 1 }; in SubtractionBroadcast1ElementInt16Test() local
[all …]
H A DMinimumTestImpl.cpp26 unsigned int shape1[] = { 1, 1, 1, 1 }; in MinimumBroadcast1ElementTest1() local
52 unsigned int shape1[] = { 1, 1, 1, 1 }; in MinimumBroadcast1ElementTest2() local
78 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MinimumBroadcast1DVectorUint8Test() local
153 const unsigned int shape1[] = { 1, 1, 1, 1 }; in MinimumBroadcast1ElementFloat16Test() local
188 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MinimumBroadcast1DVectorFloat16Test() local
259 const unsigned int shape1[] = { 1, 1, 1, 1 }; in MinimumBroadcast1ElementInt16Test() local
293 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MinimumBroadcast1DVectorInt16Test() local
364 const unsigned int shape1[] = { 1, 1, 1, 1 }; in MinimumBroadcast1ElementInt32Test() local
398 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MinimumBroadcast1DVectorInt32Test() local
H A DMaximumTestImpl.cpp68 unsigned int shape1[] = { 1, 1, 1, 1 }; in MaximumBroadcast1ElementTest() local
94 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MaximumBroadcast1DVectorTest() local
174 unsigned int shape1[] = { 1, 1, 1, 1 }; in MaximumBroadcast1ElementFloat16Test() local
202 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MaximumBroadcast1DVectorFloat16Test() local
274 const unsigned int shape1[] = { 1, 1, 1, 1 }; in MaximumBroadcast1ElementUint8Test() local
308 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MaximumBroadcast1DVectorUint8Test() local
369 const unsigned int shape1[] = { 1, 1, 1, 1 }; in MaximumBroadcast1ElementInt16Test() local
403 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MaximumBroadcast1DVectorInt16Test() local
465 const unsigned int shape1[] = { 1, 1, 1, 1 }; in MaximumBroadcast1ElementInt32Test() local
499 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MaximumBroadcast1DVectorInt32Test() local
H A DDivisionTestImpl.cpp110 unsigned int shape1[] = { 1, 1, 1, 1 }; in DivisionBroadcast1ElementTest() local
136 unsigned int shape1[] = { 1, 1, 1, 2 }; in DivisionBroadcast1DVectorTest() local
218 unsigned int shape1[] = { 1, 1, 1, 1 }; in DivisionBroadcast1ElementFloat16Test() local
246 unsigned int shape1[] = { 1, 1, 1, 2 }; in DivisionBroadcast1DVectorFloat16Test() local
326 unsigned int shape1[] = { 1, 1, 1, 1 }; in DivisionBroadcast1ElementUint8Test() local
352 unsigned int shape1[] = { 1, 1, 1, 2 }; in DivisionBroadcast1DVectorUint8Test() local
427 unsigned int shape1[] = { 1, 1, 1, 1 }; in DivisionBroadcast1ElementInt16Test() local
453 unsigned int shape1[] = { 1, 1, 1, 2 }; in DivisionBroadcast1DVectorInt16Test() local
534 unsigned int shape1[] = { 1, 1, 1, 1 }; in DivisionBroadcast1ElementInt32Test() local
560 unsigned int shape1[] = { 1, 1, 1, 2 }; in DivisionBroadcast1DVectorInt32Test() local
H A DMultiplicationTestImpl.cpp139 unsigned int shape1[] = { 1, 1, 1, 1 }; in MultiplicationBroadcast1ElementTest() local
165 unsigned int shape1[] = { 1, 1, 1, 2 }; in MultiplicationBroadcast1DVectorTest() local
253 const unsigned int shape1[] = { 1, 1, 1, 1 }; in MultiplicationBroadcast1ElementUint8Test() local
287 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MultiplicationBroadcast1DVectorUint8Test() local
358 const unsigned int shape1[] = { 1, 1, 1, 1 }; in MultiplicationBroadcast1ElementInt16Test() local
392 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MultiplicationBroadcast1DVectorInt16Test() local
463 const unsigned int shape1[] = { 1, 1, 1, 1 }; in MultiplicationBroadcast1ElementInt32Test() local
497 const unsigned int shape1[] = { 1, 1, 1, 3 }; in MultiplicationBroadcast1DVectorInt32Test() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DElementwiseOperationsFixture.h86 …TensorType compute_target(const TensorShape &shape0, const TensorShape &shape1, DataType data_type… in compute_target()
162 SimpleTensor<T> compute_reference(const TensorShape &shape0, const TensorShape &shape1, in compute_reference()
203 …TensorType compute_target(const TensorShape &shape0, const TensorShape &shape1, DataType data_type… in compute_target()
263 SimpleTensor<T> compute_reference(const TensorShape &shape0, const TensorShape &shape1, in compute_reference()
281 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type0, DataType dat… in setup()
307 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type0, DataType dat… in setup()
333 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type0, DataType dat… in setup()
387 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type0, DataType dat… in setup()
413 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type0, DataType dat… in setup()
454 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type0, DataType dat… in setup()
[all …]
H A DArithmeticOperationsFixture.h49 …p(reference::ArithmeticOperation op, const TensorShape &shape0, const TensorShape &shape1, DataTyp… in setup()
66 …TensorType compute_target(const TensorShape &shape0, const TensorShape &shape1, DataType data_type… in compute_target()
126 …SimpleTensor<T> compute_reference(const TensorShape &shape0, const TensorShape &shape1, DataType d… in compute_reference()
154 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type, ConvertPolicy… in setup()
178 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type, ConvertPolicy… in setup()
215 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type, ConvertPolicy… in setup()
228 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type, ConvertPolicy… in setup()
240 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type, ConvertPolicy… in setup()
290 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type, ConvertPolicy… in setup()
H A DNormalizePlanarYUVLayerFixture.h48 …void setup(TensorShape shape0, TensorShape shape1, DataType dt, DataLayout data_layout, Quantizati… in setup()
83 …TensorType compute_target(TensorShape shape0, const TensorShape &shape1, DataType dt, DataLayout d… in compute_target()
125 …SimpleTensor<T> compute_reference(const TensorShape &shape0, const TensorShape &shape1, DataType d… in compute_reference()
148 void setup(TensorShape shape0, TensorShape shape1, DataType dt, DataLayout data_layout) in setup()
159 …void setup(TensorShape shape0, TensorShape shape1, DataType dt, DataLayout data_layout, Quantizati… in setup()
H A DComparisonFixture.h48 …void setup(ComparisonOperation op, const TensorShape &shape0, const TensorShape &shape1, DataType … in setup()
62 … const TensorShape &shape0, const TensorShape &shape1, DataType data_type, in compute_target()
98 … const TensorShape &shape0, const TensorShape &shape1, DataType data_type, in compute_reference()
121 …void setup(ComparisonOperation op, const TensorShape &shape0, const TensorShape &shape1, DataType … in setup()
155 …void setup(ComparisonOperation op, const TensorShape &shape0, const TensorShape &shape1, DataType … in setup()
H A DPixelWiseMultiplicationFixture.h49 const TensorShape &shape1, in setup()
74 …TensorType compute_target(const TensorShape &shape0, const TensorShape &shape1, DataType dt_in1, D… in compute_target()
133 …SimpleTensor<T3> compute_reference(const TensorShape &shape0, const TensorShape &shape1, DataType … in compute_reference()
171 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType dt_in1, DataType dt_in2,… in setup()
208 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType dt_in1, DataType dt_in2,… in setup()
234 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType dt_in1, DataType dt_in2,… in setup()
H A DArithmeticDivisionFixture.h48 void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type) in setup()
65 …TensorType compute_target(const TensorShape &shape0, const TensorShape &shape1, DataType data_type) in compute_target()
99 …SimpleTensor<T> compute_reference(const TensorShape &shape0, const TensorShape &shape1, DataType d… in compute_reference()
H A DBatchNormalizationLayerFixture.h48 …void setup(TensorShape shape0, TensorShape shape1, float epsilon, bool use_beta, bool use_gamma, A… in setup()
93 …TensorType compute_target(TensorShape shape0, const TensorShape &shape1, float epsilon, Activation… in compute_target()
145 …SimpleTensor<T> compute_reference(const TensorShape &shape0, const TensorShape &shape1, float epsi… in compute_reference()
H A DLogicalFixture.h83 void setup(TensorShape shape0, TensorShape shape1) in setup()
90 TensorType compute_target(const TensorShape &shape0, const TensorShape &shape1) in compute_target()
110 SimpleTensor<T> compute_reference(const TensorShape &shape0, const TensorShape &shape1) in compute_reference()
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DBroadcast.cpp80 bool binaryOperationFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, in binaryOperationFloat16()
96 bool addFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in addFloat32()
123 bool addFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in addFloat16()
130 bool addQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in addQuant8()
235 bool mulFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in mulFloat32()
262 bool mulFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in mulFloat16()
269 bool mulQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in mulQuant8()
316 bool subFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in subFloat32()
333 bool subFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in subFloat16()
340 bool subQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in subQuant8()
[all …]
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/operators/
H A DMulFixture.h74 …TensorType compute_target(const TensorShape &shape0, const TensorShape &shape1, const TensorShape … in compute_target()
160 …SimpleTensor<T> compute_reference(const TensorShape &shape0, const TensorShape &shape1, const Tens… in compute_reference()
215 …void setup(const TensorShape &shape0, const TensorShape &shape1, DataType data_type, bool is_inpla… in setup()
226 …void setup(const TensorShape &shape0, const TensorShape &shape1, const TensorShape &shape2, DataTy… in setup()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/
H A DElementwiseBinaryFixture.h98 …TensorType compute_target(const TensorShape &shape0, const TensorShape &shape1, const TensorShape … in compute_target()
184 …SimpleTensor<T> compute_reference(const TensorShape &shape0, const TensorShape &shape1, const Tens… in compute_reference()
235 …void setup(ArithmeticOperation ref_op, const TensorShape &shape0, const TensorShape &shape1, DataT… in setup()
246 …void setup(ArithmeticOperation ref_op, const TensorShape &shape0, const TensorShape &shape1, const… in setup()
/aosp_15_r20/external/armnn/src/backends/tosaReference/test/
H A DTosaRefLayerSupportTests.cpp23 TensorShape shape1 = {1,1,3,4}; variable
45 TensorShape shape1 = {4}; variable
196 TensorShape shape1 = {1,1,3,4}; variable
218 TensorShape shape1 = {1,2,3,4}; variable
466 TensorShape shape1 = {1,1,3,4}; variable
488 TensorShape shape1 = {4}; variable
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dcommon_shapes_test.py29 def _assert_incompatible_broadcast(self, shape1, shape2): argument
46 def _assert_broadcast(self, expected, shape1, shape2): argument
128 def _assert_broadcast_with_unknown_dims(self, expected, shape1, shape2): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
H A Delementwise_test.cc249 const BHWC shape1(1, 1, 1, 2); in TEST() local
359 const BHWC shape1(1, 1, 1, 2); in TEST() local
423 const BHWC shape1(1, 1, 1, 2); in TEST() local
487 const BHWC shape1(1, 1, 1, 2); in TEST() local
551 const BHWC shape1(1, 1, 1, 2); in TEST() local
614 const BHWC shape1(1, 1, 1, 2); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util_test.cc80 Shape shape1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
138 Shape shape1 = ShapeUtil::MakeShape(BF16, {3, 2}); in TEST() local
144 Shape shape1 = ShapeUtil::MakeShape(BF16, {3, 2}); in TEST() local
273 Shape shape1 = ShapeUtil::MakeShape(F32, {}); in TEST() local
285 Shape shape1 = ShapeUtil::MakeShape(F32, {5, 7}); in TEST() local
572 Shape shape1 = ShapeUtil::MakeShape(S32, {1, 9, 4, 1}); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Doutfeed_receiver_test.cc168 const Shape shape1 = ShapeUtil::MakeShape(U32, {128}); in TEST() local
211 const Shape shape1 = ShapeUtil::MakeShape(U32, {128}); in TEST() local
253 const Shape shape1 = ShapeUtil::MakeShape(U32, {128}); in TEST() local
/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_shape.cpp54 auto shape1 = Shape(); in TEST() local
64 auto shape1 = Shape(c10::ScalarType::Float, {}); in TEST() local

123