Searched defs:IterArgs (Results 1 – 2 of 2) sorted by relevance
19 struct IterArgs { struct42 // using IterArgs<YourStructName>::operator() argument49 void operator()(c10::IListRef<T> args) { in operator()58 void operator()(at::ArrayRef<T> args) { in operator()67 void operator()(const torch::List<T>& args) { in operator()78 void operator()(const std::vector<T>& args) { in operator()82 constexpr bool short_circuit() const { in short_circuit()87 inline F& self() { in self()
583 class IterArgs { class in mlir::lmhlo::__anon91b1dc7e0111::SelectAndScatterOpConverter585 explicit IterArgs(ValueRange ivsValFlag) : ivsValFlag(ivsValFlag) {} in IterArgs() function in mlir::lmhlo::__anon91b1dc7e0111::SelectAndScatterOpConverter::IterArgs586 IterArgs(ValueRange ivs, Value value, Value flag) { in IterArgs() function in mlir::lmhlo::__anon91b1dc7e0111::SelectAndScatterOpConverter::IterArgs