Home
last modified time | relevance | path

Searched refs:RandomShape (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Drandom_ops.cc34 .SetShapeFn(shape_inference::RandomShape);
60 return shape_inference::RandomShape(c); in __anon9ce6b60a0102()
71 .SetShapeFn(shape_inference::RandomShape);
92 return shape_inference::RandomShape(c); in __anon9ce6b60a0202()
103 .SetShapeFn(shape_inference::RandomShape);
Drisc_ops.cc300 .SetShapeFn(shape_inference::RandomShape);
/external/XNNPACK/test/
Dsubgraph-binary-tester.h43 std::vector<size_t> input1_shape = RandomShape(); in SetUp()
49 input2_shape = RandomShape(input2_num_dims); in SetUp()
81 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
Dsubgraph-unary-tester.h37 dims = RandomShape(); in UnaryTest()
54 std::vector<size_t> RandomShape() { in RandomShape() function
Dprelu.cc29 input_dims = RandomShape(4); in SetUp()
40 std::vector<size_t> RandomShape(size_t num_dims) in RandomShape() function in PreluTestF32
Dconcatenate3.cc37 input1_dims = RandomShape(); in Concatenate3Test()
39 input2_dims = RandomShape(input1_dims, axis); in Concatenate3Test()
40 input3_dims = RandomShape(input1_dims, axis); in Concatenate3Test()
70 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
Dconcatenate4.cc37 input1_dims = RandomShape(); in Concatenate4Test()
39 input2_dims = RandomShape(input1_dims, axis); in Concatenate4Test()
40 input3_dims = RandomShape(input1_dims, axis); in Concatenate4Test()
41 input4_dims = RandomShape(input1_dims, axis); in Concatenate4Test()
74 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
Dconcatenate2.cc37 input1_dims = RandomShape(); in Concatenate2Test()
39 input2_dims = RandomShape(input1_dims, axis); in Concatenate2Test()
66 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
Ddivide2.cc31 std::vector<size_t> dims = RandomShape(); in TEST_F()
Ddepth-to-space.cc37 input_dims = RandomShape(4); in DepthToSpaceTest()
54 std::vector<size_t> RandomShape(size_t num_dims) in RandomShape() function in DepthToSpaceTest
Dglobal-average-pooling-2d.cc38 input_dims = RandomShape(4); in GlobalAveragePooling2DTest()
48 std::vector<size_t> RandomShape(size_t num_dims) in RandomShape() function in GlobalAveragePooling2DTest
Dglobal-average-pooling-1d.cc38 input_dims = RandomShape(); in GlobalAveragePooling1DTest()
54 std::vector<size_t> RandomShape() in RandomShape() function in GlobalAveragePooling1DTest
Deven-split2.cc37 output1_dims = RandomShape(); in EvenSplit2Test()
65 std::vector<size_t> RandomShape() in RandomShape() function in EvenSplit2Test
Dfully-connected.cc47 input_dims = RandomShape(num_input_dims); in FullyConnectedTestBase()
65 std::vector<size_t> RandomShape(size_t num_dims) in RandomShape() function in FullyConnectedTestBase
Deven-split3.cc37 output1_dims = RandomShape(); in EvenSplit3Test()
68 std::vector<size_t> RandomShape() in RandomShape() function in EvenSplit3Test
Deven-split4.cc37 output1_dims = RandomShape(); in EvenSplit4Test()
71 std::vector<size_t> RandomShape() in RandomShape() function in EvenSplit4Test
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.h250 Status RandomShape(shape_inference::InferenceContext* c);
Dcommon_shape_fns.cc2173 Status RandomShape(shape_inference::InferenceContext* c) { in RandomShape() function
/external/eigen/unsupported/test/
Dcxx11_tensor_block_access.cpp31 static TensorBlockShapeType RandomShape() { in RandomShape() function
135 dims, {RandomShape(), RandomTargetSize(dims), zeroCost()}); in test_block_mapper_maps_every_element()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test.cc960 TEST(XlaCompilationTest, RandomShape) { in TEST() argument