is_fast_path_index_select_scale Class — pytorch Architecture

Architecture documentation for the is_fast_path_index_select_scale class in EmbeddingBag.cpp from the pytorch codebase.

Class cpp

Entity Profile

Source Code

aten/src/ATen/native/EmbeddingBag.cpp lines 92–98

template<typename index_t>
bool is_fast_path_index_select_scale(const Tensor& src, const Tensor& scale, Tensor& output, index_t padding_idx) {
  return (src.scalar_type() == kFloat || src.scalar_type() == kHalf ||
          src.scalar_type() == kBFloat16) &&
      src.strides()[1] == 1 && output.strides()[1] == 1 &&
      scale.strides()[0] == 1 && padding_idx < static_cast<index_t>(0);
}

Source

View on GitHub

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free