Searched refs:AsIntTensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 1011 static Tensor AsIntTensor(DataType dtype, const std::vector<int64_t>& values) { in AsIntTensor() function
1174 a.paddings.CopyFrom(AsIntTensor(a.paddings_type, paddings_vec), in ChoosePadArguments()
1908 CHECK(crops.CopyFrom(AsIntTensor(DT_INT32, crop_vals), in TEST_F()
1946 CHECK(crops.CopyFrom(AsIntTensor(DT_INT32, crop_vals), in TEST_F()
2046 .Input(AsIntTensor(type, dims.first)) in TEST_F()
2047 .Input(AsIntTensor(type, dims.second)) in TEST_F()
2060 .Input(AsIntTensor(type, dims.first)) in TEST_F()
2061 .Input(AsIntTensor(type, dims.second)) in TEST_F()
2075 .Input(AsIntTensor(type_idx, dims_to)) in TEST_F()
2732 .Input(AsIntTensor(DT_INT32, RandomDims())) in TEST_F()
[all …]
|