Home
last modified time | relevance | path

Searched defs:unroll_factors (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
H A Dkernel_creator.cc112 llvm::ArrayRef<int64_t> unroll_factors, in LowerTFToJITInvocation()
141 llvm::ArrayRef<int64_t> unroll_factors, in LowerTFtoLoops()
427 llvm::ArrayRef<int64_t> tile_sizes, llvm::ArrayRef<int64_t> unroll_factors, in GenerateKernelForTfCode()
H A Dtf_to_kernel.cc110 llvm::ArrayRef<int64_t> unroll_factors, int64_t max_supported_rank, in Run()
182 llvm::cl::list<int64_t> unroll_factors( in main() local
H A Dtf_framework_c_interface.cc153 llvm::SmallVectorImpl<int64_t>& unroll_factors, int64_t max_supported_rank, in Compile()
280 llvm::SmallVector<int64_t, 8> unroll_factors = in _mlir_ciface_tf_jit_compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dtf_to_jit_invocations.cc244 llvm::ArrayRef<int64_t> unroll_factors; 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 Dtf_framework_legalize_to_llvm.cc272 std::pair<Value, Value> unroll_factors = in matchAndRewrite() local