Searched defs:tile_sizes (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| H A D | kernel_creator.cc | 111 llvm::ArrayRef<int64_t> tile_sizes, in LowerTFToJITInvocation()
140 Status LowerTFtoLoops(mlir::ModuleOp module, llvm::ArrayRef<int64_t> tile_sizes, in LowerTFtoLoops()
427 llvm::ArrayRef<int64_t> tile_sizes, llvm::ArrayRef<int64_t> unroll_factors, in GenerateKernelForTfCode()
|
| H A D | tf_to_kernel.cc | 109 llvm::ArrayRef<int64_t> tile_sizes, in Run()
179 llvm::cl::list<int64_t> tile_sizes( in main() local
|
| H A D | tf_framework_c_interface.cc | 152 llvm::SmallVectorImpl<int64_t>& tile_sizes, in Compile()
278 llvm::SmallVector<int64_t, 8> tile_sizes = in _mlir_ciface_tf_jit_compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | tf_to_jit_invocations.cc | 243 llvm::ArrayRef<int64_t> tile_sizes; member
292 llvm::ArrayRef<int64_t> tile_sizes, llvm::ArrayRef<int64_t> unroll_factors, in PopulateTFToJITInvocationPatterns()
309 llvm::ArrayRef<int64_t> tile_sizes, llvm::ArrayRef<int64_t> unroll_factors, in CreateTFToJITInvocationPass()
|
| H A D | tf_framework_legalize_to_llvm.cc | 268 std::pair<Value, Value> tile_sizes = in matchAndRewrite() local
|
| /aosp_15_r20/external/mesa3d/src/broadcom/common/ |
| H A D | v3d_util.c | 122 static const uint8_t tile_sizes[] = { in v3d_choose_tile_size() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | kernel_mapping_scheme.h | 49 TilingScheme(Vector3 dims_in_elems, Vector3 tile_sizes, Vector3 num_threads, in TilingScheme()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| H A D | tf_jitrt_tile_reduction.cc | 311 auto tile_sizes = mlir::linalg::computeTileSizes( in TileAndReshapeInputTensors() local
|