Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_options.cc87 int64 tile_size_n_in_vector_width; in LlvmIrGemmTileSize() local
96 &tile_size_n_in_vector_width)); in LlvmIrGemmTileSize()
99 tile_size_n_in_vector_width); in LlvmIrGemmTileSize()
Ddot_op_emitter.cc249 int64 tile_size_m, tile_size_k, tile_size_n_in_vector_width; in EmitTiledLlvmIrGemm() local
250 std::tie(tile_size_m, tile_size_k, tile_size_n_in_vector_width) = in EmitTiledLlvmIrGemm()
257 /*max_vector_count=*/tile_size_n_in_vector_width, in EmitTiledLlvmIrGemm()