Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc1249 int input_index_to_concat_node = 0; in CreateConcatNodesForRetval() local
1254 inputs_to_sharded_retval.begin() + input_index_to_concat_node; in CreateConcatNodesForRetval()
1257 input_index_to_concat_node += num_splits; in CreateConcatNodesForRetval()