Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc276 const std::vector<llvm::Value*>& rhs_tile,
297 std::vector<llvm::Value*> rhs_tile = in EmitOuterLoopBody() local
299 EmitInnerLoopTiled(&lhs_memory_tile, rhs_tile, in EmitOuterLoopBody()
322 MemoryTile* lhs_memory_tile, const std::vector<llvm::Value*>& rhs_tile, in EmitInnerLoopTiled() argument
335 accumulator = vsl_.MulAdd(lhs_tile[i], rhs_tile[i], accumulator); in EmitInnerLoopTiled()
924 std::vector<llvm::Value*> rhs_tile = rhs_memory_tile.LoadTile(n_i); in EmitTiledGemm() local
929 vsl->MulAdd(lhs_tile[r_m_i][r_k_i], rhs_tile[r_k_i], in EmitTiledGemm()