Searched refs:RandomDims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 296 std::vector<int64> RandomDims(int min_rank = 0,
444 std::vector<int64> OpTest::RandomDims(int min_rank, int max_rank, in RandomDims() function in tensorflow::__anon68708c3b0111::OpTest
560 return RandomTensor(dtype, /*needs_unique_values=*/false, RandomDims()); in RandomTensor()
603 return RandomNonNegativeTensor(dtype, RandomDims()); in RandomNonNegativeTensor()
633 return BroadcastableDims(RandomDims(0, 3)); in BroadcastableDims()
844 dims = RandomDims(); in ExpectTfAndXlaOutputsAreClose()
989 auto shape = RandomDims(); in TEST_F()
1003 std::vector<int64> data_dims = RandomDims(); in TEST_F()
1023 std::vector<int64> data_dims = RandomDims(); in TEST_F()
1046 std::vector<int64> dims = RandomDims(1, 5, 1); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_mkldnn_contraction_kernel_test.cc | 31 Eigen::array<Index, NumDims> RandomDims(int min_dim = 1, int max_dim = 20) { in RandomDims() function
56 Eigen::array<Index, 2> dims = RandomDims<Index, 2>(1, 500); in TEST()
63 Eigen::array<Index, 2> slice_start = RandomDims<Index, 2>(0, 250); in TEST()
64 Eigen::array<Index, 2> slice_size = RandomDims<Index, 2>(100, 500); in TEST()
|