Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc2239 std::vector<const Edge*> replicate_input_edges; in BuildDynamicShapeNodes() local
2240 TF_RETURN_IF_ERROR(replicate_node.input_edges(&replicate_input_edges)); in BuildDynamicShapeNodes()
2259 TF_RET_CHECK(i < replicate_input_edges.size()); in BuildDynamicShapeNodes()
2262 src = replicate_input_edges[i]->src(); in BuildDynamicShapeNodes()
2263 src_output = replicate_input_edges[i]->src_output(); in BuildDynamicShapeNodes()
2270 input_num < replicate_input_edges.size()); in BuildDynamicShapeNodes()
2271 src = replicate_input_edges[input_num]->src(); in BuildDynamicShapeNodes()
2272 src_output = replicate_input_edges[input_num]->src_output(); in BuildDynamicShapeNodes()
2906 std::vector<const Edge*> replicate_input_edges; in BuildExecuteNodes() local
2907 TF_RETURN_IF_ERROR(replicate_node.input_edges(&replicate_input_edges)); in BuildExecuteNodes()
[all …]