Searched refs:RandomBoundedTensor (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
H A D | randomized_tests.cc | 345 Tensor RandomBoundedTensor(DataType dtype, T lo, T hi, 349 Tensor RandomBoundedTensor(DataType dtype, T lo, T hi, 353 Tensor RandomBoundedTensor(DataType dtype, Tensor lo, Tensor hi); 844 Tensor OpTest::RandomBoundedTensor(DataType dtype, T lo, T hi, in RandomBoundedTensor() function in tensorflow::__anon92bff46a0111::OpTest 867 Tensor OpTest::RandomBoundedTensor(DataType dtype, T lo, T hi, in RandomBoundedTensor() function in tensorflow::__anon92bff46a0111::OpTest 869 return RandomBoundedTensor<T>(dtype, lo, hi, needs_unique_values, in RandomBoundedTensor() 873 Tensor OpTest::RandomBoundedTensor(DataType dtype, Tensor lo, Tensor hi) { in RandomBoundedTensor() function in tensorflow::__anon92bff46a0111::OpTest 1149 a.indices = RandomBoundedTensor<int32>(DT_INT32, 0, a.params_shape[axis] - 1, in ChooseGatherArguments() 1205 a.indices = RandomBoundedTensor(a.indices_type, indices_lo, indices_hi); in ChooseScatterArguments() 1585 .Input(RandomBoundedTensor<float>(DT_FLOAT, -1, 1, false)) in TEST_F() [all …]
|