Searched defs:input_dp (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/cpp/api/ | ||
| D | parallel.cpp | 258 auto input_dp = torch::ones({16, 2, 3, 3}); in TEST_F() local |
| /external/pytorch/test/ | ||
| D | test_datapipe.py | 691 def __init__(self, input_dp): argument |