Searched refs:is_first_column (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | tiled_dot_emitter.cc | 254 bool is_first_column); 277 int64 columns, bool is_first_column); 293 llvm::Value* column, int64 column_count, bool is_first_column) { in EmitOuterLoopBody() argument 300 /*columns=*/column_count, is_first_column); in EmitOuterLoopBody() 301 EmitInnerLoopEpilogue(column, /*columns=*/column_count, is_first_column); in EmitOuterLoopBody() 311 [&](llvm::Value* column, bool is_first_column) { in Emit() argument 312 EmitOuterLoopBody(column, tile_cols(), is_first_column); in Emit() 323 int64 columns, bool is_first_column) { in EmitInnerLoopTiled() argument 331 is_first_column ? (addend_ ? vsl_.LoadVector(addend_, row) in EmitInnerLoopTiled()
|