Searched defs:max_supported_rank (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | kernel_creator.cc | 113 int64_t max_supported_rank, bool enable_ftz, in LowerTFToJITInvocation()
142 int64_t max_supported_rank, bool enable_ftz, in LowerTFtoLoops()
428 int64_t max_supported_rank, bool embed_memref_prints, bool print_ptx, in GenerateKernelForTfCode()
|
| D | tf_to_kernel.cc | 110 llvm::ArrayRef<int64_t> unroll_factors, int64_t max_supported_rank, in Run()
174 llvm::cl::opt<int64_t> max_supported_rank( in main() local
|
| D | tf_framework_c_interface.cc | 153 llvm::SmallVectorImpl<int64_t>& unroll_factors, int64_t max_supported_rank, in Compile()
243 int64_t* unroll_factors_ptr, int64_t max_supported_rank, bool enable_ftz, in _mlir_ciface_tf_jit_compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_to_jit_invocations.cc | 245 int64_t max_supported_rank; member
293 int64_t max_supported_rank, bool enable_ftz, bool index_64bit, in PopulateTFToJITInvocationPatterns()
310 int64_t max_supported_rank, bool enable_ftz, bool index_64bit, in CreateTFToJITInvocationPass()
|
| D | tf_framework_legalize_to_llvm.cc | 276 Value max_supported_rank = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite() local
|