Searched defs:orig_src_output (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 727 const int orig_src_output, DataType dtype, in CreateSplitNode()
807 int orig_src_output, Node* orig_src, Node* control_predecessor, in CreateOrGetSplitNodesForInputSharding()
1014 const int orig_src_output, Graph* graph, in CreateOrGetXlaSplitNodeForShardedPerReplicaArg()
1041 const int orig_src_output, Graph* graph, in CreateOrGetXlaSplitNodeForDistributedArg()
1071 const int orig_src_output, Graph* graph, in CreateOrGetXlaSplitNodeForVariableArg()
|