Split ops_history file into a directory with one file per op.

PiperOrigin-RevId: 260550077
This commit is contained in:
A. Unique TensorFlower 2019-07-29 12:03:36 -07:00 committed by TensorFlower Gardener
parent a3c0837d85
commit 7b372c7da2
1211 changed files with 95101 additions and 65 deletions
tensorflow/core/ops/compat
BUILDop_compatibility_lib.ccop_compatibility_lib.h
ops_history_v1
Abort.pbtxtAbs.pbtxtAccumulateNV2.pbtxtAccumulatorApplyGradient.pbtxtAccumulatorNumAccumulated.pbtxtAccumulatorSetGlobalStep.pbtxtAccumulatorTakeGradient.pbtxtAcos.pbtxtAcosh.pbtxtAdd.pbtxtAddManySparseToTensorsMap.pbtxtAddN.pbtxtAddSparseToTensorsMap.pbtxtAddV2.pbtxtAdjustContrast.pbtxtAdjustContrastv2.pbtxtAdjustHue.pbtxtAdjustSaturation.pbtxtAll.pbtxtAllCandidateSampler.pbtxtAllToAll.pbtxtAngle.pbtxtAnonymousIterator.pbtxtAnonymousIteratorV2.pbtxtAnonymousMultiDeviceIterator.pbtxtAny.pbtxtApplyAdaMax.pbtxtApplyAdadelta.pbtxtApplyAdagrad.pbtxtApplyAdagradDA.pbtxtApplyAdam.pbtxtApplyAddSign.pbtxtApplyCenteredRMSProp.pbtxtApplyFtrl.pbtxtApplyFtrlV2.pbtxtApplyGradientDescent.pbtxtApplyMomentum.pbtxtApplyPowerSign.pbtxtApplyProximalAdagrad.pbtxtApplyProximalGradientDescent.pbtxtApplyRMSProp.pbtxtApproximateEqual.pbtxtArgMax.pbtxtArgMin.pbtxtAsString.pbtxtAsin.pbtxtAsinh.pbtxtAssert.pbtxtAssertNextDataset.pbtxtAssign.pbtxtAssignAdd.pbtxtAssignAddVariableOp.pbtxtAssignSub.pbtxtAssignSubVariableOp.pbtxtAssignVariableOp.pbtxtAtan.pbtxtAtan2.pbtxtAtanh.pbtxtAudioSpectrogram.pbtxtAudioSummary.pbtxtAudioSummaryV2.pbtxtAutoShardDataset.pbtxtAvgPool.pbtxtAvgPool3D.pbtxtAvgPool3DGrad.pbtxtAvgPoolGrad.pbtxtBarrier.pbtxtBarrierClose.pbtxtBarrierIncompleteSize.pbtxtBarrierInsertMany.pbtxtBarrierReadySize.pbtxtBarrierTakeMany.pbtxtBatch.pbtxtBatchCholesky.pbtxtBatchCholeskyGrad.pbtxtBatchDataset.pbtxtBatchDatasetV2.pbtxtBatchFFT.pbtxtBatchFFT2D.pbtxtBatchFFT3D.pbtxtBatchFunction.pbtxtBatchIFFT.pbtxtBatchIFFT2D.pbtxtBatchIFFT3D.pbtxtBatchMatMul.pbtxtBatchMatMulV2.pbtxtBatchMatrixBandPart.pbtxtBatchMatrixDeterminant.pbtxtBatchMatrixDiag.pbtxtBatchMatrixDiagPart.pbtxtBatchMatrixInverse.pbtxtBatchMatrixSetDiag.pbtxtBatchMatrixSolve.pbtxtBatchMatrixSolveLs.pbtxtBatchMatrixTriangularSolve.pbtxtBatchNormWithGlobalNormalization.pbtxtBatchNormWithGlobalNormalizationGrad.pbtxt

View File

