Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc898 ShardedInputInfo sharded_input_info{root_split_node, in CreateOrGetSplitNodesForInputSharding() local
900 (*arg_index_to_sharded_input_map)[input_index] = sharded_input_info; in CreateOrGetSplitNodesForInputSharding()
901 return sharded_input_info; in CreateOrGetSplitNodesForInputSharding()
3170 ShardedInputInfo sharded_input_info, in BuildExecuteNodes()
3177 sharded_input_info.sharded_inputs.at(core); in BuildExecuteNodes()
3253 ShardedInputInfo sharded_input_info, in BuildExecuteNodes()
3261 sharded_input_info.sharded_inputs[core]; in BuildExecuteNodes()