Searched defs:orig_src_output (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 728 const int orig_src_output, DataType dtype, in CreateSplitNode()
805 int orig_src_output, Node* orig_src, Node* control_predecessor, in CreateOrGetSplitNodesForInputSharding()
1010 const int orig_src_output, Graph* graph, in CreateOrGetXlaSplitNodeForShardedPerReplicaArg()
1037 const int orig_src_output, Graph* graph, in CreateOrGetXlaSplitNodeForDistributedArg()
1067 const int orig_src_output, Graph* graph, in CreateOrGetXlaSplitNodeForVariableArg()
|