Searched refs:min_vectorization_width (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | tiled_dot_emitter.cc | 667 int64_t min_vectorization_width, int64_t tile_size_m, in Config() argument
673 min_vectorization_width_(min_vectorization_width), in Config()
680 "_", min_vectorization_width(), "_", tile_size_m(), in GetCacheKey()
688 int64_t min_vectorization_width() const { return min_vectorization_width_; } in min_vectorization_width() function in xla::cpu::__anondccdbc850111::TiledSmallGemmEmitter::Config
719 min_vectorization_width() > 0 && in TiledSmallGemmEmitter()
720 absl::has_single_bit(static_cast<uint64_t>(min_vectorization_width()))); in TiledSmallGemmEmitter()
721 CHECK_GE(max_vectorization_width(), min_vectorization_width()); in TiledSmallGemmEmitter()
757 int64_t min_vectorization_width() const { in min_vectorization_width() function in xla::cpu::__anondccdbc850111::TiledSmallGemmEmitter
758 return config().min_vectorization_width(); in min_vectorization_width()
788 current_vectorization_width >= min_vectorization_width()) { in HandleResiduesOnN()
[all …]
|
| H A D | tiled_dot_emitter.h | 43 int64_t min_vectorization_width, int64_t tile_size_m,
|