Searched refs:max_vector_count (Results 1 – 2 of 2) sorted by relevance
657 int64 max_vectorization_width, int64 max_vector_count, in Config() argument663 max_vector_count_(max_vector_count), in Config()678 int64 max_vector_count() const { return max_vector_count_; } in max_vector_count() function in xla::cpu::__anon016ac5180111::TiledSmallGemmEmitter::Config707 CHECK_GT(max_vector_count(), 0); in TiledSmallGemmEmitter()745 int64 max_vector_count() const { return config().max_vector_count(); } in max_vector_count() function in xla::cpu::__anon016ac5180111::TiledSmallGemmEmitter772 max_vector_count() * max_vectorization_width(); in HandleResiduesOnN()773 int64 current_vector_count = max_vector_count(); in HandleResiduesOnN()1045 int64 max_vectorization_width, int64 max_vector_count, in EmitSmallGemm() argument1054 /*max_vector_count=*/max_vector_count, in EmitSmallGemm()
46 tensorflow::int64 max_vector_count,