Home
last modified time | relevance | path

Searched refs:TFToI64JITInvocationForLargeTensorsPattern (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dtf_to_jit_invocations.cc114 struct TFToI64JITInvocationForLargeTensorsPattern : public RewritePattern { struct
115 explicit TFToI64JITInvocationForLargeTensorsPattern(MLIRContext *ctx) in TFToI64JITInvocationForLargeTensorsPattern() function
296 patterns->add<TFToI64JITInvocationForLargeTensorsPattern>(ctx); in PopulateTFToJITInvocationPatterns()