Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
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 …]
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 …]
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 …]
/external/armnn/src/backends/backendsCommon/test/layerTests/
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 …]
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
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
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
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
/external/ComputeLibrary/tests/validation/fixtures/
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()
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()
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()
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 …]
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()
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()
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()
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()
/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/operators/
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()
/external/tensorflow/tensorflow/python/framework/
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
/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/
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()
/external/armnn/src/backends/tosaReference/test/
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
/external/pytorch/test/cpp/lazy/
Dtest_shape.cpp54 auto shape1 = Shape(); in TEST() local
64 auto shape1 = Shape(c10::ScalarType::Float, {}); in TEST() local
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_ops.cc84 ShapeHandle shape0, shape1, shape2; in __anon1714fc320202() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dprocess_broadcast_shapes.h41 const RuntimeShape& shape1, in ProcessBroadcastShapes()
/external/ComputeLibrary/tests/datasets/
DDynamicFusionDataset.h94 void add_config(TensorShape shape0, TensorShape shape1, TensorShape shape2) in add_config()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
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

123