Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dtf_to_jit_invocations.cc254 struct TFToJITInvocationPass struct
255 : public TFToJITInvocationPassBase<TFToJITInvocationPass> {
260 explicit TFToJITInvocationPass(llvm::ArrayRef<int64_t> tile_sizes, in TFToJITInvocationPass() function
312 return std::make_unique<TFToJITInvocationPass>( in CreateTFToJITInvocationPass()
H A Dpasses.td41 def TFToJITInvocationPass : Pass<"tf-to-jit-invocation", "mlir::func::FuncOp"> {