Searched defs:orig_arg_num (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 803 const xla::OpSharding& sharding, int orig_arg_num, DataType dtype, in CreateOrGetSplitNodesForInputSharding()
1008 const int orig_arg_num, DataType dtype, in CreateOrGetXlaSplitNodeForShardedPerReplicaArg()
1035 const int replica_id, const int orig_arg_num, DataType dtype, in CreateOrGetXlaSplitNodeForDistributedArg()
1065 const int replica_id, const int orig_arg_num, DataType dtype, in CreateOrGetXlaSplitNodeForVariableArg()
3078 const int orig_arg_num, DataType dtype, const string& device, Graph* graph, in CreatePartitionedDummyVarArgs()
3173 int64_t orig_arg_num = i + params_info.NumPerReplicaArgs() + in CreateOrGetPerHostVariableCopy() local
3199 int64_t orig_arg_num = i + params_info.NumPerReplicaArgs() + in CreateOrGetPerHostVariableCopy() local
3507 int64_t orig_arg_num = core_arg_nums[core][i]; in BuildExecuteNodes() local
3829 int orig_arg_num = in BuildExecuteNodes() local
|