Searched refs:tiling_scheme (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 3355 static llvm::Value* GetStartOffsetX(const TilingScheme& tiling_scheme, in GetStartOffsetX() argument
3359 int64_t multiplier = tiling_scheme.GetIndexingOrder() == kStridedIndexingX in GetStartOffsetX()
3360 ? tiling_scheme.GetVectorSize() in GetStartOffsetX()
3361 : tiling_scheme.GetTileSizeFor(kDimX); in GetStartOffsetX()
3380 const TilingScheme& tiling_scheme, bool check_x_tile_bounds, in EmitXTileLoop() argument
3392 int64_t vector_size = tiling_scheme.GetVectorSize(); in EmitXTileLoop()
3393 int64_t stride_x = tiling_scheme.GetIndexingOrder() == kLinearIndexingX in EmitXTileLoop()
3395 : tiling_scheme.GetNumThreadsFor(kDimX); in EmitXTileLoop()
3400 /*end=*/constant(tiling_scheme.GetTileSizeFor(kDimX) / vector_size), in EmitXTileLoop()
3428 const TilingScheme& tiling_scheme, const IrArray::Index& tile_origin_index, in EmitTile() argument
[all …]
|
| H A D | ir_emitter_unnested.h | 532 const TilingScheme& tiling_scheme,
555 const TilingScheme& tiling_scheme, llvm::Type* index_ty,
583 const TilingScheme& tiling_scheme,
597 const llvm_ir::IrArray::Index& index, const TilingScheme& tiling_scheme,
672 const TilingScheme& tiling_scheme, llvm::Type* element_type,
732 StatusOr<ThreadIdInfo> EmitThreadIdInfo(const TilingScheme& tiling_scheme,
|