Searched refs:tile_size_k (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 659 int64 tile_size_k) in Config() argument
666 tile_size_k_(tile_size_k) {} in Config()
672 "_", tile_size_k()); in GetCacheKey()
682 int64 tile_size_k() const { return tile_size_k_; } in tile_size_k() function in xla::cpu::__anon016ac5180111::TiledSmallGemmEmitter::Config
711 CHECK_GT(tile_size_k(), 0); in TiledSmallGemmEmitter()
725 void HandleResiduesOnM(VectorSupportLibrary* vsl, int64 tile_size_k,
731 void EmitTiledGemm(VectorSupportLibrary* vsl, int64 tile_size_k,
750 int64 tile_size_k() const { return config().tile_size_k(); } in tile_size_k() function in xla::cpu::__anon016ac5180111::TiledSmallGemmEmitter
807 int64 k_end = dims().k() - (dims().k() % tile_size_k()); in HandleResiduesOnK()
809 HandleResiduesOnM(vsl, tile_size_k(), GetInt64(k_start), GetInt64(k_end), in HandleResiduesOnK()
[all …]
|
| D | cpu_options.cc | 86 int64 tile_size_k; in LlvmIrGemmTileSize() local
90 CHECK(absl::SimpleAtoi(tile_components[1], &tile_size_k)); in LlvmIrGemmTileSize()
98 return std::tuple<int64, int64, int64>(tile_size_m, tile_size_k, in LlvmIrGemmTileSize()
|
| D | tiled_dot_emitter.h | 48 tensorflow::int64 tile_size_m, tensorflow::int64 tile_size_k,
|
| D | dot_op_emitter.cc | 249 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()
259 /*tile_size_m=*/tile_size_m, /*tile_size_k=*/tile_size_k, /*lhs=*/lhs, in EmitTiledLlvmIrGemm()
|