Searched defs:unroll_factors (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | kernel_creator.cc | 155 llvm::ArrayRef<int64_t> unroll_factors) { in TileLoops()
214 llvm::ArrayRef<int64_t> unroll_factors, in LowerTFToJITInvocation()
235 llvm::ArrayRef<int64_t> unroll_factors, in LowerTFtoLoops()
517 llvm::ArrayRef<int64_t> tile_sizes, llvm::ArrayRef<int64_t> unroll_factors, in GenerateKernelForTfCode()
|
| D | tf_to_kernel.cc | 110 llvm::ArrayRef<int64_t> unroll_factors, in Run()
179 llvm::cl::list<int64_t> unroll_factors( in main() local
|
| D | tf_framework_c_interface.cc | 145 llvm::SmallVectorImpl<int64_t>& unroll_factors, int64_t max_supported_rank, in Compile()
260 llvm::SmallVector<int64_t> unroll_factors = in _mlir_ciface_tf_jit_compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_to_jit_invocations.cc | 59 llvm::ArrayRef<int64_t> unroll_factors; member
210 llvm::ArrayRef<int64_t> unroll_factors; member
258 llvm::ArrayRef<int64_t> unroll_factors, in PopulateTFToJITInvocationPatterns()
269 llvm::ArrayRef<int64_t> tile_sizes, llvm::ArrayRef<int64_t> unroll_factors, in CreateTFToJITInvocationPass()
|
| D | tf_framework_legalize_to_llvm.cc | 325 std::pair<Value, Value> unroll_factors = in matchAndRewrite() local
|