Home
last modified time | relevance | path

Searched defs:shape2 (Results 1 – 25 of 45) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dunsigned_quantized_concatenation_test.cc39 const std::vector<int32_t> shape2({shape_rng()}); variable
64 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
88 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
113 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
134 const std::vector<int32_t> shape2({shape_rng()}); variable
160 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
185 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
211 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
233 const std::vector<int32_t> shape2({shape_rng()}); variable
260 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
[all …]
H A Dconcatenation_test.cc39 const std::vector<int32_t> shape2({shape_rng()}); variable
64 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
88 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
113 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
134 const std::vector<int32_t> shape2({shape_rng()}); variable
160 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
185 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
211 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
233 const std::vector<int32_t> shape2({shape_rng()}); variable
260 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
[all …]
H A Dsigned_quantized_concatenation_test.cc39 const std::vector<int32_t> shape2({shape_rng()}); variable
64 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
88 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
113 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
134 const std::vector<int32_t> shape2({shape_rng()}); variable
160 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
185 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
211 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
233 const std::vector<int32_t> shape2({shape_rng()}); variable
260 auto shape2 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
[all …]
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DBroadcast.cpp81 const Shape& shape2, int32_t activation, _Float16* out, 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 …]
DMaximumMinimum.cpp95 bool eval(const void* in1, const Shape& shape1, const void* in2, const Shape& shape2, in eval()
/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/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/operators/
H A DMulFixture.h74 …ype compute_target(const TensorShape &shape0, const TensorShape &shape1, const TensorShape &shape2) in compute_target()
160 … compute_reference(const TensorShape &shape0, const TensorShape &shape1, const TensorShape &shape2) in compute_reference()
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 …ype compute_target(const TensorShape &shape0, const TensorShape &shape1, const TensorShape &shape2) in compute_target()
184 … compute_reference(const TensorShape &shape0, const TensorShape &shape1, const TensorShape &shape2) in compute_reference()
246 …n ref_op, const TensorShape &shape0, const TensorShape &shape1, const TensorShape &shape2, DataTyp… in setup()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util_test.cc81 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
139 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
145 Shape shape2 = ShapeUtil::MakeShape(F32, {2, 2}); in TEST() local
274 Shape shape2 = ShapeUtil::MakeTupleShape( in TEST() local
286 Shape shape2 = ShapeUtil::MakeOpaqueShape(); in TEST() local
573 Shape shape2 = ShapeUtil::MakeShape(S32, {3, 1, 12}); in TEST() local
/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_shape.cpp57 auto shape2 = Shape(c10::ScalarType::Float, {1, 2, 3}); in TEST() local
65 auto shape2 = Shape(c10::ScalarType::Float, {1, 2, 3}); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsegment_reduction_ops_test.cc54 TensorShape shape2({num_rows}); in BM_UnsortedSegmentReduction() local
122 TensorShape shape2({num_rows}); in BM_SegmentReduction() local
H A Dset_kernels.cc419 Status CheckShapesMatch(VarDimArray shape1, VarDimArray shape2) { in CheckShapesMatch()
430 Status GroupShapeFromInputs(VarDimArray shape1, VarDimArray shape2, in GroupShapeFromInputs()
475 const auto shape2 = TensorShapeToArray(set2_t.shape()); in ComputeDenseToDense() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DUtilsTest.cpp43 Shape shape2; in TEST() local
60 Shape shape2; in TEST() local
/aosp_15_r20/developers/build/prebuilts/gradle/ElevationBasic/Application/src/main/java/com/example/android/elevationbasic/
H A DElevationBasicFragment.java47 View shape2 = rootView.findViewById(R.id.floating_shape_2); in onCreateView() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_ops.cc84 ShapeHandle shape0, shape1, shape2; in __anon3c2861b20202() local
/aosp_15_r20/developers/samples/android/ui/views/Elevation/ElevationBasic/Application/src/main/java/com/example/android/elevationbasic/
DElevationBasicFragment.java47 View shape2 = rootView.findViewById(R.id.floating_shape_2); in onCreateView() local
/aosp_15_r20/development/samples/browseable/ElevationBasic/src/com.example.android.elevationbasic/
H A DElevationBasicFragment.java47 View shape2 = rootView.findViewById(R.id.floating_shape_2); in onCreateView() local
/aosp_15_r20/external/ComputeLibrary/tests/datasets/
H A DDynamicFusionDataset.h94 void add_config(TensorShape shape0, TensorShape shape1, TensorShape shape2) in add_config()
/aosp_15_r20/external/tensorflow/tensorflow/lite/swift/Tests/
H A DTensorTests.swift96 var shape2: Tensor.Shape = [2, 2, 3] in testEquatable() variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbroadcast_args.cc41 const TfLiteTensor* shape2; member
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dshape_optimizer.cc170 const TensorShapeProto& shape2 = prop2[0].shape(); in Optimize() local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dhelpers.cpp123 Shape GetPromotedBinaryOpShape(const Shape& shape1, const Shape& shape2) { in GetPromotedBinaryOpShape()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dio_ops.cc94 ShapeHandle shape0, shape1, shape2; in __anon2a9f7d410302() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DExpandUtils.h60 inline bool are_expandable(IntArrayRef shape1, IntArrayRef shape2) { in are_expandable()
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/
H A Dmerge.py47 def _compute_elemwise_op_output_shape(self, shape1, shape2): argument

12