Home
last modified time | relevance | path

Searched defs:jit_i64_indexed_for_large_tensors (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
H A Dkernel_creator.cc115 bool jit_i64_indexed_for_large_tensors, in LowerTFToJITInvocation()
143 bool index_64bit, bool jit_i64_indexed_for_large_tensors, in LowerTFtoLoops()
430 bool jit_i64_indexed_for_large_tensors, bool apply_cl_options) { in GenerateKernelForTfCode()
H A Dtf_to_kernel.cc113 bool jit_i64_indexed_for_large_tensors) { in Run()
186 llvm::cl::opt<bool> jit_i64_indexed_for_large_tensors( in main() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dtf_to_jit_invocations.cc294 bool cpu_codegen, bool jit_i64_indexed_for_large_tensors) { in PopulateTFToJITInvocationPatterns()
311 bool cpu_codegen, bool jit_i64_indexed_for_large_tensors) { in CreateTFToJITInvocationPass()