Searched defs:tile_sizes (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_fuse_linalg.cc | 48 llvm::ArrayRef<unsigned> tile_sizes) { in LhloFuseLinalgPass()
152 SmallVector<int64_t, 2> tile_sizes(tile_sizes_.begin(), in runOnFunction() local
194 bool tileGenericOp(LinalgOp op, ArrayRef<int64_t> tile_sizes, OpBuilder* b) { in tileGenericOp()
221 bool use_parallel_loops, ArrayRef<unsigned> tile_sizes) { in createLhloFuseLinalgPass()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_to_kernel.cc | 109 llvm::ArrayRef<int64_t> tile_sizes, in Run()
164 llvm::cl::list<int64_t> tile_sizes( in main() local
|
| D | kernel_creator.cc | 152 explicit TileLoops(llvm::ArrayRef<int64_t> tile_sizes, in TileLoops()
196 Status LowerTFtoLoops(mlir::ModuleOp module, llvm::ArrayRef<int64_t> tile_sizes, in LowerTFtoLoops()
432 llvm::ArrayRef<int64_t> tile_sizes, llvm::ArrayRef<int64_t> unroll_factors, in GenerateKernelForTfCode()
|
| /external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/ |
| D | kernel_lowering.h | 27 llvm::ArrayRef<unsigned> tile_sizes = {16, 64}; member
|
| /external/llvm-project/polly/lib/External/ppcg/ |
| D | gpu_hybrid.c | 121 int *tile_sizes) in gpu_hybrid_tile()
|
| D | hybrid.c | 677 __isl_keep isl_multi_val *tile_sizes, in compute_space_sizes()
880 __isl_keep isl_multi_val *tile_sizes) in ppcg_ht_bounds_construct_tiling()
|
| /external/mesa3d/src/broadcom/vulkan/ |
| D | v3dv_pass.c | 263 static const uint8_t tile_sizes[] = { in subpass_get_granularity() local
|
| D | v3dv_cmd_buffer.c | 451 static const uint8_t tile_sizes[] = { in job_compute_frame_tiling() local
|
| /external/mesa3d/src/gallium/drivers/v3d/ |
| D | v3d_job.c | 280 static const uint8_t tile_sizes[] = { in v3d_job_set_tile_buffer_size() local
|