Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc109 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph() function
230 kXlaClusterIdAttr, **graph, RewriteSubgraph, in Encapsulate()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc76 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph() function
2533 kTPUReplicateAttr, **graph, RewriteSubgraph, in Encapsulate()