| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | concatenation_test.cc | 39 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 …]
|
| D | unsigned_quantized_concatenation_test.cc | 39 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 …]
|
| D | signed_quantized_concatenation_test.cc | 39 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 …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes_test.py | 29 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/operators/ |
| D | MulFixture.h | 74 …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()
|
| /external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/ |
| D | ElementwiseBinaryFixture.h | 98 …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()
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_ops.cc | 84 ShapeHandle shape0, shape1, shape2; in __anonac88f2010202() local
|
| /external/ComputeLibrary/tests/datasets/ |
| D | DynamicFusionDataset.h | 94 void add_config(TensorShape shape0, TensorShape shape1, TensorShape shape2) in add_config()
|
| /external/tensorflow/tensorflow/lite/swift/Tests/ |
| D | TensorTests.swift | 96 var shape2: Tensor.Shape = [2, 2, 3] in testEquatable() variable
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 81 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
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | broadcast_args.cc | 41 const TfLiteTensor* shape2; member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_test.cc | 54 TensorShape shape2({num_rows}); in BM_UnsortedSegmentReduction() local
122 TensorShape shape2({num_rows}); in BM_SegmentReduction() local
|
| D | set_kernels.cc | 419 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
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 170 const TensorShapeProto& shape2 = prop2[0].shape(); in Optimize() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | io_ops.cc | 94 ShapeHandle shape0, shape1, shape2; in __anonc18e58700302() local
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | merge.py | 47 def _compute_elemwise_op_output_shape(self, shape1, shape2): argument
|
| /external/gemmlowp/test/ |
| D | benchmark_all_sizes.cc | 216 bool operator<(const Shape& shape1, const Shape& shape2) { in operator <()
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 704 shape2, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover_test.cc | 495 auto shape2 = ShapeUtil::MakeShape(F32, {8, 7, 1}); in TEST_F() local
|
| D | memory_space_assignment_test.cc | 995 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST_P() local
2539 Shape shape2 = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_P() local
2668 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
2739 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
2806 Shape shape2 = ShapeUtil::MakeShape(xla::F32, {3, 3}); in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 993 def reduce_with_shapes(shape1, shape2, shape3, dimensions_to_reduce=(1,)): argument
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | WorkloadDataValidation.cpp | 374 unsigned int shape2[] = {1, 1, 3, 2}; variable
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 1068 Shape shape2, int version) { in SimpleTwoInputsVersioningTest()
1092 Shape shape2, Shape shape3, int version) { in SimpleThreeInputsVersioningTest()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 855 Shape shape2 = ShapeUtil::MakeShape(F32, {param2_dim0, param2_dim1}); in BM_ParallelFusion() local
|
| /external/XNNPACK/src/xnnpack/ |
| D | subgraph.h | 314 struct xnn_shape shape2; member
|