Searched refs:NestedIntSymNodeImpl (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
H A D | NestedIntSymNodeImpl.cpp | 39 c10::SymNode NestedIntSymNodeImpl::eq(const c10::SymNode& other) { in eq() 44 c10::SymNode NestedIntSymNodeImpl::ne(const c10::SymNode& other) { in ne() 49 c10::SymNode NestedIntSymNodeImpl::ge(const c10::SymNode& other) { in ge() 54 c10::SymNode NestedIntSymNodeImpl::gt(const c10::SymNode& other) { in gt() 59 c10::SymNode NestedIntSymNodeImpl::lt(const c10::SymNode& other) { in lt() 64 c10::SymNode NestedIntSymNodeImpl::le(const c10::SymNode& other) { in le() 69 c10::SymNode NestedIntSymNodeImpl::mul(const c10::SymNode& other) { in mul() 73 return SymNode(c10::make_intrusive<NestedIntSymNodeImpl>(val_, coeff_ * *c)); in mul() 76 c10::SymNode NestedIntSymNodeImpl::clone() { in clone() 77 return SymNode(c10::make_intrusive<NestedIntSymNodeImpl>(val_, coeff_)); in clone()
|
H A D | NestedIntSymNodeImpl.h | 37 class TORCH_API NestedIntSymNodeImpl : public SymNodeImpl { 41 explicit NestedIntSymNodeImpl(int64_t val, int64_t coeff) in NestedIntSymNodeImpl() function
|
/aosp_15_r20/external/pytorch/test/cpp/api/ |
H A D | nested_int.cpp | 12 c10::SymNode(c10::make_intrusive<c10::NestedIntSymNodeImpl>(1, 1))); in TEST() 14 c10::SymNode(c10::make_intrusive<c10::NestedIntSymNodeImpl>(1, 1))); in TEST() 16 c10::SymNode(c10::make_intrusive<c10::NestedIntSymNodeImpl>(2, 1))); in TEST() 90 c10::SymNode(c10::make_intrusive<c10::NestedIntSymNodeImpl>(1, 5))); in TEST() 92 c10::SymNode(c10::make_intrusive<c10::NestedIntSymNodeImpl>(1, 10))); in TEST()
|
/aosp_15_r20/external/pytorch/torch/csrc/utils/ |
H A D | python_dispatch.cpp | 942 c10::make_intrusive<c10::NestedIntSymNodeImpl>(data, coeff))); in initDispatchBindings()
|
/aosp_15_r20/external/pytorch/ |
H A D | build_variables.bzl | 1065 "aten/src/ATen/core/NestedIntSymNodeImpl.cpp",
|