Searched refs:tile_origin_index (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2316 const IrArray::Index& tile_origin_index, in EmitFullElementalTile() argument
2331 tile_origin_index.AddOffsetToDim(y, KernelMappingScheme::DimY, builder) in EmitFullElementalTile()
2354 const IrArray::Index& tile_origin_index, in EmitPartialElementalTile() argument
2370 tile_origin_index.AddOffsetToDim(y, KernelMappingScheme::DimY, builder) in EmitPartialElementalTile()
2419 const IrArray::Index& tile_origin_index, const string& loop_name, in EmitTiledElementalCodeWithBoundsCheck() argument
2435 EmitFullElementalTile(mapping_scheme, tile_origin_index, loop_name, ksl, in EmitTiledElementalCodeWithBoundsCheck()
2439 EmitPartialElementalTile(mapping_scheme, tile_origin_index, loop_name, in EmitTiledElementalCodeWithBoundsCheck()
|