Searched refs:xla_cluster (Results 1 – 5 of 5) sorted by relevance
115 NodeBuilder& NodeBuilder::XlaCluster(StringPiece xla_cluster) { in XlaCluster() argument116 def_builder_.Attr("_XlaCluster", xla_cluster); in XlaCluster()
109 NodeBuilder& XlaCluster(StringPiece xla_cluster);
253 const string& xla_cluster) in Impl() argument266 xla_cluster_(xla_cluster), in Impl()457 Scope Scope::WithXlaCluster(const string& xla_cluster) const { in WithXlaCluster()458 return Scope(new Impl(*this, Impl::Tags::XlaCluster(), xla_cluster)); in WithXlaCluster()
148 Scope WithXlaCluster(const string& xla_cluster) const;
82 Impl(const Scope& other, Tags::XlaCluster, const string& xla_cluster);