Searched defs:dep_token (Results 1 – 3 of 3) sorted by relevance
/external/pytorch/aten/src/ATen/native/ |
D | Constraints.cpp | 58 const Tensor& dep_token) { in _functional_sym_constrain_range() 75 const Tensor& dep_token) { in _functional_sym_constrain_range_for_size()
|
D | TensorCompare.cpp | 433 …ctional_assert_scalar(const Scalar& scalar, c10::string_view assert_msg, const Tensor& dep_token) { in _functional_assert_scalar() 441 const Tensor& dep_token) { in _functional_assert_async_msg_cpu()
|
/external/pytorch/torch/ |
D | _meta_registrations.py | 709 def functional_sym_constrain_range(size, min=None, max=None, dep_token=None): argument 725 def functional_sym_constrain_range_for_size(size, min, max, dep_token): argument 731 def functional_assert_async_meta(val, assert_msg, dep_token): argument
|