Home
last modified time | relevance | path

Searched defs:sizes_vec (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dcomm.hpp27 std::vector<c10::IntArrayRef> sizes_vec, in GradBucket()
H A Dreducer.hpp374 std::vector<c10::IntArrayRef> sizes_vec; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinear.cpp67 c10::SymDimVector sizes_vec(new_size.begin(), new_size.end()); in _flatten_nd_linear() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc2219 auto sizes_vec = sizes_value->vec<T>(); in SliceHelper() local