Searched refs:EmitOuterLoopBody (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 253 void EmitOuterLoopBody(llvm::Value* column, int64 column_count,
292 void ColumnMajorMatrixVectorProductEmitter::EmitOuterLoopBody( in EmitOuterLoopBody() function in xla::cpu::__anon016ac5180111::ColumnMajorMatrixVectorProductEmitter
312 EmitOuterLoopBody(column, tile_cols(), is_first_column); in Emit()
316 EmitOuterLoopBody(b_->getInt64(column_limit), column_remainder, in Emit()
485 void EmitOuterLoopBody(llvm::Value* row, int64 row_count);
503 void RowMajorMatrixVectorProductEmitter::EmitOuterLoopBody(llvm::Value* row, in EmitOuterLoopBody() function in xla::cpu::__anon016ac5180111::RowMajorMatrixVectorProductEmitter
555 [&](llvm::Value* row) { EmitOuterLoopBody(row, tile_rows()); }); in Emit()
558 EmitOuterLoopBody(b_->getInt64(row_limit), row_remainder); in Emit()
|