Searched refs:k_end (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 726 llvm::Value* k_start, llvm::Value* k_end,
732 llvm::Value* k_start, llvm::Value* k_end,
807 int64 k_end = dims().k() - (dims().k() % tile_size_k()); in HandleResiduesOnK() local
808 if (k_end != k_start) { in HandleResiduesOnK()
809 HandleResiduesOnM(vsl, tile_size_k(), GetInt64(k_start), GetInt64(k_end), in HandleResiduesOnK()
811 k_start = k_end; in HandleResiduesOnK()
822 llvm::Value* k_end, llvm::Value* n_start, llvm::Value* n_end) { in HandleResiduesOnM() argument
824 EmitTiledGemm(vsl, tile_size_k, k_start, k_end, n_start, n_end, tile_size_m(), in HandleResiduesOnM()
828 EmitTiledGemm(vsl, tile_size_k, k_start, k_end, n_start, n_end, in HandleResiduesOnM()
905 llvm::Value* k_end, llvm::Value* n_start, llvm::Value* n_end, in EmitTiledGemm() argument
[all …]
|