Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
H A Dpropagate_static_shapes_to_kernel.cc72 class PropagateStaticShapesToKernelPass class
74 PropagateStaticShapesToKernelPass> {
76 explicit PropagateStaticShapesToKernelPass(Type pointerType) in PropagateStaticShapesToKernelPass() function in mlir::__anonaa7420f50111::PropagateStaticShapesToKernelPass
209 void PropagateStaticShapesToKernelPass::runOnOperation() { in runOnOperation()
232 return std::make_unique<PropagateStaticShapesToKernelPass>(pointerType); in createPropagateStaticShapesToKernelPass()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Transforms/
H A Dpasses.td142 def PropagateStaticShapesToKernelPass : Pass<"propagate-static-shapes", "ModuleOp"> {