Home
last modified time | relevance | path

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

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