Home
last modified time | relevance | path

Searched refs:unroll_factors (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mlir_generated/
DBUILD487 unroll_factors = "4",
510 # unroll_factors = "4",
525 unroll_factors = "2",
558 unroll_factors = "4",
569 unroll_factors = "4",
580 unroll_factors = "2",
592 # unroll_factors = "4",
604 unroll_factors = "4",
616 unroll_factors = "4",
643 unroll_factors = "4",
[all …]
Dbuild_defs.bzl132 if ctx.attr.unroll_factors:
133 cmd_args.append("--unroll_factors=%s" % ctx.attr.unroll_factors)
174 "unroll_factors": attr.string(),
205 unroll_factors = None,
222 unroll_factors: The unrolling specification, e.g. "4,4"
255 unroll_factors = unroll_factors,
Dbuild_test.sh28 ${TF_TO_KERNEL} --input=${INPUT} --output=${OUTPUT_FILE} --unroll_factors=4 --tile_sizes=256 ${PLAT…
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dkernel_creator.cc153 llvm::ArrayRef<int64_t> unroll_factors) { in TileLoops() argument
159 if (unroll_factors.size() == tile_sizes.size()) { in TileLoops()
160 inner_tile_ = llvm::to_vector<4>(unroll_factors); in TileLoops()
161 for (auto en : llvm::enumerate(unroll_factors)) { in TileLoops()
197 llvm::ArrayRef<int64_t> unroll_factors, in LowerTFtoLoops() argument
249 std::make_unique<TileLoops>(tile_sizes, unroll_factors)); in LowerTFtoLoops()
432 llvm::ArrayRef<int64_t> tile_sizes, llvm::ArrayRef<int64_t> unroll_factors, in GenerateKernelForTfCode() argument
448 LowerTFtoLoops(module.get(), tile_sizes, unroll_factors, cpu_codegen)); in GenerateKernelForTfCode()
Dtf_to_kernel.cc110 llvm::ArrayRef<int64_t> unroll_factors, in Run() argument
122 unroll_factors, embed_memref_prints, in Run()
167 llvm::cl::list<int64_t> unroll_factors( in main() local
179 input_file, output_file, architectures, tile_sizes, unroll_factors, in main()
Dkernel_creator.h41 llvm::ArrayRef<int64_t> unroll_factors = {},
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dkernel_lowering.cc59 if (!options.unroll_factors.empty()) { in LowerLHLOToGPU()
61 for (auto pair : llvm::zip(options.tile_sizes, options.unroll_factors)) { in LowerLHLOToGPU()
104 if (!options.unroll_factors.empty()) { in LowerLHLOToGPU()
125 if (!options.unroll_factors.empty()) { in LowerLHLOToGPU()
Dkernel_lowering.h28 llvm::ArrayRef<unsigned> unroll_factors = {}; member
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/tf_to_kernel/
Dtanh.mlir1 // RUN: tf_to_kernel --input=%s --output=%t --unroll_factors=4 --tile_sizes=256 --arch=sm_70
Dadd_v2.mlir1 // RUN: tf_to_kernel --input=%s --output=%t --unroll_factors=4 --tile_sizes=256 --arch=sm_70