@ -34,11 +34,11 @@ tf_cc_test(
size = "small",
srcs = ["backwards_compatibility_test.cc"],
data = [
":ops_history.v0.pbtxt",
":ops_history.v1.pbtxt",
":ops_history.v2.pbtxt",
"//tensorflow/core:ops/ops.pbtxt",
],
] + glob([
"ops_history_v*/*.pbtxt",
"ops_history.v*.pbtxt",
]),
deps = [
":op_compatibility_lib",
"//tensorflow/core:framework",

View File

@ -27,17 +27,83 @@ limitations under the License.
namespace tensorflow {
static string OpsHistoryDirectory(const string& ops_prefix,
const string& history_version) {
return io::JoinPath(ops_prefix,
strings::StrCat("compat/ops_history_", history_version));
}
static string OpsHistoryFile(const string& ops_prefix,
const string& history_version) {
return io::JoinPath(ops_prefix, strings::StrCat("compat/ops_history.",
history_version, ".pbtxt"));
}
static string FileNameFromOpName(const string& op_name) {
return strings::StrCat(op_name, ".pbtxt");
}
static void AddNewOpToHistory(const OpDef& op,
OpCompatibilityLib::OpHistory* out_op_history) {
if (out_op_history != nullptr) {
out_op_history->emplace_back(FileNameFromOpName(op.name()), OpList());
*out_op_history->back().second.add_op() = op;
}
}
static Status ReadOpHistory(Env* env, const string& file,
const string& directory,
OpCompatibilityLib::OpHistory* out) {
// Read op history form `directory` if it exists there.
std::vector<string> matching_files;
Status status = env->GetMatchingPaths(io::JoinPath(directory, "*.pbtxt"),
&matching_files);
if (status.ok() && !matching_files.empty()) {
printf("Reading op history from %s/*.pbtxt...\n", directory.c_str());
std::sort(matching_files.begin(), matching_files.end());
for (const string& full_file : matching_files) {
string op_history_str;
TF_RETURN_IF_ERROR(ReadFileToString(env, full_file, &op_history_str));
OpList in_op_history;
protobuf::TextFormat::ParseFromString(op_history_str, &in_op_history);
const string file_tail = FileNameFromOpName(in_op_history.op(0).name());
const string expected = io::JoinPath(directory, file_tail);
if (full_file != expected) {
return errors::Internal("Expected file paths to match but '", full_file,
"' != '", expected, "'");
}
out->emplace_back(file_tail, in_op_history);
}
} else { // Otherwise, fall back to reading op history from `file`.
printf("Reading op history from %s...\n", file.c_str());
string op_history_str;
TF_RETURN_IF_ERROR(ReadFileToString(env, file, &op_history_str));
OpList in_op_history;
protobuf::TextFormat::ParseFromString(op_history_str, &in_op_history);
// Convert from a linear OpList to OpHistory format with one OpList per
// unique op name.
int start = 0;
while (start < in_op_history.op_size()) {
int end = start + 1;
while (end < in_op_history.op_size() &&
in_op_history.op(start).name() == in_op_history.op(end).name()) {
++end;
}
AddNewOpToHistory(in_op_history.op(start), out);
for (++start; start < end; ++start) {
*out->back().second.add_op() = in_op_history.op(start);
}
}
}
return Status::OK();
}
OpCompatibilityLib::OpCompatibilityLib(const string& ops_prefix,
const string& history_version,
const std::set<string>* stable_ops)
: ops_file_(io::JoinPath(ops_prefix, "ops.pbtxt")),
op_history_file_(OpsHistoryFile(ops_prefix, history_version)),
op_history_directory_(OpsHistoryDirectory(ops_prefix, history_version)),
stable_ops_(stable_ops) {
// Get the sorted list of all registered OpDefs.
printf("Getting all registered ops...\n");
@ -46,7 +112,7 @@ OpCompatibilityLib::OpCompatibilityLib(const string& ops_prefix,
Status OpCompatibilityLib::ValidateCompatible(Env* env, int* changed_ops,
int* added_ops,
OpList* out_op_history) {
OpHistory* out_op_history) {
*changed_ops = 0;
*added_ops = 0;
@ -78,104 +144,90 @@ Status OpCompatibilityLib::ValidateCompatible(Env* env, int* changed_ops,
}
}
OpList in_op_history;
{ // Read op history.
printf("Reading op history from %s...\n", op_history_file_.c_str());
string op_history_str;
TF_RETURN_IF_ERROR(
ReadFileToString(env, op_history_file_, &op_history_str));
protobuf::TextFormat::ParseFromString(op_history_str, &in_op_history);
}
OpHistory in_op_history;
TF_RETURN_IF_ERROR(ReadOpHistory(env, op_history_file_, op_history_directory_,
&in_op_history));
int cur = 0;
int start = 0;
int hist = 0;
printf("Verifying updates are compatible...\n");
// Note: Op history is in (alphabetical, oldest-first) order.
while (cur < op_list_.op_size() && start < in_op_history.op_size()) {
const string& op_name = op_list_.op(cur).name();
if (stable_ops_ != nullptr && stable_ops_->count(op_name) == 0) {
// Note: Op history is one OpList per unique op name in alphabetical order.
// Within the OplList it has versions in oldest-first order.
while (cur < op_list_.op_size() && hist < in_op_history.size()) {
const OpDef& cur_op = op_list_.op(cur);
const string& cur_op_name = cur_op.name();
const OpList& history_op_list = in_op_history[hist].second;
const string& history_op_name = history_op_list.op(0).name();
if (stable_ops_ != nullptr && stable_ops_->count(cur_op_name) == 0) {
// Ignore unstable op.
for (++cur; cur < op_list_.op_size(); ++cur) {
if (op_list_.op(cur).name() != op_name) break;
if (op_list_.op(cur).name() != cur_op_name) break;
}
} else if (op_name < in_op_history.op(start).name()) {
} else if (cur_op_name < history_op_name) {
// New op: add it.
if (out_op_history != nullptr) {
*out_op_history->add_op() = op_list_.op(cur);
}
AddNewOpToHistory(cur_op, out_op_history);
++*added_ops;
++cur;
} else if (op_name > in_op_history.op(start).name()) {
} else if (cur_op_name > history_op_name) {
if (stable_ops_ != nullptr) {
// Okay to remove ops from the history that have been made unstable.
for (++start; start < in_op_history.op_size(); ++start) {
if (op_name <= in_op_history.op(start).name()) break;
}
++hist;
} else {
// Op removed: error.
return errors::InvalidArgument("Error, removed op: ",
SummarizeOpDef(in_op_history.op(start)));
SummarizeOpDef(history_op_list.op(0)));
}
} else {
// Op match.
// Find all historical version of this op.
int end = start + 1;
for (; end < in_op_history.op_size(); ++end) {
if (in_op_history.op(end).name() != op_name) break;
}
if (out_op_history != nullptr) {
// Copy from in_op_history to *out_op_history.
for (int i = start; i < end; ++i) {
*out_op_history->add_op() = in_op_history.op(i);
}
out_op_history->push_back(in_op_history[hist]);
}
const int end = history_op_list.op_size();
// Is the last op in the history the same as the current op?
// Compare using their serialized representations.
string history_str, cur_str;
in_op_history.op(end - 1).SerializeToString(&history_str);
op_list_.op(cur).SerializeToString(&cur_str);
history_op_list.op(end - 1).SerializeToString(&history_str);
cur_op.SerializeToString(&cur_str);
if (history_str != cur_str) {
// Op changed, verify the change is compatible.
for (int i = start; i < end; ++i) {
TF_RETURN_IF_ERROR(
OpDefCompatible(in_op_history.op(i), op_list_.op(cur)));
for (int i = 0; i < end; ++i) {
TF_RETURN_IF_ERROR(OpDefCompatible(history_op_list.op(i), cur_op));
}
// Verify default value of attrs has not been added/removed/modified
// as compared to only the last historical version.
TF_RETURN_IF_ERROR(OpDefAttrDefaultsUnchanged(in_op_history.op(end - 1),
op_list_.op(cur)));
TF_RETURN_IF_ERROR(
OpDefAttrDefaultsUnchanged(history_op_list.op(end - 1), cur_op));
// Check that attrs missing from in_op_history.op(start) don't
// change their defaults.
if (start < end - 1) {
// Check that attrs missing from history_op_list.op(0) don't change
// their defaults.
if (end > 1) {
TF_RETURN_IF_ERROR(OpDefAddedDefaultsUnchanged(
in_op_history.op(start), in_op_history.op(end - 1),
op_list_.op(cur)));
history_op_list.op(0), history_op_list.op(end - 1), cur_op));
}
// Compatible! Add changed op to the end of the history.
if (out_op_history != nullptr) {
*out_op_history->add_op() = op_list_.op(cur);
*out_op_history->back().second.add_op() = cur_op;
}
++*changed_ops;
}
// Advance past this op.
start = end;
++hist;
++cur;
}
}
// Error if missing ops.
if (stable_ops_ == nullptr && start < in_op_history.op_size()) {
return errors::InvalidArgument("Error, removed op: ",
SummarizeOpDef(in_op_history.op(start)));
if (stable_ops_ == nullptr && hist < in_op_history.size()) {
return errors::InvalidArgument(
"Error, removed op: ",
SummarizeOpDef(in_op_history[hist].second.op(0)));
}
// Add remaining new ops.
@ -184,9 +236,7 @@ Status OpCompatibilityLib::ValidateCompatible(Env* env, int* changed_ops,
if (stable_ops_ != nullptr && stable_ops_->count(op_name) == 0) {
// Ignore unstable op.
} else {
if (out_op_history) {
*out_op_history->add_op() = op_list_.op(cur);
}
AddNewOpToHistory(op_list_.op(cur), out_op_history);
++*added_ops;
}
}

View File

@ -45,6 +45,11 @@ class OpCompatibilityLib {
// order.
const string& op_history_file() const { return op_history_file_; }
// Name of the directory that contains all versions of *stable* ops,
// without docs. Op history is one file per op, in oldest-first
// order within the file.
const string& op_history_directory() const { return op_history_directory_; }
// Should match the contents of ops_file(). Run before calling
// ValidateCompatible().
string OpsString() const { return op_list_.DebugString(); }
@ -53,17 +58,21 @@ class OpCompatibilityLib {
// just stable ops.
int num_all_ops() const { return op_list_.op_size(); }
// <file name, file contents> pairs representing op history.
typedef std::vector<std::pair<string, OpList>> OpHistory;
// Make sure the current version of the *stable* ops are compatible
// with the historical versions, and if out_op_history != nullptr,
// generate a new history adding all changed ops. Sets
// *changed_ops/*added_ops to the number of changed/added ops
// (ignoring doc changes).
Status ValidateCompatible(Env* env, int* changed_ops, int* added_ops,
OpList* out_op_history);
OpHistory* out_op_history);
private:
const string ops_file_;
const string op_history_file_;
const string op_history_directory_;
const std::set<string>* stable_ops_;
OpList op_list_;
};

View File

@ -0,0 +1,17 @@
op {
name: "Abort"
attr {
name: "error_msg"
type: "string"
default_value {
s: ""
}
}
attr {
name: "exit_without_error"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,74 @@
op {
name: "Abs"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "Abs"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "Abs"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
}
}
}
}

View File

@ -0,0 +1,146 @@
op {
name: "AccumulateNV2"
input_arg {
name: "inputs"
type_attr: "T"
number_attr: "N"
}
output_arg {
name: "sum"
type_attr: "T"
}
attr {
name: "N"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "shape"
type: "shape"
}
is_aggregate: true
is_commutative: true
}
op {
name: "AccumulateNV2"
input_arg {
name: "inputs"
type_attr: "T"
number_attr: "N"
}
output_arg {
name: "sum"
type_attr: "T"
}
attr {
name: "N"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "shape"
type: "shape"
}
is_aggregate: true
is_commutative: true
}
op {
name: "AccumulateNV2"
input_arg {
name: "inputs"
type_attr: "T"
number_attr: "N"
}
output_arg {
name: "sum"
type_attr: "T"
}
attr {
name: "N"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "shape"
type: "shape"
}
is_aggregate: true
is_commutative: true
}

View File

@ -0,0 +1,160 @@
op {
name: "AccumulatorApplyGradient"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
input_arg {
name: "local_step"
type: DT_INT64
}
input_arg {
name: "gradient"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
}
op {
name: "AccumulatorApplyGradient"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
input_arg {
name: "local_step"
type: DT_INT64
}
input_arg {
name: "gradient"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
}
op {
name: "AccumulatorApplyGradient"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
input_arg {
name: "local_step"
type: DT_INT64
}
input_arg {
name: "gradient"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
}
op {
name: "AccumulatorApplyGradient"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
input_arg {
name: "local_step"
type: DT_INT64
}
input_arg {
name: "gradient"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
}

View File

@ -0,0 +1,12 @@
op {
name: "AccumulatorNumAccumulated"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
output_arg {
name: "num_accumulated"
type: DT_INT32
}
}

View File

@ -0,0 +1,12 @@
op {
name: "AccumulatorSetGlobalStep"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
input_arg {
name: "new_global_step"
type: DT_INT64
}
}

View File

@ -0,0 +1,160 @@
op {
name: "AccumulatorTakeGradient"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
input_arg {
name: "num_required"
type: DT_INT32
}
output_arg {
name: "average"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
}
op {
name: "AccumulatorTakeGradient"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
input_arg {
name: "num_required"
type: DT_INT32
}
output_arg {
name: "average"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
}
op {
name: "AccumulatorTakeGradient"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
input_arg {
name: "num_required"
type: DT_INT32
}
output_arg {
name: "average"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
}
op {
name: "AccumulatorTakeGradient"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
input_arg {
name: "num_required"
type: DT_INT32
}
output_arg {
name: "average"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
}

View File

@ -0,0 +1,80 @@
op {
name: "Acos"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Acos"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Acos"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}

View File

@ -0,0 +1,74 @@
op {
name: "Acosh"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Acosh"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Acosh"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}

View File

@ -0,0 +1,104 @@
op {
name: "Add"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_UINT8
type: DT_INT8
type: DT_INT16
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_STRING
}
}
}
}
op {
name: "Add"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
type: DT_UINT8
type: DT_INT8
type: DT_INT16
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_STRING
}
}
}
}
op {
name: "Add"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_UINT8
type: DT_INT8
type: DT_INT16
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_STRING
}
}
}
}

View File

@ -0,0 +1,38 @@
op {
name: "AddManySparseToTensorsMap"
input_arg {
name: "sparse_indices"
type: DT_INT64
}
input_arg {
name: "sparse_values"
type_attr: "T"
}
input_arg {
name: "sparse_shape"
type: DT_INT64
}
output_arg {
name: "sparse_handles"
type: DT_INT64
}
attr {
name: "T"
type: "type"
}
attr {
name: "container"
type: "string"
default_value {
s: ""
}
}
attr {
name: "shared_name"
type: "string"
default_value {
s: ""
}
}
is_stateful: true
}

View File

@ -0,0 +1,222 @@
op {
name: "AddN"
input_arg {
name: "inputs"
type_attr: "T"
number_attr: "N"
}
output_arg {
name: "sum"
type_attr: "T"
}
attr {
name: "N"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
is_aggregate: true
is_commutative: true
}
op {
name: "AddN"
input_arg {
name: "inputs"
type_attr: "T"
number_attr: "N"
}
output_arg {
name: "sum"
type_attr: "T"
}
attr {
name: "N"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_VARIANT
}
}
}
is_aggregate: true
is_commutative: true
}
op {
name: "AddN"
input_arg {
name: "inputs"
type_attr: "T"
number_attr: "N"
}
output_arg {
name: "sum"
type_attr: "T"
}
attr {
name: "N"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_VARIANT
}
}
}
is_aggregate: true
is_commutative: true
}
op {
name: "AddN"
input_arg {
name: "inputs"
type_attr: "T"
number_attr: "N"
}
output_arg {
name: "sum"
type_attr: "T"
}
attr {
name: "N"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
type: DT_VARIANT
}
}
}
is_aggregate: true
is_commutative: true
}
op {
name: "AddN"
input_arg {
name: "inputs"
type_attr: "T"
number_attr: "N"
}
output_arg {
name: "sum"
type_attr: "T"
}
attr {
name: "N"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_VARIANT
}
}
}
is_aggregate: true
is_commutative: true
}

