コード例 #1
0
        public void MergeFrom(pb::CodedInputStream input)
        {
            uint tag;

            while ((tag = input.ReadTag()) != 0)
            {
                switch (tag)
                {
                default:
                    _unknownFields = pb::UnknownFieldSet.MergeFieldFrom(_unknownFields, input);
                    break;

                case 10: {
                    tableDescriptor_.AddEntriesFrom(input, _repeated_tableDescriptor_codec);
                    break;
                }

                case 16: {
                    Mode = (global::Tensorflow.Tpu.TPUEmbeddingConfiguration.Types.Mode)input.ReadEnum();
                    break;
                }

                case 24: {
                    BatchSizePerTensorCore = input.ReadInt32();
                    break;
                }

                case 32: {
                    NumHosts = input.ReadInt32();
                    break;
                }

                case 40: {
                    NumTensorCores = input.ReadInt32();
                    break;
                }

                case 48: {
                    ShardingStrategy = (global::Tensorflow.Tpu.TPUEmbeddingConfiguration.Types.ShardingStrategy)input.ReadEnum();
                    break;
                }

                case 56: {
                    PipelineExecutionWithTensorCore = input.ReadBool();
                    break;
                }

                case 66: {
                    if (outputLayout_ == null)
                    {
                        OutputLayout = new global::Tensorflow.Tpu.TPUEmbeddingOutputLayout();
                    }
                    input.ReadMessage(OutputLayout);
                    break;
                }
                }
            }
        }
コード例 #2
0
 public void MergeFrom(TPUEmbeddingConfiguration other)
 {
     if (other == null)
     {
         return;
     }
     tableDescriptor_.Add(other.tableDescriptor_);
     if (other.Mode != 0)
     {
         Mode = other.Mode;
     }
     if (other.BatchSizePerTensorCore != 0)
     {
         BatchSizePerTensorCore = other.BatchSizePerTensorCore;
     }
     if (other.NumHosts != 0)
     {
         NumHosts = other.NumHosts;
     }
     if (other.NumTensorCores != 0)
     {
         NumTensorCores = other.NumTensorCores;
     }
     if (other.ShardingStrategy != 0)
     {
         ShardingStrategy = other.ShardingStrategy;
     }
     if (other.PipelineExecutionWithTensorCore != false)
     {
         PipelineExecutionWithTensorCore = other.PipelineExecutionWithTensorCore;
     }
     if (other.outputLayout_ != null)
     {
         if (outputLayout_ == null)
         {
             OutputLayout = new global::Tensorflow.Tpu.TPUEmbeddingOutputLayout();
         }
         OutputLayout.MergeFrom(other.OutputLayout);
     }
     _unknownFields = pb::UnknownFieldSet.MergeFrom(_unknownFields, other._unknownFields);
 }