Home
last modified time | relevance | path

Searched defs:RandomShape (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/XNNPACK/test/
H A Dsubgraph-binary-tester.h81 std::vector<size_t> RandomShape(size_t num_dims) in RandomShape() function
88 std::vector<size_t> RandomShape() { return RandomShape(shape_dist(rng)); } in RandomShape() function
H A Dconcatenate2.cc66 std::vector<size_t> RandomShape() in RandomShape() function in Concatenate2Test
73 std::vector<size_t> RandomShape(const std::vector<size_t> base_dims, size_t axis) in RandomShape() function in Concatenate2Test
H A Dconcatenate3.cc70 std::vector<size_t> RandomShape() in RandomShape() function in Concatenate3Test
77 std::vector<size_t> RandomShape(const std::vector<size_t> base_dims, size_t axis) in RandomShape() function in Concatenate3Test
H A Dconcatenate4.cc74 std::vector<size_t> RandomShape() in RandomShape() function in Concatenate4Test
81 std::vector<size_t> RandomShape(const std::vector<size_t> base_dims, size_t axis) in RandomShape() function in Concatenate4Test
H A Dsubgraph-unary-tester.h54 std::vector<size_t> RandomShape() { in RandomShape() function
H A Dprelu.cc40 std::vector<size_t> RandomShape(size_t num_dims) in RandomShape() function in PreluTestF32
H A Deven-split2.cc65 std::vector<size_t> RandomShape() in RandomShape() function in EvenSplit2Test
H A Dglobal-average-pooling-1d.cc54 std::vector<size_t> RandomShape() in RandomShape() function in GlobalAveragePooling1DTest
H A Dglobal-average-pooling-2d.cc48 std::vector<size_t> RandomShape(size_t num_dims) in RandomShape() function in GlobalAveragePooling2DTest
H A Deven-split3.cc68 std::vector<size_t> RandomShape() in RandomShape() function in EvenSplit3Test
H A Ddepth-to-space.cc54 std::vector<size_t> RandomShape(size_t num_dims) in RandomShape() function in DepthToSpaceTest
H A Deven-split4.cc71 std::vector<size_t> RandomShape() in RandomShape() function in EvenSplit4Test
H A Dfully-connected.cc65 std::vector<size_t> RandomShape(size_t num_dims) in RandomShape() function in FullyConnectedTestBase
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_block_access.cpp31 static TensorBlockShapeType RandomShape() { in RandomShape() function
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc2173 Status RandomShape(shape_inference::InferenceContext* c) { in RandomShape() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dmark_for_compilation_pass_test.cc960 TEST(XlaCompilationTest, RandomShape) { in TEST() argument