Searched defs:TensorIteratorBase (Results 1 – 2 of 2) sorted by relevance
248 struct TORCH_API TensorIteratorBase : public impl::MetaBase { struct253 TensorIteratorBase(); argument269 using loop_subiter_t = c10::function_ref<void(TensorIteratorBase& subiter)>; argument273 int ndim() const { in ndim()276 IntArrayRef shape() const { in shape()280 int ntensors() const { in ntensors()283 int noutputs() const { in noutputs()286 int ninputs() const { in ninputs()289 IntArrayRef view_offsets() const { in view_offsets()307 IntArrayRef strides(int64_t arg) const { in strides()[all …]
136 for_each_in_tuple(const std::tuple<tuple_t...>& /*t*/, const TensorIteratorBase& /*iter*/, const in… in for_each_in_tuple()