View File

@ -0,0 +1,38 @@
op {
name: "AddSparseToTensorsMap"
input_arg {
name: "sparse_indices"
type: DT_INT64
}
input_arg {
name: "sparse_values"
type_attr: "T"
}
input_arg {
name: "sparse_shape"
type: DT_INT64
}
output_arg {
name: "sparse_handle"
type: DT_INT64
}
attr {
name: "T"
type: "type"
}
attr {
name: "container"
type: "string"
default_value {
s: ""
}
}
attr {
name: "shared_name"
type: "string"
default_value {
s: ""
}
}
is_stateful: true
}

View File

@ -0,0 +1,107 @@
op {
name: "AddV2"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_UINT8
type: DT_INT8
type: DT_INT16
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
is_aggregate: true
is_commutative: true
}
op {
name: "AddV2"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
type: DT_UINT8
type: DT_INT8
type: DT_INT16
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
is_aggregate: true
is_commutative: true
}
op {
name: "AddV2"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_UINT8
type: DT_INT8
type: DT_INT16
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
is_aggregate: true
is_commutative: true
}

View File

@ -0,0 +1,41 @@
op {
name: "AdjustContrast"
input_arg {
name: "images"
type_attr: "T"
}
input_arg {
name: "contrast_factor"
type: DT_FLOAT
}
input_arg {
name: "min_value"
type: DT_FLOAT
}
input_arg {
name: "max_value"
type: DT_FLOAT
}
output_arg {
name: "output"
type: DT_FLOAT
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_UINT8
type: DT_INT8
type: DT_INT16
type: DT_INT32
type: DT_INT64
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
deprecation {
version: 2
}
}

View File

@ -0,0 +1,43 @@
op {
name: "AdjustContrastv2"
input_arg {
name: "images"
type: DT_FLOAT
}
input_arg {
name: "contrast_factor"
type: DT_FLOAT
}
output_arg {
name: "output"
type: DT_FLOAT
}
}
op {
name: "AdjustContrastv2"
input_arg {
name: "images"
type_attr: "T"
}
input_arg {
name: "contrast_factor"
type: DT_FLOAT
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
}
}
}
}

View File

@ -0,0 +1,43 @@
op {
name: "AdjustHue"
input_arg {
name: "images"
type: DT_FLOAT
}
input_arg {
name: "delta"
type: DT_FLOAT
}
output_arg {
name: "output"
type: DT_FLOAT
}
}
op {
name: "AdjustHue"
input_arg {
name: "images"
type_attr: "T"
}
input_arg {
name: "delta"
type: DT_FLOAT
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
}
}
}
}

View File

@ -0,0 +1,43 @@
op {
name: "AdjustSaturation"
input_arg {
name: "images"
type: DT_FLOAT
}
input_arg {
name: "scale"
type: DT_FLOAT
}
output_arg {
name: "output"
type: DT_FLOAT
}
}
op {
name: "AdjustSaturation"
input_arg {
name: "images"
type_attr: "T"
}
input_arg {
name: "scale"
type: DT_FLOAT
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
}
}
}
}

View File

