Searched refs:max_vectorization_width (Results 1 – 2 of 2) sorted by relevance
657 int64 max_vectorization_width, int64 max_vector_count, in Config() argument662 max_vectorization_width_(max_vectorization_width), in Config()670 dims().ToString(), "_", max_vectorization_width(), in GetCacheKey()677 int64 max_vectorization_width() const { return max_vectorization_width_; } in max_vectorization_width() function in xla::cpu::__anon1dc5b04f0111::TiledSmallGemmEmitter::Config705 CHECK(max_vectorization_width() > 0 && in TiledSmallGemmEmitter()706 IsPowerOfTwo(static_cast<uint64>(max_vectorization_width()))); in TiledSmallGemmEmitter()710 CHECK_GE(max_vectorization_width(), min_vectorization_width()); in TiledSmallGemmEmitter()742 int64 max_vectorization_width() const { in max_vectorization_width() function in xla::cpu::__anon1dc5b04f0111::TiledSmallGemmEmitter743 return config().max_vectorization_width(); in max_vectorization_width()772 max_vector_count() * max_vectorization_width(); in HandleResiduesOnN()[all …]
45 tensorflow::int64 max_vectorization_width,