tensorflow/core/ops/compat/ops_history_v1/LoadTPUEmbeddingADAMParameters.pbtxt

op {
  name: "LoadTPUEmbeddingADAMParameters"
  input_arg {
    name: "parameters"
    type: DT_FLOAT
  }
  input_arg {
    name: "momenta"
    type: DT_FLOAT
  }
  input_arg {
    name: "velocities"
    type: DT_FLOAT
  }
  attr {
    name: "table_id"
    type: "int"
    default_value {
      i: -1
    }
    has_minimum: true
    minimum: -1
  }
  attr {
    name: "table_name"
    type: "string"
    default_value {
      s: ""
    }
  }
  attr {
    name: "num_shards"
    type: "int"
  }
  attr {
    name: "shard_id"
    type: "int"
  }
  is_stateful: true
}