Searched refs:RandomDims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 295 std::vector<int64> RandomDims(int min_rank = 0,
443 std::vector<int64> OpTest::RandomDims(int min_rank, int max_rank, in RandomDims() function in tensorflow::__anon30dbcadd0111::OpTest
559 return RandomTensor(dtype, /*needs_unique_values=*/false, RandomDims()); in RandomTensor()
602 return RandomNonNegativeTensor(dtype, RandomDims()); in RandomNonNegativeTensor()
632 return BroadcastableDims(RandomDims(0, 3)); in BroadcastableDims()
843 dims = RandomDims(); in ExpectTfAndXlaOutputsAreClose()
988 auto shape = RandomDims(); in TEST_F()
1002 std::vector<int64> data_dims = RandomDims(); in TEST_F()
1022 std::vector<int64> data_dims = RandomDims(); in TEST_F()
1045 std::vector<int64> dims = RandomDims(1, 5, 1); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_mkldnn_contraction_kernel_test.cc | 24 Eigen::array<Index, NumDims> RandomDims(int min_dim = 1, int max_dim = 20) { in RandomDims() function
49 Eigen::array<Index, 2> dims = RandomDims<Index, 2>(1, 500); in TEST()
56 Eigen::array<Index, 2> slice_start = RandomDims<Index, 2>(0, 250); in TEST()
57 Eigen::array<Index, 2> slice_size = RandomDims<Index, 2>(100, 500); in TEST()
|