@ -0,0 +1,35 @@
op {
name: "All"
input_arg {
name: "input"
type: DT_BOOL
}
input_arg {
name: "reduction_indices"
type_attr: "Tidx"
}
output_arg {
name: "output"
type: DT_BOOL
}
attr {
name: "keep_dims"
type: "bool"
default_value {
b: false
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}

View File

@ -0,0 +1,99 @@
op {
name: "AllCandidateSampler"
input_arg {
name: "true_classes"
type: DT_INT64
}
output_arg {
name: "sampled_candidates"
type: DT_INT64
}
output_arg {
name: "true_expected_count"
type: DT_FLOAT
}
output_arg {
name: "sampled_expected_count"
type: DT_FLOAT
}
attr {
name: "num_true"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "num_sampled"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "unique"
type: "bool"
}
attr {
name: "seed"
type: "int"
default_value {
i: 0
}
}
attr {
name: "seed2"
type: "int"
default_value {
i: 0
}
}
}
op {
name: "AllCandidateSampler"
input_arg {
name: "true_classes"
type: DT_INT64
}
output_arg {
name: "sampled_candidates"
type: DT_INT64
}
output_arg {
name: "true_expected_count"
type: DT_FLOAT
}
output_arg {
name: "sampled_expected_count"
type: DT_FLOAT
}
attr {
name: "num_true"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "num_sampled"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "unique"
type: "bool"
}
attr {
name: "seed"
type: "int"
default_value {
i: 0
}
}
attr {
name: "seed2"
type: "int"
default_value {
i: 0
}
}
is_stateful: true
}

View File

@ -0,0 +1,90 @@
op {
name: "AllToAll"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "group_assignment"
type: DT_INT32
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
}
}
}
attr {
name: "concat_dimension"
type: "int"
}
attr {
name: "split_dimension"
type: "int"
}
attr {
name: "split_count"
type: "int"
}
}
op {
name: "AllToAll"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "group_assignment"
type: DT_INT32
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BOOL
}
}
}
attr {
name: "concat_dimension"
type: "int"
}
attr {
name: "split_dimension"
type: "int"
}
attr {
name: "split_count"
type: "int"
}
}

View File

