Home
last modified time | relevance | path

Searched defs:cluster_op (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcluster_outlining.cc103 void OutlineCluster(tf_device::ClusterOp cluster_op, SymbolTable* symbol_table, in OutlineCluster()
Dcluster_ops_by_policy.cc569 auto cluster_op = in CreateClusterOp() local
Dshape_inference.cc1162 if (auto cluster_op = dyn_cast<tf_device::ClusterOp>(op)) { in InferShapeForNonTFDialectOperation() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dlower_saved_model.cc117 mlir::Operation *cluster_op = in ReplaceHoistedValues() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Drank_specialization.cc128 auto cluster_op = rewriter.create<chlo::RankSpecializationClusterOp>( in matchAndRewrite() local