Home
last modified time | relevance | path

Searched defs:shape3 (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dsigned_quantized_concatenation_test.cc135 const std::vector<int32_t> shape3({shape_rng()}); variable
161 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
186 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
212 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
234 const std::vector<int32_t> shape3({shape_rng()}); variable
261 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
287 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
314 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
Dunsigned_quantized_concatenation_test.cc135 const std::vector<int32_t> shape3({shape_rng()}); variable
161 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
186 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
212 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
234 const std::vector<int32_t> shape3({shape_rng()}); variable
261 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
287 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
314 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
Dconcatenation_test.cc135 const std::vector<int32_t> shape3({shape_rng()}); variable
161 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
186 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
212 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
234 const std::vector<int32_t> shape3({shape_rng()}); variable
261 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
287 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
314 auto shape3 = SameShapeDifferentAxis(shape1, i, shape_rng()); variable
/external/pytorch/test/cpp/lazy/
Dtest_shape.cpp66 auto shape3 = Shape(c10::ScalarType::Long, {1, 2, 3}); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover_test.cc496 auto shape3 = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F() local
Dalgebraic_simplifier_test.cc6688 auto shape3 = ShapeUtil::MakeShape(F32, {2, 3, 2}); in TEST_F() local
6718 auto shape3 = ShapeUtil::MakeShape(F32, {2, 2, 3}); in TEST_F() local
6743 auto shape3 = ShapeUtil::MakeShape(F32, {2, 2, 3}); in TEST_F() local
6796 auto shape3 = ShapeUtil::MakeShape(F32, {2, 2, 3, 2}); in TEST_F() local
6917 auto shape3 = ShapeUtil::MakeShape(F32, {1, 2, 3, 4}); in TEST_F() local
Dmemory_space_assignment_test.cc996 Shape shape3 = ShapeUtil::MakeShape(F32, {1, 6}); in TEST_P() local
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py993 def reduce_with_shapes(shape1, shape2, shape3, dimensions_to_reduce=(1,)): argument
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc1092 Shape shape2, Shape shape3, int version) { in SimpleThreeInputsVersioningTest()