Searched defs:cluster_op (Results 1 – 2 of 2) sorted by relevance
53 tf_device::ClusterOp cluster_op, SymbolTable* symbol_table, in BuildFunction()102 void OutlineCluster(tf_device::ClusterOp cluster_op, SymbolTable* symbol_table, in OutlineCluster()
950 if (auto cluster_op = dyn_cast<tf_device::ClusterOp>(op)) { in InferShapeForNonTFDialectOperation() local