@ -0,0 +1,37 @@
op {
name: "Angle"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "Tout"
}
attr {
name: "T"
type: "type"
default_value {
type: DT_COMPLEX64
}
allowed_values {
list {
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
attr {
name: "Tout"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,20 @@
op {
name: "AnonymousIterator"
output_arg {
name: "handle"
type: DT_RESOURCE
}
attr {
name: "output_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "output_shapes"
type: "list(shape)"
has_minimum: true
minimum: 1
}
is_stateful: true
}

View File

@ -0,0 +1,24 @@
op {
name: "AnonymousIteratorV2"
output_arg {
name: "handle"
type: DT_RESOURCE
}
output_arg {
name: "deleter"
type: DT_VARIANT
}
attr {
name: "output_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "output_shapes"
type: "list(shape)"
has_minimum: true
minimum: 1
}
is_stateful: true
}

View File

@ -0,0 +1,30 @@
op {
name: "AnonymousMultiDeviceIterator"
output_arg {
name: "handle"
type: DT_RESOURCE
}
output_arg {
name: "deleter"
type: DT_VARIANT
}
attr {
name: "devices"
type: "list(string)"
has_minimum: true
minimum: 1
}
attr {
name: "output_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "output_shapes"
type: "list(shape)"
has_minimum: true
minimum: 1
}
is_stateful: true
}

View File

@ -0,0 +1,35 @@
op {
name: "Any"
input_arg {
name: "input"
type: DT_BOOL
}
input_arg {
name: "reduction_indices"
type_attr: "Tidx"
}
output_arg {
name: "output"
type: DT_BOOL
}
attr {
name: "keep_dims"
type: "bool"
default_value {
b: false
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}

View File

@ -0,0 +1,79 @@
op {
name: "ApplyAdaMax"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "v"
type_attr: "T"
is_ref: true
}
input_arg {
name: "beta1_power"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "beta1"
type_attr: "T"
}
input_arg {
name: "beta2"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,280 @@
op {
name: "ApplyAdadelta"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum_update"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdadelta"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum_update"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdadelta"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum_update"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdadelta"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum_update"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,293 @@
op {
name: "ApplyAdagrad"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdagrad"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdagrad"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdagrad"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdagrad"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
attr {
name: "update_slots"
type: "bool"
default_value {
b: true
}
}
}

View File

@ -0,0 +1,296 @@
op {
name: "ApplyAdagradDA"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "gradient_accumulator"
type_attr: "T"
is_ref: true
}
input_arg {
name: "gradient_squared_accumulator"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "global_step"
type: DT_INT64
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdagradDA"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "gradient_accumulator"
type_attr: "T"
is_ref: true
}
input_arg {
name: "gradient_squared_accumulator"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "global_step"
type: DT_INT64
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdagradDA"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "gradient_accumulator"
type_attr: "T"
is_ref: true
}
input_arg {
name: "gradient_squared_accumulator"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "global_step"
type: DT_INT64
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdagradDA"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "gradient_accumulator"
type_attr: "T"
is_ref: true
}
input_arg {
name: "gradient_squared_accumulator"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "global_step"
type: DT_INT64
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,436 @@
op {
name: "ApplyAdam"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "v"
type_attr: "T"
is_ref: true
}
input_arg {
name: "beta1_power"
type_attr: "T"
}
input_arg {
name: "beta2_power"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "beta1"
type_attr: "T"
}
input_arg {
name: "beta2"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdam"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "v"
type_attr: "T"
is_ref: true
}
input_arg {
name: "beta1_power"
type_attr: "T"
}
input_arg {
name: "beta2_power"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "beta1"
type_attr: "T"
}
input_arg {
name: "beta2"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
attr {
name: "use_nesterov"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdam"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "v"
type_attr: "T"
is_ref: true
}
input_arg {
name: "beta1_power"
type_attr: "T"
}
input_arg {
name: "beta2_power"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "beta1"
type_attr: "T"
}
input_arg {
name: "beta2"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
attr {
name: "use_nesterov"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdam"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "v"
type_attr: "T"
is_ref: true
}
input_arg {
name: "beta1_power"
type_attr: "T"
}
input_arg {
name: "beta2_power"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "beta1"
type_attr: "T"
}
input_arg {
name: "beta2"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
attr {
name: "use_nesterov"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAdam"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "v"
type_attr: "T"
is_ref: true
}
input_arg {
name: "beta1_power"
type_attr: "T"
}
input_arg {
name: "beta2_power"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "beta1"
type_attr: "T"
}
input_arg {
name: "beta2"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
attr {
name: "use_nesterov"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,209 @@
op {
name: "ApplyAddSign"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "alpha"
type_attr: "T"
}
input_arg {
name: "sign_decay"
type_attr: "T"
}
input_arg {
name: "beta"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAddSign"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "alpha"
type_attr: "T"
}
input_arg {
name: "sign_decay"
type_attr: "T"
}
input_arg {
name: "beta"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyAddSign"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "alpha"
type_attr: "T"
}
input_arg {
name: "sign_decay"
type_attr: "T"
}
input_arg {
name: "beta"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,316 @@
op {
name: "ApplyCenteredRMSProp"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mg"
type_attr: "T"
is_ref: true
}
input_arg {
name: "ms"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mom"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyCenteredRMSProp"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mg"
type_attr: "T"
is_ref: true
}
input_arg {
name: "ms"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mom"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyCenteredRMSProp"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mg"
type_attr: "T"
is_ref: true
}
input_arg {
name: "ms"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mom"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyCenteredRMSProp"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mg"
type_attr: "T"
is_ref: true
}
input_arg {
name: "ms"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mom"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,296 @@
op {
name: "ApplyFtrl"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "linear"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "lr_power"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyFtrl"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "linear"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "lr_power"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyFtrl"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "linear"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "lr_power"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyFtrl"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "linear"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "lr_power"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,312 @@
op {
name: "ApplyFtrlV2"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "linear"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "l2_shrinkage"
type_attr: "T"
}
input_arg {
name: "lr_power"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyFtrlV2"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "linear"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "l2_shrinkage"
type_attr: "T"
}
input_arg {
name: "lr_power"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyFtrlV2"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "linear"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "l2_shrinkage"
type_attr: "T"
}
input_arg {
name: "lr_power"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyFtrlV2"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "linear"
type_attr: "T"
is_ref: true
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "l2_shrinkage"
type_attr: "T"
}
input_arg {
name: "lr_power"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,208 @@
op {
name: "ApplyGradientDescent"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "alpha"
type_attr: "T"
}
input_arg {
name: "delta"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyGradientDescent"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "alpha"
type_attr: "T"
}
input_arg {
name: "delta"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyGradientDescent"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "alpha"
type_attr: "T"
}
input_arg {
name: "delta"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyGradientDescent"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "alpha"
type_attr: "T"
}
input_arg {
name: "delta"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,272 @@
op {
name: "ApplyMomentum"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
attr {
name: "use_nesterov"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyMomentum"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
attr {
name: "use_nesterov"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyMomentum"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
attr {
name: "use_nesterov"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyMomentum"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
attr {
name: "use_nesterov"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,209 @@
op {
name: "ApplyPowerSign"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "logbase"
type_attr: "T"
}
input_arg {
name: "sign_decay"
type_attr: "T"
}
input_arg {
name: "beta"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyPowerSign"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "logbase"
type_attr: "T"
}
input_arg {
name: "sign_decay"
type_attr: "T"
}
input_arg {
name: "beta"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyPowerSign"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "m"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "logbase"
type_attr: "T"
}
input_arg {
name: "sign_decay"
type_attr: "T"
}
input_arg {
name: "beta"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,260 @@
op {
name: "ApplyProximalAdagrad"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyProximalAdagrad"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyProximalAdagrad"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyProximalAdagrad"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "accum"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,240 @@
op {
name: "ApplyProximalGradientDescent"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "alpha"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "delta"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyProximalGradientDescent"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "alpha"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "delta"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyProximalGradientDescent"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "alpha"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "delta"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyProximalGradientDescent"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "alpha"
type_attr: "T"
}
input_arg {
name: "l1"
type_attr: "T"
}
input_arg {
name: "l2"
type_attr: "T"
}
input_arg {
name: "delta"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,296 @@
op {
name: "ApplyRMSProp"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "ms"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mom"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyRMSProp"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "ms"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mom"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyRMSProp"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "ms"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mom"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "ApplyRMSProp"
input_arg {
name: "var"
type_attr: "T"
is_ref: true
}
input_arg {
name: "ms"
type_attr: "T"
is_ref: true
}
input_arg {
name: "mom"
type_attr: "T"
is_ref: true
}
input_arg {
name: "lr"
type_attr: "T"
}
input_arg {
name: "rho"
type_attr: "T"
}
input_arg {
name: "momentum"
type_attr: "T"
}
input_arg {
name: "epsilon"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "out"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,188 @@
op {
name: "ApproximateEqual"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type: DT_BOOL
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "tolerance"
type: "float"
default_value {
f: 1e-05
}
}
is_commutative: true
}
op {
name: "ApproximateEqual"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type: DT_BOOL
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "tolerance"
type: "float"
default_value {
f: 1e-05
}
}
is_commutative: true
}
op {
name: "ApproximateEqual"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type: DT_BOOL
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "tolerance"
type: "float"
default_value {
f: 1e-05
}
}
is_commutative: true
}
op {
name: "ApproximateEqual"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type: DT_BOOL
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "tolerance"
type: "float"
default_value {
f: 1e-05
}
}
is_commutative: true
}

View File

@ -0,0 +1,310 @@
op {
name: "ArgMax"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type: DT_INT64
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "ArgMax"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type_attr: "output_type"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "output_type"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "ArgMax"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type_attr: "output_type"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "output_type"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "ArgMax"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type_attr: "output_type"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "output_type"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "ArgMax"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type_attr: "output_type"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "output_type"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}

View File

@ -0,0 +1,310 @@
op {
name: "ArgMin"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type: DT_INT64
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "ArgMin"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type_attr: "output_type"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "output_type"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "ArgMin"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type_attr: "output_type"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "output_type"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "ArgMin"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type_attr: "output_type"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "output_type"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "ArgMin"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type_attr: "output_type"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "output_type"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}

View File

@ -0,0 +1,186 @@
op {
name: "AsString"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type: DT_STRING
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_FLOAT
type: DT_DOUBLE
type: DT_BOOL
type: DT_INT8
}
}
}
attr {
name: "precision"
type: "int"
default_value {
i: -1
}
}
attr {
name: "scientific"
type: "bool"
default_value {
b: false
}
}
attr {
name: "shortest"
type: "bool"
default_value {
b: false
}
}
attr {
name: "width"
type: "int"
default_value {
i: -1
}
}
attr {
name: "fill"
type: "string"
default_value {
s: ""
}
}
}
op {
name: "AsString"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type: DT_STRING
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_INT8
type: DT_INT16
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_FLOAT
type: DT_DOUBLE
type: DT_BOOL
}
}
}
attr {
name: "precision"
type: "int"
default_value {
i: -1
}
}
attr {
name: "scientific"
type: "bool"
default_value {
b: false
}
}
attr {
name: "shortest"
type: "bool"
default_value {
b: false
}
}
attr {
name: "width"
type: "int"
default_value {
i: -1
}
}
attr {
name: "fill"
type: "string"
default_value {
s: ""
}
}
}
op {
name: "AsString"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type: DT_STRING
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_INT8
type: DT_INT16
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_FLOAT
type: DT_DOUBLE
type: DT_BOOL
}
}
}
attr {
name: "precision"
type: "int"
default_value {
i: -1
}
}
attr {
name: "scientific"
type: "bool"
default_value {
b: false
}
}
attr {
name: "shortest"
type: "bool"
default_value {
b: false
}
}
attr {
name: "width"
type: "int"
default_value {
i: -1
}
}
attr {
name: "fill"
type: "string"
default_value {
s: ""
}
}
}

View File

@ -0,0 +1,80 @@
op {
name: "Asin"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Asin"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Asin"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}

View File

@ -0,0 +1,74 @@
op {
name: "Asinh"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Asinh"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Asinh"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}

View File

@ -0,0 +1,25 @@
op {
name: "Assert"
input_arg {
name: "condition"
type: DT_BOOL
}
input_arg {
name: "data"
type_list_attr: "T"
}
attr {
name: "T"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "summarize"
type: "int"
default_value {
i: 3
}
}
is_stateful: true
}

View File

@ -0,0 +1,27 @@
op {
name: "AssertNextDataset"
input_arg {
name: "input_dataset"
type: DT_VARIANT
}
input_arg {
name: "transformations"
type: DT_STRING
}
output_arg {
name: "handle"
type: DT_VARIANT
}
attr {
name: "output_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "output_shapes"
type: "list(shape)"
has_minimum: true
minimum: 1
}
}

View File

@ -0,0 +1,36 @@
op {
name: "Assign"
input_arg {
name: "ref"
type_attr: "T"
is_ref: true
}
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output_ref"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
}
attr {
name: "validate_shape"
type: "bool"
default_value {
b: true
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: true
}
}
allows_uninitialized_input: true
}

View File

@ -0,0 +1,192 @@
op {
name: "AssignAdd"
input_arg {
name: "ref"
type_attr: "T"
is_ref: true
}
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output_ref"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "AssignAdd"
input_arg {
name: "ref"
type_attr: "T"
is_ref: true
}
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output_ref"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "AssignAdd"
input_arg {
name: "ref"
type_attr: "T"
is_ref: true
}
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output_ref"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "AssignAdd"
input_arg {
name: "ref"
type_attr: "T"
is_ref: true
}
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output_ref"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,16 @@
op {
name: "AssignAddVariableOp"
input_arg {
name: "resource"
type: DT_RESOURCE
}
input_arg {
name: "value"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
}
is_stateful: true
}

View File

@ -0,0 +1,192 @@
op {
name: "AssignSub"
input_arg {
name: "ref"
type_attr: "T"
is_ref: true
}
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output_ref"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "AssignSub"
input_arg {
name: "ref"
type_attr: "T"
is_ref: true
}
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output_ref"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "AssignSub"
input_arg {
name: "ref"
type_attr: "T"
is_ref: true
}
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output_ref"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "AssignSub"
input_arg {
name: "ref"
type_attr: "T"
is_ref: true
}
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output_ref"
type_attr: "T"
is_ref: true
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "use_locking"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,16 @@
op {
name: "AssignSubVariableOp"
input_arg {
name: "resource"
type: DT_RESOURCE
}
input_arg {
name: "value"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
}
is_stateful: true
}

View File

@ -0,0 +1,16 @@
op {
name: "AssignVariableOp"
input_arg {
name: "resource"
type: DT_RESOURCE
}
input_arg {
name: "value"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
}
is_stateful: true
}

View File

@ -0,0 +1,80 @@
op {
name: "Atan"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Atan"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Atan"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}

View File

@ -0,0 +1,78 @@
op {
name: "Atan2"
input_arg {
name: "y"
type_attr: "T"
}
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "Atan2"
input_arg {
name: "y"
type_attr: "T"
}
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "Atan2"
input_arg {
name: "y"
type_attr: "T"
}
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,74 @@
op {
name: "Atanh"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Atanh"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}
op {
name: "Atanh"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
}

View File

@ -0,0 +1,26 @@
op {
name: "AudioSpectrogram"
input_arg {
name: "input"
type: DT_FLOAT
}
output_arg {
name: "spectrogram"
type: DT_FLOAT
}
attr {
name: "window_size"
type: "int"
}
attr {
name: "stride"
type: "int"
}
attr {
name: "magnitude_squared"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,31 @@
op {
name: "AudioSummary"
input_arg {
name: "tag"
type: DT_STRING
}
input_arg {
name: "tensor"
type: DT_FLOAT
}
output_arg {
name: "summary"
type: DT_STRING
}
attr {
name: "sample_rate"
type: "float"
}
attr {
name: "max_outputs"
type: "int"
default_value {
i: 3
}
has_minimum: true
minimum: 1
}
deprecation {
version: 15
}
}

View File

@ -0,0 +1,28 @@
op {
name: "AudioSummaryV2"
input_arg {
name: "tag"
type: DT_STRING
}
input_arg {
name: "tensor"
type: DT_FLOAT
}
input_arg {
name: "sample_rate"
type: DT_FLOAT
}
output_arg {
name: "summary"
type: DT_STRING
}
attr {
name: "max_outputs"
type: "int"
default_value {
i: 3
}
has_minimum: true
minimum: 1
}
}

View File

@ -0,0 +1,31 @@
op {
name: "AutoShardDataset"
input_arg {
name: "input_dataset"
type: DT_VARIANT
}
input_arg {
name: "num_workers"
type: DT_INT64
}
input_arg {
name: "index"
type: DT_INT64
}
output_arg {
name: "handle"
type: DT_VARIANT
}
attr {
name: "output_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "output_shapes"
type: "list(shape)"
has_minimum: true
minimum: 1
}
}

View File

@ -0,0 +1,229 @@
op {
name: "AvgPool"
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NHWC"
}
allowed_values {
list {
s: "NHWC"
s: "NCHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_HALF
type: DT_DOUBLE
}
}
}
}
op {
name: "AvgPool"
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NHWC"
}
allowed_values {
list {
s: "NHWC"
s: "NCHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_HALF
}
}
}
}
op {
name: "AvgPool"
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NHWC"
}
allowed_values {
list {
s: "NHWC"
s: "NCHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "AvgPool"
input_arg {
name: "value"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NHWC"
}
allowed_values {
list {
s: "NHWC"
s: "NCHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,214 @@
op {
name: "AvgPool3D"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "AvgPool3D"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NDHWC"
}
allowed_values {
list {
s: "NDHWC"
s: "NCDHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "AvgPool3D"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NDHWC"
}
allowed_values {
list {
s: "NDHWC"
s: "NCDHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "AvgPool3D"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NDHWC"
}
allowed_values {
list {
s: "NDHWC"
s: "NCDHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,230 @@
op {
name: "AvgPool3DGrad"
input_arg {
name: "orig_input_shape"
type: DT_INT32
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "AvgPool3DGrad"
input_arg {
name: "orig_input_shape"
type: DT_INT32
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NDHWC"
}
allowed_values {
list {
s: "NDHWC"
s: "NCDHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "AvgPool3DGrad"
input_arg {
name: "orig_input_shape"
type: DT_INT32
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NDHWC"
}
allowed_values {
list {
s: "NDHWC"
s: "NCDHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "AvgPool3DGrad"
input_arg {
name: "orig_input_shape"
type: DT_INT32
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 5
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NDHWC"
}
allowed_values {
list {
s: "NDHWC"
s: "NCDHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,245 @@
op {
name: "AvgPoolGrad"
input_arg {
name: "orig_input_shape"
type: DT_INT32
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NHWC"
}
allowed_values {
list {
s: "NHWC"
s: "NCHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_HALF
type: DT_DOUBLE
}
}
}
}
op {
name: "AvgPoolGrad"
input_arg {
name: "orig_input_shape"
type: DT_INT32
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NHWC"
}
allowed_values {
list {
s: "NHWC"
s: "NCHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_HALF
}
}
}
}
op {
name: "AvgPoolGrad"
input_arg {
name: "orig_input_shape"
type: DT_INT32
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NHWC"
}
allowed_values {
list {
s: "NHWC"
s: "NCHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "AvgPoolGrad"
input_arg {
name: "orig_input_shape"
type: DT_INT32
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "ksize"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "strides"
type: "list(int)"
has_minimum: true
minimum: 4
}
attr {
name: "padding"
type: "string"
allowed_values {
list {
s: "SAME"
s: "VALID"
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NHWC"
}
allowed_values {
list {
s: "NHWC"
s: "NCHW"
}
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,45 @@
op {
name: "Barrier"
output_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
attr {
name: "component_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "shapes"
type: "list(shape)"
default_value {
list {
}
}
has_minimum: true
}
attr {
name: "capacity"
type: "int"
default_value {
i: -1
}
}
attr {
name: "container"
type: "string"
default_value {
s: ""
}
}
attr {
name: "shared_name"
type: "string"
default_value {
s: ""
}
}
is_stateful: true
}

View File

@ -0,0 +1,15 @@
op {
name: "BarrierClose"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
attr {
name: "cancel_pending_enqueues"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,12 @@
op {
name: "BarrierIncompleteSize"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
output_arg {
name: "size"
type: DT_INT32
}
}

View File

@ -0,0 +1,24 @@
op {
name: "BarrierInsertMany"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
input_arg {
name: "keys"
type: DT_STRING
}
input_arg {
name: "values"
type_attr: "T"
}
attr {
name: "T"
type: "type"
}
attr {
name: "component_index"
type: "int"
}
}

View File

@ -0,0 +1,12 @@
op {
name: "BarrierReadySize"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
output_arg {
name: "size"
type: DT_INT32
}
}

View File

@ -0,0 +1,51 @@
op {
name: "BarrierTakeMany"
input_arg {
name: "handle"
type: DT_STRING
is_ref: true
}
input_arg {
name: "num_elements"
type: DT_INT32
}
output_arg {
name: "indices"
type: DT_INT64
}
output_arg {
name: "keys"
type: DT_STRING
}
output_arg {
name: "values"
type_list_attr: "component_types"
}
attr {
name: "component_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "allow_small_batch"
type: "bool"
default_value {
b: false
}
}
attr {
name: "wait_for_incomplete"
type: "bool"
default_value {
b: false
}
}
attr {
name: "timeout_ms"
type: "int"
default_value {
i: -1
}
}
}

View File

@ -0,0 +1,147 @@
op {
name: "Batch"
input_arg {
name: "in_tensors"
type_list_attr: "T"
}
output_arg {
name: "batched_tensors"
type_list_attr: "T"
}
output_arg {
name: "batch_index"
type: DT_INT64
}
output_arg {
name: "id"
type: DT_INT64
}
attr {
name: "num_batch_threads"
type: "int"
}
attr {
name: "max_batch_size"
type: "int"
}
attr {
name: "batch_timeout_micros"
type: "int"
}
attr {
name: "allowed_batch_sizes"
type: "list(int)"
default_value {
list {
}
}
}
attr {
name: "grad_timeout_micros"
type: "int"
}
attr {
name: "container"
type: "string"
default_value {
s: ""
}
}
attr {
name: "shared_name"
type: "string"
default_value {
s: ""
}
}
attr {
name: "batching_queue"
type: "string"
default_value {
s: ""
}
}
attr {
name: "T"
type: "list(type)"
has_minimum: true
minimum: 1
}
}
op {
name: "Batch"
input_arg {
name: "in_tensors"
type_list_attr: "T"
}
output_arg {
name: "batched_tensors"
type_list_attr: "T"
}
output_arg {
name: "batch_index"
type: DT_INT64
}
output_arg {
name: "id"
type: DT_INT64
}
attr {
name: "num_batch_threads"
type: "int"
}
attr {
name: "max_batch_size"
type: "int"
}
attr {
name: "max_enqueued_batches"
type: "int"
default_value {
i: 10
}
}
attr {
name: "batch_timeout_micros"
type: "int"
}
attr {
name: "allowed_batch_sizes"
type: "list(int)"
default_value {
list {
}
}
}
attr {
name: "grad_timeout_micros"
type: "int"
}
attr {
name: "container"
type: "string"
default_value {
s: ""
}
}
attr {
name: "shared_name"
type: "string"
default_value {
s: ""
}
}
attr {
name: "batching_queue"
type: "string"
default_value {
s: ""
}
}
attr {
name: "T"
type: "list(type)"
has_minimum: true
minimum: 1
}
}

View File

@ -0,0 +1,24 @@
op {
name: "BatchCholesky"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_DOUBLE
type: DT_FLOAT
}
}
}
deprecation {
version: 13
}
}

View File

@ -0,0 +1,28 @@
op {
name: "BatchCholeskyGrad"
input_arg {
name: "l"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
deprecation {
version: 13
}
}

View File

@ -0,0 +1,55 @@
op {
name: "BatchDataset"
input_arg {
name: "input_dataset"
type: DT_VARIANT
}
input_arg {
name: "batch_size"
type: DT_INT64
}
output_arg {
name: "handle"
type: DT_VARIANT
}
attr {
name: "output_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "output_shapes"
type: "list(shape)"
has_minimum: true
minimum: 1
}
is_stateful: true
}
op {
name: "BatchDataset"
input_arg {
name: "input_dataset"
type: DT_VARIANT
}
input_arg {
name: "batch_size"
type: DT_INT64
}
output_arg {
name: "handle"
type: DT_VARIANT
}
attr {
name: "output_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "output_shapes"
type: "list(shape)"
has_minimum: true
minimum: 1
}
}

View File

@ -0,0 +1,69 @@
op {
name: "BatchDatasetV2"
input_arg {
name: "input_dataset"
type: DT_VARIANT
}
input_arg {
name: "batch_size"
type: DT_INT64
}
input_arg {
name: "drop_remainder"
type: DT_BOOL
}
output_arg {
name: "handle"
type: DT_VARIANT
}
attr {
name: "output_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "output_shapes"
type: "list(shape)"
has_minimum: true
minimum: 1
}
}
op {
name: "BatchDatasetV2"
input_arg {
name: "input_dataset"
type: DT_VARIANT
}
input_arg {
name: "batch_size"
type: DT_INT64
}
input_arg {
name: "drop_remainder"
type: DT_BOOL
}
output_arg {
name: "handle"
type: DT_VARIANT
}
attr {
name: "parallel_copy"
type: "bool"
default_value {
b: false
}
}
attr {
name: "output_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "output_shapes"
type: "list(shape)"
has_minimum: true
minimum: 1
}
}

View File

@ -0,0 +1,14 @@
op {
name: "BatchFFT"
input_arg {
name: "input"
type: DT_COMPLEX64
}
output_arg {
name: "output"
type: DT_COMPLEX64
}
deprecation {
version: 15
}
}

View File

@ -0,0 +1,14 @@
op {
name: "BatchFFT2D"
input_arg {
name: "input"
type: DT_COMPLEX64
}
output_arg {
name: "output"
type: DT_COMPLEX64
}
deprecation {
version: 15
}
}

View File

@ -0,0 +1,14 @@
op {
name: "BatchFFT3D"
input_arg {
name: "input"
type: DT_COMPLEX64
}
output_arg {
name: "output"
type: DT_COMPLEX64
}
deprecation {
version: 15
}
}

View File

@ -0,0 +1,84 @@
op {
name: "BatchFunction"
input_arg {
name: "in_tensors"
type_list_attr: "Tin"
}
input_arg {
name: "captured_tensors"
type_list_attr: "Tcaptured"
}
output_arg {
name: "out_tensors"
type_list_attr: "Tout"
}
attr {
name: "f"
type: "func"
}
attr {
name: "num_batch_threads"
type: "int"
}
attr {
name: "max_batch_size"
type: "int"
}
attr {
name: "batch_timeout_micros"
type: "int"
}
attr {
name: "max_enqueued_batches"
type: "int"
default_value {
i: 10
}
}
attr {
name: "allowed_batch_sizes"
type: "list(int)"
default_value {
list {
}
}
}
attr {
name: "container"
type: "string"
default_value {
s: ""
}
}
attr {
name: "shared_name"
type: "string"
default_value {
s: ""
}
}
attr {
name: "batching_queue"
type: "string"
default_value {
s: ""
}
}
attr {
name: "Tin"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "Tcaptured"
type: "list(type)"
has_minimum: true
}
attr {
name: "Tout"
type: "list(type)"
has_minimum: true
minimum: 1
}
}

View File

@ -0,0 +1,14 @@
op {
name: "BatchIFFT"
input_arg {
name: "input"
type: DT_COMPLEX64
}
output_arg {
name: "output"
type: DT_COMPLEX64
}
deprecation {
version: 15
}
}

View File

@ -0,0 +1,14 @@
op {
name: "BatchIFFT2D"
input_arg {
name: "input"
type: DT_COMPLEX64
}
output_arg {
name: "output"
type: DT_COMPLEX64
}
deprecation {
version: 15
}
}

View File

@ -0,0 +1,14 @@
op {
name: "BatchIFFT3D"
input_arg {
name: "input"
type: DT_COMPLEX64
}
output_arg {
name: "output"
type: DT_COMPLEX64
}
deprecation {
version: 15
}
}

View File

@ -0,0 +1,176 @@
op {
name: "BatchMatMul"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
attr {
name: "adj_x"
type: "bool"
default_value {
b: false
}
}
attr {
name: "adj_y"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "BatchMatMul"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
attr {
name: "adj_x"
type: "bool"
default_value {
b: false
}
}
attr {
name: "adj_y"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "BatchMatMul"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
attr {
name: "adj_x"
type: "bool"
default_value {
b: false
}
}
attr {
name: "adj_y"
type: "bool"
default_value {
b: false
}
}
}
op {
name: "BatchMatMul"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
attr {
name: "adj_x"
type: "bool"
default_value {
b: false
}
}
attr {
name: "adj_y"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,45 @@
op {
name: "BatchMatMulV2"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
attr {
name: "adj_x"
type: "bool"
default_value {
b: false
}
}
attr {
name: "adj_y"
type: "bool"
default_value {
b: false
}
}
}

View File

@ -0,0 +1,26 @@
op {
name: "BatchMatrixBandPart"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "num_lower"
type: DT_INT64
}
input_arg {
name: "num_upper"
type: DT_INT64
}
output_arg {
name: "band"
type_attr: "T"
}
attr {
name: "T"
type: "type"
}
deprecation {
version: 14
}
}

View File

@ -0,0 +1,50 @@
op {
name: "BatchMatrixDeterminant"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
deprecation {
version: 13
}
}
op {
name: "BatchMatrixDeterminant"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
deprecation {
version: 13
}
}

View File

@ -0,0 +1,18 @@
op {
name: "BatchMatrixDiag"
input_arg {
name: "diagonal"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
}
deprecation {
version: 14
}
}

View File

@ -0,0 +1,18 @@
op {
name: "BatchMatrixDiagPart"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "diagonal"
type_attr: "T"
}
attr {
name: "T"
type: "type"
}
deprecation {
version: 14
}
}

View File

@ -0,0 +1,31 @@
op {
name: "BatchMatrixInverse"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "adjoint"
type: "bool"
default_value {
b: false
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_DOUBLE
type: DT_FLOAT
}
}
}
deprecation {
version: 13
}
}

View File

@ -0,0 +1,22 @@
op {
name: "BatchMatrixSetDiag"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "diagonal"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
}
deprecation {
version: 14
}
}

View File

@ -0,0 +1,35 @@
op {
name: "BatchMatrixSolve"
input_arg {
name: "matrix"
type_attr: "T"
}
input_arg {
name: "rhs"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "adjoint"
type: "bool"
default_value {
b: false
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_DOUBLE
type: DT_FLOAT
}
}
}
deprecation {
version: 13
}
}

View File

@ -0,0 +1,39 @@
op {
name: "BatchMatrixSolveLs"
input_arg {
name: "matrix"
type_attr: "T"
}
input_arg {
name: "rhs"
type_attr: "T"
}
input_arg {
name: "l2_regularizer"
type: DT_DOUBLE
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_DOUBLE
type: DT_FLOAT
}
}
}
attr {
name: "fast"
type: "bool"
default_value {
b: true
}
}
deprecation {
version: 13
}
}

View File

@ -0,0 +1,42 @@
op {
name: "BatchMatrixTriangularSolve"
input_arg {
name: "matrix"
type_attr: "T"
}
input_arg {
name: "rhs"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "lower"
type: "bool"
default_value {
b: true
}
}
attr {
name: "adjoint"
type: "bool"
default_value {
b: false
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_DOUBLE
type: DT_FLOAT
}
}
}
deprecation {
version: 13
}
}

View File

@ -0,0 +1,248 @@
op {
name: "BatchNormWithGlobalNormalization"
input_arg {
name: "t"
type_attr: "T"
}
input_arg {
name: "m"
type_attr: "T"
}
input_arg {
name: "v"
type_attr: "T"
}
input_arg {
name: "beta"
type_attr: "T"
}
input_arg {
name: "gamma"
type_attr: "T"
}
output_arg {
name: "result"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "variance_epsilon"
type: "float"
}
attr {
name: "scale_after_normalization"
type: "bool"
}
deprecation {
version: 9
}
}
op {
name: "BatchNormWithGlobalNormalization"
input_arg {
name: "t"
type_attr: "T"
}
input_arg {
name: "m"
type_attr: "T"
}
input_arg {
name: "v"
type_attr: "T"
}
input_arg {
name: "beta"
type_attr: "T"
}
input_arg {
name: "gamma"
type_attr: "T"
}
output_arg {
name: "result"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "variance_epsilon"
type: "float"
}
attr {
name: "scale_after_normalization"
type: "bool"
}
deprecation {
version: 9
}
}
op {
name: "BatchNormWithGlobalNormalization"
input_arg {
name: "t"
type_attr: "T"
}
input_arg {
name: "m"
type_attr: "T"
}
input_arg {
name: "v"
type_attr: "T"
}
input_arg {
name: "beta"
type_attr: "T"
}
input_arg {
name: "gamma"
type_attr: "T"
}
output_arg {
name: "result"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "variance_epsilon"
type: "float"
}
attr {
name: "scale_after_normalization"
type: "bool"
}
deprecation {
version: 9
}
}
op {
name: "BatchNormWithGlobalNormalization"
input_arg {
name: "t"
type_attr: "T"
}
input_arg {
name: "m"
type_attr: "T"
}
input_arg {
name: "v"
type_attr: "T"
}
input_arg {
name: "beta"
type_attr: "T"
}
input_arg {
name: "gamma"
type_attr: "T"
}
output_arg {
name: "result"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "variance_epsilon"
type: "float"
}
attr {
name: "scale_after_normalization"
type: "bool"
}
deprecation {
version: 9
}
}

View File

@ -0,0 +1,312 @@
op {
name: "BatchNormWithGlobalNormalizationGrad"
input_arg {
name: "t"
type_attr: "T"
}
input_arg {
name: "m"
type_attr: "T"
}
input_arg {
name: "v"
type_attr: "T"
}
input_arg {
name: "gamma"
type_attr: "T"
}
input_arg {
name: "backprop"
type_attr: "T"
}
output_arg {
name: "dx"
type_attr: "T"
}
output_arg {
name: "dm"
type_attr: "T"
}
output_arg {
name: "dv"
type_attr: "T"
}
output_arg {
name: "db"
type_attr: "T"
}
output_arg {
name: "dg"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
}
}
}
attr {
name: "variance_epsilon"
type: "float"
}
attr {
name: "scale_after_normalization"
type: "bool"
}
deprecation {
version: 9
}
}
op {
name: "BatchNormWithGlobalNormalizationGrad"
input_arg {
name: "t"
type_attr: "T"
}
input_arg {
name: "m"
type_attr: "T"
}
input_arg {
name: "v"
type_attr: "T"
}
input_arg {
name: "gamma"
type_attr: "T"
}
input_arg {
name: "backprop"
type_attr: "T"
}
output_arg {
name: "dx"
type_attr: "T"
}
output_arg {
name: "dm"
type_attr: "T"
}
output_arg {
name: "dv"
type_attr: "T"
}
output_arg {
name: "db"
type_attr: "T"
}
output_arg {
name: "dg"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "variance_epsilon"
type: "float"
}
attr {
name: "scale_after_normalization"
type: "bool"
}
deprecation {
version: 9
}
}
op {
name: "BatchNormWithGlobalNormalizationGrad"
input_arg {
name: "t"
type_attr: "T"
}
input_arg {
name: "m"
type_attr: "T"
}
input_arg {
name: "v"
type_attr: "T"
}
input_arg {
name: "gamma"
type_attr: "T"
}
input_arg {
name: "backprop"
type_attr: "T"
}
output_arg {
name: "dx"
type_attr: "T"
}
output_arg {
name: "dm"
type_attr: "T"
}
output_arg {
name: "dv"
type_attr: "T"
}
output_arg {
name: "db"
type_attr: "T"
}
output_arg {
name: "dg"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT64
type: DT_INT32
type: DT_UINT8
type: DT_UINT16
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_COMPLEX128
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
attr {
name: "variance_epsilon"
type: "float"
}
attr {
name: "scale_after_normalization"
type: "bool"
}
deprecation {
version: 9
}
}
op {
name: "BatchNormWithGlobalNormalizationGrad"
input_arg {
name: "t"
type_attr: "T"
}
input_arg {
name: "m"
type_attr: "T"
}
input_arg {
name: "v"
type_attr: "T"
}
input_arg {
name: "gamma"
type_attr: "T"
}
input_arg {
name: "backprop"
type_attr: "T"
}
output_arg {
name: "dx"
type_attr: "T"
}
output_arg {
name: "dm"
type_attr: "T"
}
output_arg {
name: "dv"
type_attr: "T"
}
output_arg {
name: "db"
type_attr: "T"
}
output_arg {
name: "dg"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "variance_epsilon"
type: "float"
}
attr {
name: "scale_after_normalization"
type: "bool"
}
deprecation {
version: 9
}
}

Some files were not shown because too many files have changed in this diff Show More