Home
last modified time | relevance | path

Searched refs:max_vectorization_width (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dtiled_dot_emitter.cc666 int64_t max_vectorization_width, int64_t max_vector_count, in Config() argument
671 max_vectorization_width_(max_vectorization_width), in Config()
679 dims().ToString(), "_", max_vectorization_width(), in GetCacheKey()
686 int64_t max_vectorization_width() const { return max_vectorization_width_; } in max_vectorization_width() function in xla::cpu::__anondccdbc850111::TiledSmallGemmEmitter::Config
715 max_vectorization_width() > 0 && in TiledSmallGemmEmitter()
716 absl::has_single_bit(static_cast<uint64_t>(max_vectorization_width()))); in TiledSmallGemmEmitter()
721 CHECK_GE(max_vectorization_width(), min_vectorization_width()); in TiledSmallGemmEmitter()
753 int64_t max_vectorization_width() const { in max_vectorization_width() function in xla::cpu::__anondccdbc850111::TiledSmallGemmEmitter
754 return config().max_vectorization_width(); in max_vectorization_width()
783 max_vector_count() * max_vectorization_width(); in HandleResiduesOnN()
[all …]
H A Dtiled_dot_emitter.h42 int64_t max_vectorization_width, int64_t max_vector_count,