Searched refs:jit_i64_indexed_for_large_tensors (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | tf_to_jit_invocations.cc | 264 bool jit_i64_indexed_for_large_tensors) { in TFToJITInvocationPass()
271 jit_i64_indexed_for_large_tensors_ = jit_i64_indexed_for_large_tensors; in TFToJITInvocationPass()
294 bool cpu_codegen, bool jit_i64_indexed_for_large_tensors) { in PopulateTFToJITInvocationPatterns() argument
295 if (jit_i64_indexed_for_large_tensors) { in PopulateTFToJITInvocationPatterns()
302 jit_i64_indexed_for_large_tensors ? true : index_64bit; in PopulateTFToJITInvocationPatterns()
311 bool cpu_codegen, bool jit_i64_indexed_for_large_tensors) { in CreateTFToJITInvocationPass() argument
314 cpu_codegen, jit_i64_indexed_for_large_tensors); in CreateTFToJITInvocationPass()
|
| H A D | rewriters.h | 60 bool cpu_codegen, bool jit_i64_indexed_for_large_tensors);
|
| H A D | passes.h | 56 bool jit_i64_indexed_for_large_tensors = false);
|
| H A D | passes.td | 58 "jit_i64_indexed_for_large_tensors", "bool", /*default=*/"false",
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| H A D | build_defs.bzl | 155 "--jit_i64_indexed_for_large_tensors=%s" % ctx.attr.jit_i64_indexed_for_large_tensors,
188 "jit_i64_indexed_for_large_tensors": attr.bool(),
324 for (type, output_type, jit, jit_i64_indexed_for_large_tensors) in all_kernels:
360 jit_i64_indexed_for_large_tensors = jit_i64_indexed_for_large_tensors,
376 "--jit_i64_indexed_for_large_tensors=%s" % jit_i64_indexed_for_large_tensors,
413 for (type, output_type, jit, jit_i64_indexed_for_large_tensors) in all_kernels
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| H A D | kernel_creator.cc | 115 bool jit_i64_indexed_for_large_tensors, in LowerTFToJITInvocation() argument
123 index_64bit, jit_i64_indexed_for_large_tensors)); in LowerTFToJITInvocation()
143 bool index_64bit, bool jit_i64_indexed_for_large_tensors, in LowerTFtoLoops() argument
147 if (jit_i64_indexed_for_large_tensors) { in LowerTFtoLoops()
430 bool jit_i64_indexed_for_large_tensors, bool apply_cl_options) { in GenerateKernelForTfCode() argument
443 jit_i64_indexed_for_large_tensors, apply_cl_options)); in GenerateKernelForTfCode()
|
| H A D | tf_to_kernel.cc | 113 bool jit_i64_indexed_for_large_tensors) { in Run() argument
126 jit_i64_indexed_for_large_tensors, in Run()
186 llvm::cl::opt<bool> jit_i64_indexed_for_large_tensors( in main() local
202 enable_ftz, index_64bit, jit_compile, jit_i64_indexed_for_large_tensors); in main()
|
| H A D | kernel_creator.h | 49 bool jit_i64_indexed_for_large_tensors, bool apply_cl_options);
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/ |
| H A D | jit_i64_indexed_for_large_tensors.mlir | 5 // RUN: jit_i64_indexed_for_large_tensors=true" | \
|