Searched defs:orig_arg_num (Results 1 – 1 of 1) sorted by relevance
792 const xla::OpSharding& sharding, int orig_arg_num, DataType dtype, in CreateOrGetSplitNodesForInputSharding()2817 int64 orig_arg_num = in CreateOrGetPerHostVariableCopy() local2856 int64 orig_arg_num = i + params_info.NumPerReplicaArgs() + in CreateOrGetPerHostVariableCopy() local3115 int64 orig_arg_num = core_arg_nums[core][i]; in BuildExecuteNodes() local3383 int orig_arg_num = in BuildExecuteNodes() local