Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Ddrop_while_shape_invariant.cc34 class DropWhileShapeInvariantInDeviceClusterPass class
36 DropWhileShapeInvariantInDeviceClusterPass> {
48 void DropWhileShapeInvariantInDeviceClusterPass::runOnOperation() { in runOnOperation()
62 return std::make_unique<DropWhileShapeInvariantInDeviceClusterPass>(); in CreateDropWhileShapeInvariantInDeviceClusterPass()
H A Dtf_passes.td1764 def DropWhileShapeInvariantInDeviceClusterPass : Pass<"tf-drop-while-shape-invariant-in-device-clus…