is_fast_path Class — pytorch Architecture
Architecture documentation for the is_fast_path class in EmbeddingBag.cpp from the pytorch codebase.
Entity Profile
Source Code
aten/src/ATen/native/EmbeddingBag.cpp lines 100–105
template<typename index_t>
bool is_fast_path(const Tensor& src, const std::optional<Tensor>& scale, Tensor& output, index_t padding_idx) {
return (scale.has_value() && scale.value().defined()) ?
is_fast_path_index_select_scale(src, scale.value(), output, padding_idx) :
is_fast_path_index_select(src, output, padding_idx);
}
Source
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free