Searched defs:tile_size_k (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_options.cc | 86 int64 tile_size_k; in LlvmIrGemmTileSize() local
|
| D | tiled_dot_emitter.cc | 659 int64 tile_size_k) in Config()
682 int64 tile_size_k() const { return tile_size_k_; } in tile_size_k() function in xla::cpu::__anon016ac5180111::TiledSmallGemmEmitter::Config
750 int64 tile_size_k() const { return config().tile_size_k(); } in tile_size_k() function in xla::cpu::__anon016ac5180111::TiledSmallGemmEmitter
821 VectorSupportLibrary* vsl, int64 tile_size_k, llvm::Value* k_start, in HandleResiduesOnM()
904 VectorSupportLibrary* vsl, int64 tile_size_k, llvm::Value* k_start, in EmitTiledGemm()
1047 int64 tile_size_k, llvm::Value* lhs, llvm::Value* rhs, in EmitSmallGemm()
|
| D | dot_op_emitter.cc | 249 int64 tile_size_m, tile_size_k, tile_size_n_in_vector_width; in EmitTiledLlvmIrGemm() local
|