Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc324 int64 row_limit = m() - (m() % tile_rows()); in EmitInnerLoopTiled() local
326 ksl_.For("dot.inner.tiled", /*start=*/0, /*end=*/row_limit, in EmitInnerLoopTiled()
551 int64 row_limit = m() - row_remainder; in Emit() local
554 /*start=*/0, /*end=*/row_limit, /*step=*/tile_rows(), in Emit()
558 EmitOuterLoopBody(b_->getInt64(row_limit), row_remainder); in Emit()