diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Abort.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Abort.pbtxt index 58448c2d17b..7d90f6d9fc7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Abort.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Abort.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Abort" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Abs.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Abs.pbtxt index ece45cf73f3..5ae7934e3cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Abs.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Abs.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Abs" endpoint { name: "math.Abs" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulateNV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulateNV2.pbtxt index 0c7a080c115..ae2d6e0c7fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulateNV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulateNV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AccumulateNV2" endpoint { name: "math.AccumulateN" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorApplyGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorApplyGradient.pbtxt index 49b7acad7d8..ecf18bfde4d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorApplyGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorApplyGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AccumulatorApplyGradient" endpoint { name: "train.AccumulatorApplyGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorNumAccumulated.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorNumAccumulated.pbtxt index 1c42e819bf4..c9f5db313ee 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorNumAccumulated.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorNumAccumulated.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AccumulatorNumAccumulated" endpoint { name: "train.AccumulatorNumAccumulated" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorSetGlobalStep.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorSetGlobalStep.pbtxt index ca85302cdb4..53dbca3a28a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorSetGlobalStep.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorSetGlobalStep.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AccumulatorSetGlobalStep" endpoint { name: "train.AccumulatorSetGlobalStep" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorTakeGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorTakeGradient.pbtxt index 4883802c637..d8482bfef55 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorTakeGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AccumulatorTakeGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AccumulatorTakeGradient" endpoint { name: "train.AccumulatorTakeGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Acos.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Acos.pbtxt index 847986b429d..d730005b322 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Acos.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Acos.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Acos" endpoint { name: "math.Acos" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Acosh.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Acosh.pbtxt index 76d8f5fad05..7f880491eae 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Acosh.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Acosh.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Acosh" endpoint { name: "math.Acosh" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Add.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Add.pbtxt index 4f78ccc9ea6..b213eb8dd32 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Add.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Add.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Add" endpoint { name: "math.Add" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AddManySparseToTensorsMap.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AddManySparseToTensorsMap.pbtxt index e009ba19d34..8dcebf4c82b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AddManySparseToTensorsMap.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AddManySparseToTensorsMap.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AddManySparseToTensorsMap" endpoint { name: "sparse.AddManySparseToTensorsMap" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AddN.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AddN.pbtxt index 20d469ae731..8807e161276 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AddN.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AddN.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AddN" endpoint { name: "math.AddN" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AddSparseToTensorsMap.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AddSparseToTensorsMap.pbtxt index 0bb20186de3..d46dc06cd51 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AddSparseToTensorsMap.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AddSparseToTensorsMap.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AddSparseToTensorsMap" endpoint { name: "sparse.AddSparseToTensorsMap" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustContrastv2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustContrastv2.pbtxt index 81f565c1d59..bbf539a05de 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustContrastv2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustContrastv2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AdjustContrastv2" endpoint { name: "image.AdjustContrast" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustHue.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustHue.pbtxt index 0847cad4031..9cfca205fb5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustHue.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustHue.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AdjustHue" endpoint { name: "image.AdjustHue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustSaturation.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustSaturation.pbtxt index d685636eb12..679b1d48ab9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustSaturation.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AdjustSaturation.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AdjustSaturation" endpoint { name: "image.AdjustSaturation" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_All.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_All.pbtxt index a6459c56b71..89ab8929419 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_All.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_All.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "All" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AllCandidateSampler.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AllCandidateSampler.pbtxt index 607c208a460..2a260b630af 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AllCandidateSampler.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AllCandidateSampler.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AllCandidateSampler" endpoint { name: "random.AllCandidateSampler" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AllToAll.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AllToAll.pbtxt index fd03c0df632..1ce77f7d74a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AllToAll.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AllToAll.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AllToAll" endpoint { name: "tpu.AllToAll" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Angle.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Angle.pbtxt index a92ccf357db..fd3770221f8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Angle.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Angle.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Angle" endpoint { name: "math.Angle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousHashTable.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousHashTable.pbtxt index 5508431f163..5b60d123270 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousHashTable.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousHashTable.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "AnonymousHashTable" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMemoryCache.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMemoryCache.pbtxt index 030b1a92bfc..fcde7026956 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMemoryCache.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMemoryCache.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AnonymousMemoryCache" endpoint { name: "data.AnonymousMemoryCache" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableDenseHashTable.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableDenseHashTable.pbtxt index 1774d42d57f..fe75322c561 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableDenseHashTable.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableDenseHashTable.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "AnonymousMutableDenseHashTable" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableHashTable.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableHashTable.pbtxt index 80aec142155..69f531da488 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableHashTable.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableHashTable.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "AnonymousMutableHashTable" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableHashTableOfTensors.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableHashTableOfTensors.pbtxt index 069a81cbac7..409abc6f6d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableHashTableOfTensors.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousMutableHashTableOfTensors.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "AnonymousMutableHashTableOfTensors" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousRandomSeedGenerator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousRandomSeedGenerator.pbtxt index 50621a2ac4c..4c3c3cd98a6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousRandomSeedGenerator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousRandomSeedGenerator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AnonymousRandomSeedGenerator" endpoint { name: "random.AnonymousRandomSeedGenerator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousSeedGenerator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousSeedGenerator.pbtxt index a3fb16094ef..cf4c8f4f339 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousSeedGenerator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AnonymousSeedGenerator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AnonymousSeedGenerator" endpoint { name: "random.AnonymousSeedGenerator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Any.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Any.pbtxt index 20b36eda3f8..c96baa7525d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Any.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Any.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Any" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdaMax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdaMax.pbtxt index 583f164e06c..b552249c876 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdaMax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdaMax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyAdaMax" endpoint { name: "train.ApplyAdaMax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdadelta.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdadelta.pbtxt index e672a8ef03b..e16875bc976 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdadelta.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdadelta.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyAdadelta" endpoint { name: "train.ApplyAdadelta" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagrad.pbtxt index 980c57c5fec..3de2b67d1b5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyAdagrad" endpoint { name: "train.ApplyAdagrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagradDA.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagradDA.pbtxt index 815df985ef9..e51c4bd8155 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagradDA.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagradDA.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyAdagradDA" endpoint { name: "train.ApplyAdagradDa" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagradV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagradV2.pbtxt index 34cfdb57ab8..cfa90ac82c2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagradV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdagradV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyAdagradV2" endpoint { name: "train.ApplyAdagradV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdam.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdam.pbtxt index 56461b1d3d5..85ff2d1bad3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdam.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAdam.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyAdam" endpoint { name: "train.ApplyAdam" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAddSign.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAddSign.pbtxt index b54ff6eca44..21a5f40a078 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAddSign.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyAddSign.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyAddSign" endpoint { name: "train.ApplyAddSign" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyCenteredRMSProp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyCenteredRMSProp.pbtxt index 1b831bca436..ec1b6380779 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyCenteredRMSProp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyCenteredRMSProp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyCenteredRMSProp" endpoint { name: "train.ApplyCenteredRmsProp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyFtrlV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyFtrlV2.pbtxt index da0fc8fcbf7..08a86347aef 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyFtrlV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyFtrlV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyFtrlV2" endpoint { name: "train.ApplyFtrl" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyGradientDescent.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyGradientDescent.pbtxt index 1fa569ed329..335095ef520 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyGradientDescent.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyGradientDescent.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyGradientDescent" endpoint { name: "train.ApplyGradientDescent" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyMomentum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyMomentum.pbtxt index 96c21199f09..4a7079316b4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyMomentum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyMomentum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyMomentum" endpoint { name: "train.ApplyMomentum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyPowerSign.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyPowerSign.pbtxt index e5c22347556..0a816803266 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyPowerSign.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyPowerSign.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyPowerSign" endpoint { name: "train.ApplyPowerSign" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyProximalAdagrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyProximalAdagrad.pbtxt index a52d8c3591c..774d00e707c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyProximalAdagrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyProximalAdagrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyProximalAdagrad" endpoint { name: "train.ApplyProximalAdagrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyProximalGradientDescent.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyProximalGradientDescent.pbtxt index 74ea29cf888..3458df77763 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyProximalGradientDescent.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyProximalGradientDescent.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyProximalGradientDescent" endpoint { name: "train.ApplyProximalGradientDescent" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyRMSProp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyRMSProp.pbtxt index 90171ccc759..259b5512e16 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyRMSProp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApplyRMSProp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApplyRMSProp" endpoint { name: "train.ApplyRmsProp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApproxTopK.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApproxTopK.pbtxt index da26057b420..51b0cc7c01f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApproxTopK.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApproxTopK.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ApproxTopK" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApproximateEqual.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApproximateEqual.pbtxt index 029dc6d2930..d392987d60a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ApproximateEqual.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ApproximateEqual.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ApproximateEqual" endpoint { name: "math.ApproximateEqual" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ArgMax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ArgMax.pbtxt index f9effd49c4a..5627186359b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ArgMax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ArgMax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ArgMax" endpoint { name: "math.ArgMax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ArgMin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ArgMin.pbtxt index 5ff04c0d1ab..e01e5f2e72b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ArgMin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ArgMin.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ArgMin" endpoint { name: "math.ArgMin" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AsString.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AsString.pbtxt index e8c875ea814..a020c7aef85 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AsString.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AsString.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AsString" endpoint { name: "dtypes.AsString" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Asin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Asin.pbtxt index 8ffc8e3e570..7b71c08eede 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Asin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Asin.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Asin" endpoint { name: "math.Asin" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Asinh.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Asinh.pbtxt index e3b30dd5125..2a371a10071 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Asinh.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Asinh.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Asinh" endpoint { name: "math.Asinh" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Assert.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Assert.pbtxt index a9e107b4780..44d1ce33dd7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Assert.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Assert.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Assert" endpoint { name: "AssertThat" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssertPrevDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssertPrevDataset.pbtxt index 610d5f8216e..246fdd58a4a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssertPrevDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssertPrevDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AssertPrevDataset" endpoint { name: "data.AssertPrevDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Assign.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Assign.pbtxt index 15d778f61e8..51c43e54d2e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Assign.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Assign.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Assign" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignAdd.pbtxt index a4118b64afd..9f29218e945 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignAdd.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "AssignAdd" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignAddVariableOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignAddVariableOp.pbtxt index 05fecb191bf..f724f706878 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignAddVariableOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignAddVariableOp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "AssignAddVariableOp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignSub.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignSub.pbtxt index aaf9246a6ac..a492c335154 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignSub.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignSub.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "AssignSub" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignSubVariableOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignSubVariableOp.pbtxt index 6e8791aed2d..768f4c47169 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignSubVariableOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignSubVariableOp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "AssignSubVariableOp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignVariableOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignVariableOp.pbtxt index 980e6968269..9e61072ca68 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignVariableOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignVariableOp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "AssignVariableOp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignVariableXlaConcatND.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignVariableXlaConcatND.pbtxt index f45351cddeb..9bf3d7734a6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignVariableXlaConcatND.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AssignVariableXlaConcatND.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AssignVariableXlaConcatND" endpoint { name: "xla.AssignVariableConcatND" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Atan.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Atan.pbtxt index e51aee9abc4..bb00076b52d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Atan.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Atan.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Atan" endpoint { name: "math.Atan" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Atan2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Atan2.pbtxt index 302b05f9dce..f313a44b032 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Atan2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Atan2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Atan2" endpoint { name: "math.Atan2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Atanh.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Atanh.pbtxt index b9c4a411544..59e98471ce1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Atanh.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Atanh.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Atanh" endpoint { name: "math.Atanh" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AudioSpectrogram.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AudioSpectrogram.pbtxt index bd8f3a5e335..8731927d50c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AudioSpectrogram.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AudioSpectrogram.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AudioSpectrogram" endpoint { name: "audio.AudioSpectrogram" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AudioSummaryV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AudioSummaryV2.pbtxt index e4eda8b09ab..954dbf9bb50 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AudioSummaryV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AudioSummaryV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AudioSummaryV2" endpoint { name: "summary.AudioSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool.pbtxt index 10d87802f0d..970557d9c96 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AvgPool" endpoint { name: "nn.AvgPool" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool3D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool3D.pbtxt index 1ae2794f48b..be8667cf31c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool3D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool3D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AvgPool3D" endpoint { name: "nn.AvgPool3d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool3DGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool3DGrad.pbtxt index 09aba78ca20..6bc2df28667 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool3DGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPool3DGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AvgPool3DGrad" endpoint { name: "nn.AvgPool3dGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPoolGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPoolGrad.pbtxt index fc8fec89b99..097ba7213f1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPoolGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_AvgPoolGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "AvgPoolGrad" endpoint { name: "nn.AvgPoolGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BandedTriangularSolve.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BandedTriangularSolve.pbtxt index 9be002004ce..9cf217624c6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BandedTriangularSolve.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BandedTriangularSolve.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BandedTriangularSolve" endpoint { name: "linalg.BandedTriangularSolve" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Barrier.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Barrier.pbtxt index 6e282ca7b39..7aada11ec00 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Barrier.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Barrier.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Barrier" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierClose.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierClose.pbtxt index 0307318763b..75d923401c4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierClose.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierClose.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "BarrierClose" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierIncompleteSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierIncompleteSize.pbtxt index a2fed2a43de..53729fe5652 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierIncompleteSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierIncompleteSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BarrierIncompleteSize" out_arg { name: "size" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierInsertMany.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierInsertMany.pbtxt index 32e29f00158..163cfbeae5b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierInsertMany.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierInsertMany.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "BarrierInsertMany" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierReadySize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierReadySize.pbtxt index 0f768476610..f648bb15560 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierReadySize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierReadySize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BarrierReadySize" out_arg { name: "size" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierTakeMany.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierTakeMany.pbtxt index 21f08878c6d..5c6508a6963 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierTakeMany.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BarrierTakeMany.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "BarrierTakeMany" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchCholesky.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchCholesky.pbtxt index 15048109fd6..c1cdb6b892e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchCholesky.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchCholesky.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchCholesky" endpoint { name: "linalg.BatchCholesky" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchCholeskyGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchCholeskyGrad.pbtxt index eb0e2c6bc83..c8e9b4060e7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchCholeskyGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchCholeskyGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchCholeskyGrad" endpoint { name: "linalg.BatchCholeskyGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT.pbtxt index 4dda7c1fb61..cf02316b08c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchFFT" endpoint { name: "signal.BatchFft" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT2D.pbtxt index e11860138a2..4b09c73a82b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchFFT2D" endpoint { name: "signal.BatchFft2d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT3D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT3D.pbtxt index 3be0b516d0e..0b4cdfac071 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT3D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFFT3D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchFFT3D" endpoint { name: "signal.BatchFft3d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFunction.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFunction.pbtxt index 8789dc6acb6..2160e9f7b8a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFunction.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchFunction.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "BatchFunction" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT.pbtxt index de37ada148a..491d21ad4c4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchIFFT" endpoint { name: "signal.BatchIfft" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT2D.pbtxt index 4ae7fb4cb0a..61a773b3f76 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchIFFT2D" endpoint { name: "signal.BatchIfft2d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT3D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT3D.pbtxt index 0ecb52714b5..6111f4c6006 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT3D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchIFFT3D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchIFFT3D" endpoint { name: "signal.BatchIfft3d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatMulV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatMulV3.pbtxt index 463513824d1..8a70e8e6e55 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatMulV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatMulV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchMatMulV3" endpoint { name: "train.BatchMatMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixBandPart.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixBandPart.pbtxt index de989c6d527..af80b346df8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixBandPart.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixBandPart.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchMatrixBandPart" endpoint { name: "linalg.BatchMatrixBandPart" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDeterminant.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDeterminant.pbtxt index a45fe25d10c..ac3c9b2a5ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDeterminant.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDeterminant.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchMatrixDeterminant" endpoint { name: "linalg.BatchMatrixDeterminant" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDiag.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDiag.pbtxt index d85d76f02f0..c30ccfb3e28 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDiag.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDiag.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchMatrixDiag" endpoint { name: "linalg.BatchMatrixDiag" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDiagPart.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDiagPart.pbtxt index 4b5350b11ee..cf215430e8e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDiagPart.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixDiagPart.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchMatrixDiagPart" endpoint { name: "linalg.BatchMatrixDiagPart" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixInverse.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixInverse.pbtxt index f40ea50d4bc..113f9e268d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixInverse.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixInverse.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchMatrixInverse" endpoint { name: "linalg.BatchMatrixInverse" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSetDiag.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSetDiag.pbtxt index ac4cd6889b6..4d402f61466 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSetDiag.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSetDiag.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchMatrixSetDiag" endpoint { name: "linalg.BatchMatrixSetDiag" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSolve.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSolve.pbtxt index 97435acb4e4..2b5a9c70205 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSolve.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSolve.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchMatrixSolve" endpoint { name: "linalg.BatchMatrixSolve" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSolveLs.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSolveLs.pbtxt index aee0b4add35..b95a4b7f1aa 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSolveLs.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixSolveLs.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchMatrixSolveLs" endpoint { name: "linalg.BatchMatrixSolveLs" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixTriangularSolve.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixTriangularSolve.pbtxt index 554eff15747..39f614c58a2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixTriangularSolve.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchMatrixTriangularSolve.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchMatrixTriangularSolve" endpoint { name: "linalg.BatchMatrixTriangularSolve" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchNormWithGlobalNormalization.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchNormWithGlobalNormalization.pbtxt index 8285ac284d8..0b8ed84a609 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchNormWithGlobalNormalization.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchNormWithGlobalNormalization.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchNormWithGlobalNormalization" endpoint { name: "nn.BatchNormWithGlobalNormalization" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchNormWithGlobalNormalizationGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchNormWithGlobalNormalizationGrad.pbtxt index 7b18bf52acc..4aa3b421147 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchNormWithGlobalNormalizationGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchNormWithGlobalNormalizationGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchNormWithGlobalNormalizationGrad" endpoint { name: "nn.BatchNormWithGlobalNormalizationGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchSelfAdjointEigV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchSelfAdjointEigV2.pbtxt index 9c973443902..4137098cf32 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchSelfAdjointEigV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchSelfAdjointEigV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchSelfAdjointEigV2" endpoint { name: "linalg.BatchSelfAdjointEig" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchSvd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchSvd.pbtxt index 8696359df8d..73f619b157c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchSvd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchSvd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchSvd" endpoint { name: "linalg.BatchSvd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchToSpace.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchToSpace.pbtxt index affbc519e51..2cd926bf567 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchToSpace.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchToSpace.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "BatchToSpace" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchToSpaceND.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchToSpaceND.pbtxt index 6c7d2fbdb9f..93d4335ac31 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchToSpaceND.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BatchToSpaceND.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BatchToSpaceND" endpoint { name: "BatchToSpaceNd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI0.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI0.pbtxt index 71b2007c630..88301e94ba7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI0.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI0.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselI0" endpoint { name: "math.BesselI0" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI0e.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI0e.pbtxt index 84eb3b5e71d..f80adf8b7e6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI0e.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI0e.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselI0e" endpoint { name: "math.BesselI0e" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI1.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI1.pbtxt index 9bb74686ef3..bbba9f7549f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI1.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI1.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselI1" endpoint { name: "math.BesselI1" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI1e.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI1e.pbtxt index 43f9113b0bb..e91b37684b8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI1e.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselI1e.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselI1e" endpoint { name: "math.BesselI1e" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselJ0.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselJ0.pbtxt index c9e7d5fad2e..1898e526094 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselJ0.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselJ0.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselJ0" endpoint { name: "math.special.BesselJ0" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselJ1.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselJ1.pbtxt index 39f7cc0513c..cbe95c525cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselJ1.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselJ1.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselJ1" endpoint { name: "math.special.BesselJ1" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK0.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK0.pbtxt index 670d0ce8ff3..ba380554645 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK0.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK0.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselK0" endpoint { name: "math.special.BesselK0" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK0e.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK0e.pbtxt index 3cfac3d9c1b..09659504093 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK0e.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK0e.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselK0e" endpoint { name: "math.special.BesselK0e" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK1.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK1.pbtxt index 6672d29a76d..91c3f998864 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK1.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK1.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselK1" endpoint { name: "math.special.BesselK1" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK1e.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK1e.pbtxt index 64c89fc43d4..334c1025b5f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK1e.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselK1e.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselK1e" endpoint { name: "math.special.BesselK1e" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselY0.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselY0.pbtxt index d55b57a5ccc..a813593994b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselY0.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselY0.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselY0" endpoint { name: "math.special.BesselY0" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselY1.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselY1.pbtxt index 76b7c0afff8..cb7a004e1a2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselY1.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BesselY1.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BesselY1" endpoint { name: "math.special.BesselY1" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Betainc.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Betainc.pbtxt index 6e9956d9ec7..1931537fa76 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Betainc.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Betainc.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Betainc" endpoint { name: "math.Betainc" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BiasAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BiasAdd.pbtxt index eb3be23bd9a..fa509206f83 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BiasAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BiasAdd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BiasAdd" endpoint { name: "nn.BiasAdd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BiasAddGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BiasAddGrad.pbtxt index 4e040bf6df8..f36f4d41ca5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BiasAddGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BiasAddGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BiasAddGrad" endpoint { name: "nn.BiasAddGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BigQueryReader.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BigQueryReader.pbtxt index 5b6e11687a2..b98f8304793 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BigQueryReader.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BigQueryReader.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BigQueryReader" endpoint { name: "io.BigQueryReader" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Bincount.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Bincount.pbtxt index b894fd6ec5e..d16999a510b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Bincount.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Bincount.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Bincount" endpoint { name: "math.Bincount" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Bitcast.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Bitcast.pbtxt index 9d2db26851d..0b55c90620a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Bitcast.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Bitcast.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Bitcast" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseAnd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseAnd.pbtxt index db5fada2461..0b791ac5dda 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseAnd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseAnd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BitwiseAnd" endpoint { name: "bitwise.BitwiseAnd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseOr.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseOr.pbtxt index 8f9d1bc2fe4..45796b0bf30 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseOr.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseOr.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BitwiseOr" endpoint { name: "bitwise.BitwiseOr" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseXor.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseXor.pbtxt index 28f405b8ada..c83fee544c6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseXor.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BitwiseXor.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BitwiseXor" endpoint { name: "bitwise.BitwiseXor" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BlockLSTMGradV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BlockLSTMGradV2.pbtxt index ed3ee5ce114..d88c6c62f86 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BlockLSTMGradV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BlockLSTMGradV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BlockLSTMGradV2" endpoint { name: "nn.BlockLSTMGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BlockLSTMV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BlockLSTMV2.pbtxt index b28283c1332..f20e824d7dc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BlockLSTMV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BlockLSTMV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "BlockLSTMV2" endpoint { name: "nn.BlockLSTM" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesAggregateStats.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesAggregateStats.pbtxt index 3d48e24f535..58978e6b6ba 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesAggregateStats.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesAggregateStats.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesAggregateStats" endpoint { name: "estimator.BoostedTreesAggregateStats" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesBucketize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesBucketize.pbtxt index 9580285ea0e..d55fffeb182 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesBucketize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesBucketize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesBucketize" endpoint { name: "estimator.BoostedTreesBucketize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestFeatureSplit.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestFeatureSplit.pbtxt index d175e607580..43ce3d8a8b8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestFeatureSplit.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestFeatureSplit.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesCalculateBestFeatureSplit" endpoint { name: "estimator.BoostedTreesCalculateBestFeatureSplit" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestFeatureSplitV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestFeatureSplitV2.pbtxt index 3af78f46d8d..d920e9bf6b5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestFeatureSplitV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestFeatureSplitV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesCalculateBestFeatureSplitV2" endpoint { name: "estimator.BoostedTreesCalculateBestFeatureSplitV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestGainsPerFeature.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestGainsPerFeature.pbtxt index 87685cceff3..cab624efd61 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestGainsPerFeature.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCalculateBestGainsPerFeature.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesCalculateBestGainsPerFeature" endpoint { name: "estimator.BoostedTreesCalculateBestGainsPerFeature" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCenterBias.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCenterBias.pbtxt index d4c4e63ffb6..055cb5b067d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCenterBias.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCenterBias.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesCenterBias" endpoint { name: "estimator.BoostedTreesCenterBias" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCreateEnsemble.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCreateEnsemble.pbtxt index ecd5599e03d..01e25eb2270 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCreateEnsemble.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCreateEnsemble.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesCreateEnsemble" endpoint { name: "estimator.BoostedTreesCreateEnsemble" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCreateQuantileStreamResource.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCreateQuantileStreamResource.pbtxt index 42091b1f414..7105d2a13ca 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCreateQuantileStreamResource.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesCreateQuantileStreamResource.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesCreateQuantileStreamResource" endpoint { name: "estimator.BoostedTreesCreateQuantileStreamResource" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesDeserializeEnsemble.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesDeserializeEnsemble.pbtxt index b01487ac55c..7dbb508bad1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesDeserializeEnsemble.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesDeserializeEnsemble.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesDeserializeEnsemble" endpoint { name: "estimator.BoostedTreesDeserializeEnsemble" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesEnsembleResourceHandleOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesEnsembleResourceHandleOp.pbtxt index 36437e93985..43f0f618a9d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesEnsembleResourceHandleOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesEnsembleResourceHandleOp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesEnsembleResourceHandleOp" endpoint { name: "estimator.BoostedTreesEnsembleResourceHandleOp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesExampleDebugOutputs.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesExampleDebugOutputs.pbtxt index 83c98a3ed17..0768f7ea464 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesExampleDebugOutputs.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesExampleDebugOutputs.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesExampleDebugOutputs" endpoint { name: "estimator.BoostedTreesExampleDebugOutputs" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesFlushQuantileSummaries.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesFlushQuantileSummaries.pbtxt index 2b42aaa4487..c5949350c42 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesFlushQuantileSummaries.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesFlushQuantileSummaries.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesFlushQuantileSummaries" endpoint { name: "estimator.BoostedTreesFlushQuantileSummaries" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesGetEnsembleStates.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesGetEnsembleStates.pbtxt index f02b68487ac..1973e3ce0b6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesGetEnsembleStates.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesGetEnsembleStates.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesGetEnsembleStates" endpoint { name: "estimator.BoostedTreesGetEnsembleStates" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesMakeQuantileSummaries.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesMakeQuantileSummaries.pbtxt index c89c68d527d..f4de8855e9a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesMakeQuantileSummaries.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesMakeQuantileSummaries.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesMakeQuantileSummaries" endpoint { name: "estimator.BoostedTreesMakeQuantileSummaries" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesMakeStatsSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesMakeStatsSummary.pbtxt index e39d6a2495c..5414e2aae97 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesMakeStatsSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesMakeStatsSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesMakeStatsSummary" endpoint { name: "estimator.BoostedTreesMakeStatsSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesPredict.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesPredict.pbtxt index 59aab04d53a..7c93fcfdfc2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesPredict.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesPredict.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesPredict" endpoint { name: "estimator.BoostedTreesPredict" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceAddSummaries.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceAddSummaries.pbtxt index 656360418a9..ab449a57d5c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceAddSummaries.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceAddSummaries.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesQuantileStreamResourceAddSummaries" endpoint { name: "estimator.BoostedTreesQuantileStreamResourceAddSummaries" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceDeserialize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceDeserialize.pbtxt index 5fde27fd5b6..45103ae088a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceDeserialize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceDeserialize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesQuantileStreamResourceDeserialize" endpoint { name: "estimator.BoostedTreesQuantileStreamResourceDeserialize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceFlush.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceFlush.pbtxt index 2c9b8936d74..16b68e4ac83 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceFlush.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceFlush.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesQuantileStreamResourceFlush" endpoint { name: "estimator.BoostedTreesQuantileStreamResourceFlush" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceGetBucketBoundaries.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceGetBucketBoundaries.pbtxt index 1d4ff1aeece..990abb4effe 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceGetBucketBoundaries.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceGetBucketBoundaries.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesQuantileStreamResourceGetBucketBoundaries" endpoint { name: "estimator.BoostedTreesQuantileStreamResourceGetBucketBoundaries" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceHandleOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceHandleOp.pbtxt index 599bf156074..12600896ec9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceHandleOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesQuantileStreamResourceHandleOp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesQuantileStreamResourceHandleOp" endpoint { name: "estimator.BoostedTreesQuantileStreamResourceHandleOp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSerializeEnsemble.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSerializeEnsemble.pbtxt index 106a8a98309..5880c132063 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSerializeEnsemble.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSerializeEnsemble.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesSerializeEnsemble" endpoint { name: "estimator.BoostedTreesSerializeEnsemble" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSparseAggregateStats.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSparseAggregateStats.pbtxt index 0493060496b..109f3bae4e2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSparseAggregateStats.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSparseAggregateStats.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesSparseAggregateStats" endpoint { name: "estimator.BoostedTreesSparseAggregateStats" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSparseCalculateBestFeatureSplit.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSparseCalculateBestFeatureSplit.pbtxt index 81067bb6cbd..aae4c225f7e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSparseCalculateBestFeatureSplit.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesSparseCalculateBestFeatureSplit.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesSparseCalculateBestFeatureSplit" endpoint { name: "estimator.BoostedTreesSparseCalculateBestFeatureSplit" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesTrainingPredict.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesTrainingPredict.pbtxt index 4d1fa91560e..d4696dc6182 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesTrainingPredict.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesTrainingPredict.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesTrainingPredict" endpoint { name: "estimator.BoostedTreesTrainingPredict" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesUpdateEnsemble.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesUpdateEnsemble.pbtxt index 7a330f04e2e..77f30bc409f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesUpdateEnsemble.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesUpdateEnsemble.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesUpdateEnsemble" endpoint { name: "estimator.BoostedTreesUpdateEnsemble" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesUpdateEnsembleV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesUpdateEnsembleV2.pbtxt index f4e1665b61c..df4e978b422 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesUpdateEnsembleV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BoostedTreesUpdateEnsembleV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "BoostedTreesUpdateEnsembleV2" endpoint { name: "estimator.BoostedTreesUpdateEnsembleV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastArgs.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastArgs.pbtxt index 484742a2d02..ebc44eacd85 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastArgs.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastArgs.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "BroadcastArgs" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastGradientArgs.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastGradientArgs.pbtxt index 50f631b2a69..6e6f0d1b9b7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastGradientArgs.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastGradientArgs.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "BroadcastGradientArgs" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastTo.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastTo.pbtxt index 127458816ce..c5b07af0a18 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastTo.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_BroadcastTo.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "BroadcastTo" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Bucketize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Bucketize.pbtxt index 5a99712fd6c..a600ac3634d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Bucketize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Bucketize.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Bucketize" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixComponents.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixComponents.pbtxt index 552f52a21a9..24b7e34e16b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixComponents.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixComponents.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CSRSparseMatrixComponents" endpoint { name: "linalg.sparse.CSRSparseMatrixComponents" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixToDense.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixToDense.pbtxt index d85bf95da9c..62baeff7b47 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixToDense.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixToDense.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CSRSparseMatrixToDense" endpoint { name: "linalg.sparse.CSRSparseMatrixToDense" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixToSparseTensor.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixToSparseTensor.pbtxt index 7dc0ac62652..6be3fd9219b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixToSparseTensor.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CSRSparseMatrixToSparseTensor.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CSRSparseMatrixToSparseTensor" endpoint { name: "linalg.sparse.CSRSparseMatrixToSparseTensor" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCBeamSearchDecoder.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCBeamSearchDecoder.pbtxt index 39739f03a31..113d683f6be 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCBeamSearchDecoder.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCBeamSearchDecoder.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CTCBeamSearchDecoder" endpoint { name: "nn.CtcBeamSearchDecoder" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCGreedyDecoder.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCGreedyDecoder.pbtxt index 009742f0973..f82f1789f23 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCGreedyDecoder.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCGreedyDecoder.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CTCGreedyDecoder" endpoint { name: "nn.CtcGreedyDecoder" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCLoss.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCLoss.pbtxt index dbeefa40171..0c4d2f7843a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCLoss.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCLoss.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CTCLoss" endpoint { name: "nn.CtcLoss" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCLossV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCLossV2.pbtxt index 97df8d7a826..4ea107e1445 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCLossV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CTCLossV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CTCLossV2" endpoint { name: "nn.CTCLossV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Case.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Case.pbtxt index f1da05d19f2..eb371486f04 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Case.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Case.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Case" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cast.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cast.pbtxt index ea9f812e2a1..bd6b1b27204 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cast.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cast.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Cast" endpoint { name: "dtypes.Cast" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Ceil.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Ceil.pbtxt index d1a75f27d9a..41c23c44712 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Ceil.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Ceil.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Ceil" endpoint { name: "math.Ceil" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CheckNumericsV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CheckNumericsV2.pbtxt index cfae1f1cd11..3085f985715 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CheckNumericsV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CheckNumericsV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CheckNumericsV2" endpoint { name: "debugging.CheckNumerics" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cholesky.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cholesky.pbtxt index a60c4e3663e..0c1f48317d1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cholesky.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cholesky.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Cholesky" endpoint { name: "linalg.Cholesky" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CholeskyGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CholeskyGrad.pbtxt index 2601d415542..22e4aa89a6f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CholeskyGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CholeskyGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CholeskyGrad" endpoint { name: "linalg.CholeskyGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ClipByValue.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ClipByValue.pbtxt index c4e055c117c..b6c8fae964f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ClipByValue.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ClipByValue.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ClipByValue" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CloseSummaryWriter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CloseSummaryWriter.pbtxt index d5fbe557db0..2d1ca9631d3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CloseSummaryWriter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CloseSummaryWriter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CloseSummaryWriter" endpoint { name: "summary.CloseSummaryWriter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollateTPUEmbeddingMemory.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollateTPUEmbeddingMemory.pbtxt index 9821c816ce5..7e2b1aef93b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollateTPUEmbeddingMemory.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollateTPUEmbeddingMemory.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CollateTPUEmbeddingMemory" endpoint { name: "tpu.CollateTPUEmbeddingMemory" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveAllToAllV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveAllToAllV3.pbtxt index 7b45fb28bc9..b2356ee5b36 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveAllToAllV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveAllToAllV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CollectiveAllToAllV3" endpoint { name: "collective.CollectiveAllToAll" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveAssignGroupV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveAssignGroupV2.pbtxt index 03e7c97aeb5..d414cd66079 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveAssignGroupV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveAssignGroupV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CollectiveAssignGroupV2" endpoint { name: "collective.CollectiveAssignGroup" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveBcastRecvV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveBcastRecvV2.pbtxt index c7d78492cab..be74a35b7f9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveBcastRecvV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveBcastRecvV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CollectiveBcastRecvV2" endpoint { name: "collective.CollectiveBcastRecv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveBcastSendV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveBcastSendV2.pbtxt index 9eb747f9a4b..1fb22afed54 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveBcastSendV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveBcastSendV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CollectiveBcastSendV2" endpoint { name: "collective.CollectiveBcastSend" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveGatherV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveGatherV2.pbtxt index a9179d98926..d220f2ab11f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveGatherV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveGatherV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CollectiveGatherV2" endpoint: { name: "collective.CollectiveGather" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveInitializeCommunicator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveInitializeCommunicator.pbtxt index de44c83cdf0..fba9e620843 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveInitializeCommunicator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveInitializeCommunicator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CollectiveInitializeCommunicator" endpoint { name: "collective.CollectiveInitializeCommunicator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectivePermute.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectivePermute.pbtxt index 490242ba9c5..5fa5a659df4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectivePermute.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectivePermute.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CollectivePermute" endpoint { name: "collective.CollectivePermute" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveReduceScatterV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveReduceScatterV2.pbtxt index 07cf7f4a7c0..b36c3830ca1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveReduceScatterV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveReduceScatterV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CollectiveReduceScatterV2" endpoint { name: "collective.CollectiveReduceScatter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveReduceV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveReduceV3.pbtxt index c7234159624..3a2779461d2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveReduceV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CollectiveReduceV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CollectiveReduceV3" endpoint { name: "collective.CollectiveReduce" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CombinedNonMaxSuppression.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CombinedNonMaxSuppression.pbtxt index 44baf4d4b3e..836a46a42b2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CombinedNonMaxSuppression.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CombinedNonMaxSuppression.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CombinedNonMaxSuppression" endpoint { name: "image.CombinedNonMaxSuppression" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CompareAndBitpack.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CompareAndBitpack.pbtxt index d744fbbc90f..4e5a5e1a2af 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CompareAndBitpack.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CompareAndBitpack.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CompareAndBitpack" endpoint { name: "math.CompareAndBitpack" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Complex.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Complex.pbtxt index 4889360a96a..f649707afb8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Complex.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Complex.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Complex" endpoint { name: "dtypes.Complex" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComplexAbs.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComplexAbs.pbtxt index 42a6a3c6a1c..be6aa59c92e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComplexAbs.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComplexAbs.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ComplexAbs" endpoint { name: "math.ComplexAbs" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CompositeTensorVariantFromComponents.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CompositeTensorVariantFromComponents.pbtxt index 2a115714525..adb638940d8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CompositeTensorVariantFromComponents.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CompositeTensorVariantFromComponents.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "CompositeTensorVariantFromComponents" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CompositeTensorVariantToComponents.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CompositeTensorVariantToComponents.pbtxt index 49f121eef4a..b34054ead77 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CompositeTensorVariantToComponents.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CompositeTensorVariantToComponents.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "CompositeTensorVariantToComponents" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CompressElement.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CompressElement.pbtxt index 08158de13a5..09a543581d2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CompressElement.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CompressElement.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CompressElement" endpoint { name: "data.CompressElement" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeAccidentalHits.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeAccidentalHits.pbtxt index ca9e590fbce..8c4d834016b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeAccidentalHits.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeAccidentalHits.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ComputeAccidentalHits" endpoint { name: "nn.ComputeAccidentalHits" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeBatchSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeBatchSize.pbtxt index fd383570c1a..826f51ac87d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeBatchSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeBatchSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ComputeBatchSize" endpoint { name: "train.ComputeBatchSize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataTupleMask.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataTupleMask.pbtxt index 933b66ddf4e..946eab7e85a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataTupleMask.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ComputeDedupDataTupleMask.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ComputeDedupDataTupleMask" endpoint { name: "tpu.ComputeDedupDataTupleMask" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConcatOffset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConcatOffset.pbtxt index e8e23cf5593..876db502770 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConcatOffset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConcatOffset.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ConcatOffset" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConcatV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConcatV2.pbtxt index 7035796981f..9bf9a9b8648 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConcatV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConcatV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ConcatV2" endpoint { name: "Concat" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConditionalAccumulator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConditionalAccumulator.pbtxt index 08431982daa..3e8dd5299a1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConditionalAccumulator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConditionalAccumulator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ConditionalAccumulator" endpoint { name: "train.ConditionalAccumulator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureAndInitializeGlobalTPU.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureAndInitializeGlobalTPU.pbtxt index 9e41e0fad15..5ee6c848dee 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureAndInitializeGlobalTPU.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureAndInitializeGlobalTPU.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ConfigureAndInitializeGlobalTPU" endpoint { name: "tpu.ConfigureAndInitializeGlobalTPU" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureDistributedTPU.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureDistributedTPU.pbtxt index 2722e41b916..1dc468d8666 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureDistributedTPU.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureDistributedTPU.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ConfigureDistributedTPU" endpoint { name: "tpu.ConfigureDistributedTPU" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbedding.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbedding.pbtxt index 74bf19cfa9d..1cd8caf6d34 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbedding.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbedding.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ConfigureTPUEmbedding" endpoint { name: "tpu.ConfigureTPUEmbedding" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbeddingHost.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbeddingHost.pbtxt index 7095f59d3ae..aa4265b80ba 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbeddingHost.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbeddingHost.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ConfigureTPUEmbeddingHost" endpoint { name: "tpu.ConfigureTPUEmbeddingHost" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbeddingMemory.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbeddingMemory.pbtxt index f72a9a9804d..51b142d5c15 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbeddingMemory.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConfigureTPUEmbeddingMemory.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ConfigureTPUEmbeddingMemory" endpoint { name: "tpu.ConfigureTPUEmbeddingMemory" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conj.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conj.pbtxt index 7de199b55fa..0fb1ddc5788 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conj.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conj.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Conj" endpoint { name: "math.Conj" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConjugateTranspose.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConjugateTranspose.pbtxt index 42173088ae0..42fad3b7ee6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConjugateTranspose.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConjugateTranspose.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ConjugateTranspose" endpoint { name: "linalg.ConjugateTranspose" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConnectTPUEmbeddingHosts.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConnectTPUEmbeddingHosts.pbtxt index a59a9718d1f..030cd71468e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConnectTPUEmbeddingHosts.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConnectTPUEmbeddingHosts.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ConnectTPUEmbeddingHosts" endpoint { name: "tpu.ConnectTPUEmbeddingHosts" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConsumeMutexLock.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConsumeMutexLock.pbtxt index 1e0d136bc2f..78c8099b9ac 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ConsumeMutexLock.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ConsumeMutexLock.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ConsumeMutexLock" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ControlTrigger.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ControlTrigger.pbtxt index 4517b4373f3..8dc64a98773 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ControlTrigger.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ControlTrigger.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ControlTrigger" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv.pbtxt index 17ce54871b6..cdc59f52e68 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Conv" endpoint { name: "nn.Conv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv2D.pbtxt index 21d1398e098..1752f424f38 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Conv2D" endpoint { name: "nn.Conv2d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3D.pbtxt index 6ee1befcff1..abafc5a703d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Conv3D" endpoint { name: "nn.Conv3d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3DBackpropFilterV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3DBackpropFilterV2.pbtxt index 0643cc14a9a..257a0e6f7fe 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3DBackpropFilterV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3DBackpropFilterV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Conv3DBackpropFilterV2" endpoint { name: "nn.Conv3dBackpropFilter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3DBackpropInputV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3DBackpropInputV2.pbtxt index 33c8f5a3ce0..e192e5feedc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3DBackpropInputV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Conv3DBackpropInputV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Conv3DBackpropInputV2" endpoint { name: "nn.Conv3dBackpropInput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CopyToMesh.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CopyToMesh.pbtxt index 4228903ebde..e70bf4ade58 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CopyToMesh.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CopyToMesh.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "CopyToMesh" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CopyToMeshGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CopyToMeshGrad.pbtxt index 0780d5e0dcc..5e3d38dd349 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CopyToMeshGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CopyToMeshGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CopyToMeshGrad" endpoint { name: "CopyToMeshGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cos.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cos.pbtxt index db1f62806e2..a8006cadd6b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cos.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cos.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Cos" endpoint { name: "math.Cos" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cosh.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cosh.pbtxt index a4b5e752bf5..6f08a1b1862 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cosh.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cosh.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Cosh" endpoint { name: "math.Cosh" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CountUpTo.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CountUpTo.pbtxt index eb9f328ce0c..bdc63ba1e04 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CountUpTo.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CountUpTo.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "CountUpTo" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CreateSummaryDbWriter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CreateSummaryDbWriter.pbtxt index 299f881dd44..0c9840034b5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CreateSummaryDbWriter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CreateSummaryDbWriter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CreateSummaryDbWriter" endpoint { name: "summary.CreateSummaryDbWriter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CreateSummaryFileWriter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CreateSummaryFileWriter.pbtxt index 26c7941ce57..b85f13b6de4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CreateSummaryFileWriter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CreateSummaryFileWriter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CreateSummaryFileWriter" endpoint { name: "summary.CreateSummaryFileWriter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResize.pbtxt index cbf9aa8f996..b41932cf5ab 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CropAndResize" endpoint { name: "image.CropAndResize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResizeGradBoxes.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResizeGradBoxes.pbtxt index 44354bdfa03..8b29c975468 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResizeGradBoxes.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResizeGradBoxes.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CropAndResizeGradBoxes" endpoint { name: "image.CropAndResizeGradBoxes" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResizeGradImage.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResizeGradImage.pbtxt index 0618db9a8d7..85607c39878 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResizeGradImage.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CropAndResizeGradImage.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CropAndResizeGradImage" endpoint { name: "image.CropAndResizeGradImage" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cross.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cross.pbtxt index c027884250e..a9717d3bc7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cross.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cross.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Cross" endpoint { name: "linalg.Cross" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CrossReplicaSum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CrossReplicaSum.pbtxt index 5ef9b3b586e..f83642ef04a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CrossReplicaSum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CrossReplicaSum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CrossReplicaSum" endpoint { name: "tpu.CrossReplicaSum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNBackpropV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNBackpropV3.pbtxt index 03875605e8a..eb7800c71df 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNBackpropV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNBackpropV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CudnnRNNBackpropV3" endpoint { name: "nn.CudnnRNNBackprop" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNCanonicalToParamsV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNCanonicalToParamsV2.pbtxt index 1c2cd136bf8..99b144ed11c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNCanonicalToParamsV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNCanonicalToParamsV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CudnnRNNCanonicalToParamsV2" endpoint { name: "nn.CudnnRNNCanonicalToParams" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNParamsSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNParamsSize.pbtxt index 3f1193fe606..e0b34db1680 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNParamsSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNParamsSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CudnnRNNParamsSize" endpoint { name: "nn.CudnnRnnParamsSize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNParamsToCanonicalV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNParamsToCanonicalV2.pbtxt index fa2fd5ffbca..4542b63afcc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNParamsToCanonicalV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNParamsToCanonicalV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CudnnRNNParamsToCanonicalV2" endpoint { name: "nn.CudnnRNNParamsToCanonical" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNV3.pbtxt index 5e2ddba15d1..0e07477c874 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CudnnRNNV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CudnnRNNV3" endpoint { name: "nn.CudnnRNN" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cumprod.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cumprod.pbtxt index 0cb7862413d..b49217a6d13 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cumprod.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cumprod.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Cumprod" endpoint { name: "math.Cumprod" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cumsum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cumsum.pbtxt index e7d90765326..30db71c3b58 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Cumsum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Cumsum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Cumsum" endpoint { name: "math.Cumsum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_CumulativeLogsumexp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_CumulativeLogsumexp.pbtxt index 5d68409a4cb..5e815bd9dab 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_CumulativeLogsumexp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_CumulativeLogsumexp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "CumulativeLogsumexp" endpoint { name: "math.CumulativeLogsumexp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorRestoreV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorRestoreV2.pbtxt index 23d09b99663..c494af28b78 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorRestoreV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorRestoreV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DTensorRestoreV2" endpoint { name: "tpu.DTensorRestore" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorSetGlobalTPUArray.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorSetGlobalTPUArray.pbtxt index 512ac74139d..9eb54c892bb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorSetGlobalTPUArray.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorSetGlobalTPUArray.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DTensorSetGlobalTPUArray" endpoint { name: "tpu.ExecuteTPUEmbeddingPartitioner" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorShardedPrefix.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorShardedPrefix.pbtxt index d3eb775b2d7..28a477a0351 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorShardedPrefix.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DTensorShardedPrefix.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DTensorShardedPrefix" endpoint { name: "tpu.DTensorShardedPrefix" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DataFormatDimMap.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DataFormatDimMap.pbtxt index 36ea17793fd..8d1015ddf8a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DataFormatDimMap.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DataFormatDimMap.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DataFormatDimMap" endpoint { name: "nn.DataFormatDimMap" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DataFormatVecPermute.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DataFormatVecPermute.pbtxt index b6b7e2dc767..61766b93905 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DataFormatVecPermute.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DataFormatVecPermute.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DataFormatVecPermute" endpoint { name: "nn.DataFormatVecPermute" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Dawsn.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Dawsn.pbtxt index 253b06bcd4a..8cd2717a601 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Dawsn.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Dawsn.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Dawsn" endpoint { name: "math.special.Dawsn" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeAndCropJpeg.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeAndCropJpeg.pbtxt index c07bb7a1bdf..13ffab4d225 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeAndCropJpeg.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeAndCropJpeg.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeAndCropJpeg" endpoint { name: "image.DecodeAndCropJpeg" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeBase64.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeBase64.pbtxt index 49c93453f7b..6d091e3a52e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeBase64.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeBase64.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeBase64" endpoint { name: "io.DecodeBase64" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeBmp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeBmp.pbtxt index 049cfa153d1..03f5e2d7aa0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeBmp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeBmp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeBmp" endpoint { name: "image.DecodeBmp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeCSV.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeCSV.pbtxt index 1d60107adab..f8c881d807f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeCSV.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeCSV.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeCSV" endpoint { name: "io.DecodeCsv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeCompressed.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeCompressed.pbtxt index 91327a92ecb..e688002e944 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeCompressed.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeCompressed.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeCompressed" endpoint { name: "io.DecodeCompressed" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeGif.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeGif.pbtxt index 355643ff77c..ac36d9bc1f2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeGif.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeGif.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeGif" endpoint { name: "image.DecodeGif" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeImage.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeImage.pbtxt index 2cc6d31d1d1..80516c0e1b1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeImage.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeImage.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeImage" endpoint { name: "image.DecodeImage" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeJSONExample.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeJSONExample.pbtxt index 6ecba5ab053..d78f8891a22 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeJSONExample.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeJSONExample.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeJSONExample" endpoint { name: "io.DecodeJsonExample" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeJpeg.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeJpeg.pbtxt index c0ebf2e315f..f1d5b1238d9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeJpeg.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeJpeg.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeJpeg" endpoint { name: "image.DecodeJpeg" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodePaddedRaw.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodePaddedRaw.pbtxt index ed6dae2be3c..daaabcd76c4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodePaddedRaw.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodePaddedRaw.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodePaddedRaw" endpoint { name: "io.DecodePaddedRaw" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodePng.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodePng.pbtxt index d94537dc928..aed9c898a29 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodePng.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodePng.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodePng" endpoint { name: "image.DecodePng" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeProtoV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeProtoV2.pbtxt index 4ba118cb0e6..b831161f690 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeProtoV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeProtoV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeProtoV2" endpoint { name: "DecodeProto" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeRaw.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeRaw.pbtxt index 73067173edd..d91490cc854 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeRaw.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeRaw.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeRaw" endpoint { name: "io.DecodeRaw" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeWav.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeWav.pbtxt index 9b249cc6e95..f63a147de11 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeWav.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DecodeWav.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DecodeWav" endpoint { name: "audio.DecodeWav" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeepCopy.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeepCopy.pbtxt index 88a87c92918..e55a4c21ffe 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeepCopy.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeepCopy.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "DeepCopy" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteMemoryCache.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteMemoryCache.pbtxt index c7ab9d3c90a..e9ddbda3ed9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteMemoryCache.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteMemoryCache.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DeleteMemoryCache" endpoint { name: "data.DeleteMemoryCache" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteMultiDeviceIterator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteMultiDeviceIterator.pbtxt index b788df4402e..b93b8c3541e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteMultiDeviceIterator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteMultiDeviceIterator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DeleteMultiDeviceIterator" endpoint { name: "data.DeleteMultiDeviceIterator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteRandomSeedGenerator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteRandomSeedGenerator.pbtxt index fc61e02eba8..f1d06eccdbb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteRandomSeedGenerator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteRandomSeedGenerator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DeleteRandomSeedGenerator" endpoint { name: "random.DeleteRandomSeedGenerator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteSeedGenerator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteSeedGenerator.pbtxt index 48a546f3b83..24e5394bf3f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteSeedGenerator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteSeedGenerator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DeleteSeedGenerator" endpoint { name: "random.DeleteSeedGenerator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteSessionTensor.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteSessionTensor.pbtxt index 1865b461de7..a7e2ca5bfed 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteSessionTensor.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeleteSessionTensor.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "DeleteSessionTensor" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseBincount.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseBincount.pbtxt index aeacb17d512..38af1580e36 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseBincount.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseBincount.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DenseBincount" endpoint { name: "math.DenseBincount" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseCountSparseOutput.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseCountSparseOutput.pbtxt index fca17ecfa79..6496cb1c446 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseCountSparseOutput.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseCountSparseOutput.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DenseCountSparseOutput" endpoint { name: "sparse.DenseCountSparseOutput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToCSRSparseMatrix.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToCSRSparseMatrix.pbtxt index a8186dc4957..dc7ecd1a204 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToCSRSparseMatrix.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToCSRSparseMatrix.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DenseToCSRSparseMatrix" endpoint { name: "linalg.sparse.DenseToCSRSparseMatrix" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToDenseSetOperation.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToDenseSetOperation.pbtxt index f85def92ee3..8772c2c0e3a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToDenseSetOperation.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToDenseSetOperation.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DenseToDenseSetOperation" endpoint { name: "sparse.DenseToDenseSetOperation" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToSparseSetOperation.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToSparseSetOperation.pbtxt index 11fbef8ff1f..80455026338 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToSparseSetOperation.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DenseToSparseSetOperation.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DenseToSparseSetOperation" endpoint { name: "sparse.DenseToSparseSetOperation" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthToSpace.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthToSpace.pbtxt index 0d2cbd2b904..da338027869 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthToSpace.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthToSpace.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DepthToSpace" endpoint { name: "nn.DepthToSpace" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNative.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNative.pbtxt index 1aaa480fefd..eb20bbab725 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNative.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNative.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DepthwiseConv2dNative" endpoint { name: "nn.DepthwiseConv2dNative" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNativeBackpropFilter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNativeBackpropFilter.pbtxt index 1a62d8cf632..e534f662ea2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNativeBackpropFilter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNativeBackpropFilter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DepthwiseConv2dNativeBackpropFilter" endpoint { name: "nn.DepthwiseConv2dNativeBackpropFilter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNativeBackpropInput.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNativeBackpropInput.pbtxt index 9106dd2f8fe..892160034cd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNativeBackpropInput.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DepthwiseConv2dNativeBackpropInput.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DepthwiseConv2dNativeBackpropInput" endpoint { name: "nn.DepthwiseConv2dNativeBackpropInput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Dequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Dequantize.pbtxt index 8ee4daa2f7e..7b32cd14882 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Dequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Dequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Dequantize" endpoint { name: "quantization.Dequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeIterator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeIterator.pbtxt index bdd03f5dc67..cb296d27127 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeIterator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeIterator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DeserializeIterator" endpoint { name: "data.DeserializeIterator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeManySparse.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeManySparse.pbtxt index 826d49f5465..b57141ed844 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeManySparse.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeManySparse.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DeserializeManySparse" endpoint { name: "io.DeserializeManySparse" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeSparse.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeSparse.pbtxt index e6f24bb6257..8b46d1060b8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeSparse.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeserializeSparse.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DeserializeSparse" endpoint { name: "sparse.DeserializeSparse" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DestroyResourceOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DestroyResourceOp.pbtxt index 733e5e5029c..dbdcf2f0cea 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DestroyResourceOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DestroyResourceOp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "DestroyResourceOp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DestroyTemporaryVariable.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DestroyTemporaryVariable.pbtxt index bd416eb68fb..e9f167bd1fc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DestroyTemporaryVariable.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DestroyTemporaryVariable.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "DestroyTemporaryVariable" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeviceIndex.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeviceIndex.pbtxt index 1a656ad1290..de7b5bc2b58 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DeviceIndex.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DeviceIndex.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "DeviceIndex" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Diag.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Diag.pbtxt index 374b3c97e17..de116a55651 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Diag.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Diag.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Diag" endpoint { name: "linalg.TensorDiag" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DiagPart.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DiagPart.pbtxt index 70db2357d06..b9ef4010d99 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DiagPart.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DiagPart.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DiagPart" endpoint { name: "linalg.TensorDiagPart" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Digamma.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Digamma.pbtxt index 68dc74c64ea..fafcf4cc8bc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Digamma.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Digamma.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Digamma" endpoint { name: "math.Digamma" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2D.pbtxt index 914ea29812c..523cf20b08d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Dilation2D" endpoint { name: "nn.Dilation2d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2DBackpropFilter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2DBackpropFilter.pbtxt index db3c68e088e..0b7b84c8b5d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2DBackpropFilter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2DBackpropFilter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Dilation2DBackpropFilter" endpoint { name: "nn.Dilation2dBackpropFilter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2DBackpropInput.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2DBackpropInput.pbtxt index c935144f7af..c8d15a56c8b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2DBackpropInput.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Dilation2DBackpropInput.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Dilation2DBackpropInput" endpoint { name: "nn.Dilation2dBackpropInput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DisableCopyOnRead.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DisableCopyOnRead.pbtxt index aec65bf4c08..4e6dae43607 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DisableCopyOnRead.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DisableCopyOnRead.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DisableCopyOnRead" endpoint { name: "io.DisableCopyOnRead" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DistributedSave.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DistributedSave.pbtxt index 06d244d83bc..74a8b4ddfc1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DistributedSave.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DistributedSave.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DistributedSave" endpoint { name: "train.DistributedSave" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Div.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Div.pbtxt index 2abba7f05f3..70007de3ae0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Div.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Div.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Div" endpoint { name: "math.Div" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DivNoNan.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DivNoNan.pbtxt index c124044604b..c8dcc9f80aa 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DivNoNan.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DivNoNan.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DivNoNan" endpoint { name: "math.DivNoNan" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DrawBoundingBoxesV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DrawBoundingBoxesV2.pbtxt index 76108a4085b..1a1bcc3c284 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DrawBoundingBoxesV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DrawBoundingBoxesV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DrawBoundingBoxesV2" endpoint { name: "image.DrawBoundingBoxes" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DummyIterationCounter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DummyIterationCounter.pbtxt index 12b6ffbf5fa..837647279de 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DummyIterationCounter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DummyIterationCounter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DummyIterationCounter" endpoint { name: "data.DummyIterationCounter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DummyMemoryCache.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DummyMemoryCache.pbtxt index 87fe4495514..ac86013215f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DummyMemoryCache.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DummyMemoryCache.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "DummyMemoryCache" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DummySeedGenerator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DummySeedGenerator.pbtxt index 4d550787d03..3d2cf2618ff 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DummySeedGenerator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DummySeedGenerator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DummySeedGenerator" endpoint { name: "random.DummySeedGenerator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt index 16882e5f1d3..935786de8fa 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DynamicEnqueueTPUEmbeddingArbitraryTensorBatch" endpoint { name: "tpu.DynamicEnqueueTPUEmbeddingArbitraryTensorBatch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicEnqueueTPUEmbeddingRaggedTensorBatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicEnqueueTPUEmbeddingRaggedTensorBatch.pbtxt index 01a1c9189d2..2f59cca069b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicEnqueueTPUEmbeddingRaggedTensorBatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicEnqueueTPUEmbeddingRaggedTensorBatch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "DynamicEnqueueTPUEmbeddingRaggedTensorBatch" endpoint { name: "tpu.DynamicEnqueueTPUEmbeddingRaggedTensorBatch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicPartition.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicPartition.pbtxt index cc585676e48..4550ff6fbbc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicPartition.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicPartition.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "DynamicPartition" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicStitch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicStitch.pbtxt index ac1fef4b6af..609515974a7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicStitch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_DynamicStitch.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "DynamicStitch" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EditDistance.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EditDistance.pbtxt index ca65c2c6e58..8e6dabb659f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EditDistance.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EditDistance.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "EditDistance" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Eig.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Eig.pbtxt index b0aa6f99414..fb0d5c4b045 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Eig.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Eig.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Eig" endpoint { name: "linalg.Eig" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Einsum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Einsum.pbtxt index 07e5db98ae8..fbfc95e1380 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Einsum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Einsum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Einsum" endpoint { name: "linalg.Einsum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Elu.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Elu.pbtxt index bfe8d972cf6..432d2a70692 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Elu.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Elu.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Elu" endpoint { name: "nn.Elu" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EluGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EluGrad.pbtxt index 3757357c000..e8722cc7d24 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EluGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EluGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EluGrad" endpoint { name: "nn.EluGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Empty.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Empty.pbtxt index 6522f51d9dc..e2dfb53ab7b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Empty.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Empty.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Empty" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EmptyTensorList.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EmptyTensorList.pbtxt index ef3f533964c..df92f263af1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EmptyTensorList.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EmptyTensorList.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "EmptyTensorList" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EmptyTensorMap.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EmptyTensorMap.pbtxt index cf5cf52c931..a2141d3fbd3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EmptyTensorMap.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EmptyTensorMap.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "EmptyTensorMap" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeBase64.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeBase64.pbtxt index 66f19def9ae..a060a92104d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeBase64.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeBase64.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EncodeBase64" endpoint { name: "io.EncodeBase64" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeJpeg.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeJpeg.pbtxt index 1e151665f87..af995121608 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeJpeg.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeJpeg.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EncodeJpeg" endpoint { name: "image.EncodeJpeg" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeJpegVariableQuality.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeJpegVariableQuality.pbtxt index f18b353938c..bb8eeba21b3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeJpegVariableQuality.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeJpegVariableQuality.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EncodeJpegVariableQuality" endpoint { name: "image.EncodeJpegVariableQuality" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodePng.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodePng.pbtxt index 7a8d713c865..b806e4917ff 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodePng.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodePng.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EncodePng" endpoint { name: "image.EncodePng" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeProto.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeProto.pbtxt index ac6a04b4bc2..87b2c6ac4bc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeProto.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeProto.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "EncodeProto" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeWav.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeWav.pbtxt index f3b22fde666..96ed73270da 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeWav.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EncodeWav.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EncodeWav" endpoint { name: "audio.EncodeWav" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt index 90dc3c898d0..7335cf4e1cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EnqueueTPUEmbeddingArbitraryTensorBatch" endpoint { name: "tpu.EnqueueTPUEmbeddingArbitraryTensorBatch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingBatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingBatch.pbtxt index 6b698b2800b..a14d72b4a72 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingBatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingBatch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EnqueueTPUEmbeddingBatch" endpoint { name: "tpu.EnqueueTPUEmbeddingBatch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingIntegerBatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingIntegerBatch.pbtxt index f4a6a104d5a..97b471f0ddd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingIntegerBatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingIntegerBatch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EnqueueTPUEmbeddingIntegerBatch" endpoint { name: "tpu.EnqueueTPUEmbeddingIntegerBatch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt index 971fed264c9..d1d250dd27a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EnqueueTPUEmbeddingRaggedTensorBatch" endpoint { name: "tpu.EnqueueTPUEmbeddingRaggedTensorBatch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingSparseBatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingSparseBatch.pbtxt index 16366328074..b346dd636a9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingSparseBatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingSparseBatch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EnqueueTPUEmbeddingSparseBatch" endpoint { name: "tpu.EnqueueTPUEmbeddingSparseBatch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt index 2f8dd050421..56864f899be 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EnqueueTPUEmbeddingSparseTensorBatch" endpoint { name: "tpu.EnqueueTPUEmbeddingSparseTensorBatch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnsureShape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnsureShape.pbtxt index 6238947598f..4e7d8ac0a55 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EnsureShape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EnsureShape.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "EnsureShape" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Enter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Enter.pbtxt index ffc10c91beb..07abdf23784 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Enter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Enter.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Enter" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Equal.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Equal.pbtxt index c2256c24337..afd1c9fcf85 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Equal.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Equal.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Equal" endpoint { name: "math.Equal" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Erf.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Erf.pbtxt index 9efcc3983c4..0f3d2e6dc03 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Erf.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Erf.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Erf" endpoint { name: "math.Erf" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Erfc.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Erfc.pbtxt index c0f4db61ff4..b1da0c02862 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Erfc.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Erfc.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Erfc" endpoint { name: "math.Erfc" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Erfinv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Erfinv.pbtxt index 7937108c423..68358ebb137 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Erfinv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Erfinv.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Erfinv" endpoint { name: "math.erfinv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_EuclideanNorm.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_EuclideanNorm.pbtxt index f6a75752dfb..f4afae29cd7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_EuclideanNorm.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_EuclideanNorm.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "EuclideanNorm" endpoint { name: "linalg.EuclideanNorm" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExecuteTPUEmbeddingPartitioner.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExecuteTPUEmbeddingPartitioner.pbtxt index 465325a17a0..125aeb61d93 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExecuteTPUEmbeddingPartitioner.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExecuteTPUEmbeddingPartitioner.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExecuteTPUEmbeddingPartitioner" endpoint { name: "tpu.ExecuteTPUEmbeddingPartitioner" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Exit.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Exit.pbtxt index 6215cd22299..0ca26a5aa7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Exit.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Exit.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Exit" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Exp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Exp.pbtxt index b2790c8306f..7947019a666 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Exp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Exp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Exp" endpoint { name: "math.Exp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExpandDims.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExpandDims.pbtxt index 66902ccb5b0..01c82186792 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExpandDims.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExpandDims.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ExpandDims" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalAssertNextDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalAssertNextDataset.pbtxt index efdfebb08b8..28e46dce87d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalAssertNextDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalAssertNextDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalAssertNextDataset" endpoint { name: "data.experimental.AssertNextDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalAutoShardDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalAutoShardDataset.pbtxt index 1784a6b9a07..df08aef09b3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalAutoShardDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalAutoShardDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalAutoShardDataset" endpoint { name: "data.experimental.AutoShardDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalBytesProducedStatsDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalBytesProducedStatsDataset.pbtxt index 8e589b0d674..272f9e1eaae 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalBytesProducedStatsDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalBytesProducedStatsDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalBytesProducedStatsDataset" endpoint { name: "data.experimental.BytesProducedStatsDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalCSVDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalCSVDataset.pbtxt index 217088d589c..d548e72a07a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalCSVDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalCSVDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalCSVDataset" endpoint { name: "data.experimental.CSVDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalChooseFastestDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalChooseFastestDataset.pbtxt index ea9dbd5ca43..d818de9d33e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalChooseFastestDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalChooseFastestDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalChooseFastestDataset" endpoint { name: "data.experimental.ChooseFastestDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDatasetCardinality.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDatasetCardinality.pbtxt index cb6c9048eeb..743bc536a0f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDatasetCardinality.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDatasetCardinality.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalDatasetCardinality" endpoint { name: "data.experimental.DatasetCardinality" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDatasetToTFRecord.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDatasetToTFRecord.pbtxt index 837258d5733..45ca0a09034 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDatasetToTFRecord.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDatasetToTFRecord.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalDatasetToTFRecord" endpoint { name: "data.experimental.DatasetToTFRecord" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDenseToSparseBatchDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDenseToSparseBatchDataset.pbtxt index c45c43efe6f..492eeee03a2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDenseToSparseBatchDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDenseToSparseBatchDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalDenseToSparseBatchDataset" endpoint { name: "data.experimental.DenseToSparseBatchDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDirectedInterleaveDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDirectedInterleaveDataset.pbtxt index 9d8795e599b..d0acd7ea288 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDirectedInterleaveDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalDirectedInterleaveDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalDirectedInterleaveDataset" endpoint { name: "data.experimental.DirectedInterleaveDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResource.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResource.pbtxt index fef2a0fd2f2..f35eca43ca4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResource.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResource.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalFunctionBufferingResource" endpoint { name: "data.experimental.FunctionBufferingResource" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResourceGetNext.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResourceGetNext.pbtxt index 4c614345d59..e1b5ae6fac6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResourceGetNext.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResourceGetNext.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalFunctionBufferingResourceGetNext" endpoint { name: "data.experimental.FunctionBufferingResourceGetNext" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResourceReset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResourceReset.pbtxt index b819eeab663..8c9bdb4de26 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResourceReset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalFunctionBufferingResourceReset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalFunctionBufferingResourceReset" endpoint { name: "data.experimental.FunctionBufferingResourceReset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalGroupByReducerDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalGroupByReducerDataset.pbtxt index eeda8ab8552..8cf62d85942 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalGroupByReducerDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalGroupByReducerDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalGroupByReducerDataset" endpoint { name: "data.experimental.GroupByReducerDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalGroupByWindowDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalGroupByWindowDataset.pbtxt index 9d1210c4a60..875aaa78dd8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalGroupByWindowDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalGroupByWindowDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalGroupByWindowDataset" endpoint { name: "data.experimental.GroupByWindowDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalIgnoreErrorsDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalIgnoreErrorsDataset.pbtxt index 3ece1655397..ad31e9738d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalIgnoreErrorsDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalIgnoreErrorsDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalIgnoreErrorsDataset" endpoint { name: "data.experimental.IgnoreErrorsDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalIteratorGetDevice.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalIteratorGetDevice.pbtxt index 2f46541bf3d..b1f2dfcf5c9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalIteratorGetDevice.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalIteratorGetDevice.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalIteratorGetDevice" endpoint { name: "data.experimental.IteratorGetDevice" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalLMDBDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalLMDBDataset.pbtxt index 32345d2200e..a427a85e631 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalLMDBDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalLMDBDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalLMDBDataset" endpoint { name: "data.experimental.LmdbDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalLatencyStatsDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalLatencyStatsDataset.pbtxt index f02a5653e4f..21ed0bfde64 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalLatencyStatsDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalLatencyStatsDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalLatencyStatsDataset" endpoint { name: "data.experimental.LatencyStatsDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMapAndBatchDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMapAndBatchDataset.pbtxt index 762c1b49602..fa88e18887b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMapAndBatchDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMapAndBatchDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalMapAndBatchDataset" endpoint { name: "data.experimental.MapAndBatchDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMapDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMapDataset.pbtxt index 9406cffef9a..cdfa66a022e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMapDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMapDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalMapDataset" endpoint { name: "data.experimental.MapDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMatchingFilesDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMatchingFilesDataset.pbtxt index bcc19749d12..ae0210b3f3e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMatchingFilesDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMatchingFilesDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalMatchingFilesDataset" endpoint { name: "data.experimental.MatchingFilesDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMaxIntraOpParallelismDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMaxIntraOpParallelismDataset.pbtxt index 5336b380310..afe63cd09fe 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMaxIntraOpParallelismDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalMaxIntraOpParallelismDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalMaxIntraOpParallelismDataset" endpoint { name: "data.experimental.MaxIntraOpParallelismDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalNonSerializableDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalNonSerializableDataset.pbtxt index 61fe8ae0ce9..5e3386e8483 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalNonSerializableDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalNonSerializableDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalNonSerializableDataset" endpoint { name: "data.experimental.NonSerializableDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalParallelInterleaveDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalParallelInterleaveDataset.pbtxt index def35c75c91..ad33fe82aa8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalParallelInterleaveDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalParallelInterleaveDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalParallelInterleaveDataset" endpoint { name: "data.experimental.ParallelInterleaveDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalParseExampleDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalParseExampleDataset.pbtxt index 8a1a5286033..741b6b1a96a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalParseExampleDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalParseExampleDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalParseExampleDataset" endpoint { name: "data.experimental.ParseExampleDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalPrivateThreadPoolDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalPrivateThreadPoolDataset.pbtxt index 0f035999282..667d9f53047 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalPrivateThreadPoolDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalPrivateThreadPoolDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalPrivateThreadPoolDataset" endpoint { name: "data.experimental.PrivateThreadPoolDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalRandomDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalRandomDataset.pbtxt index a109386b43c..687e7c2782a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalRandomDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalRandomDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalRandomDataset" endpoint { name: "data.experimental.RandomDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalRebatchDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalRebatchDataset.pbtxt index 2c6b6ccbfe6..8012dbae620 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalRebatchDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalRebatchDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalRebatchDataset" endpoint { name: "data.experimental.RebatchDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalScanDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalScanDataset.pbtxt index bea58e5b1b5..910fb561988 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalScanDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalScanDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalScanDataset" endpoint { name: "data.experimental.ScanDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSetStatsAggregatorDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSetStatsAggregatorDataset.pbtxt index bd3129ae938..d3039499942 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSetStatsAggregatorDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSetStatsAggregatorDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalSetStatsAggregatorDataset" endpoint { name: "data.experimental.SetStatsAggregatorDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSleepDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSleepDataset.pbtxt index 4ab827669ba..7c160528fcc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSleepDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSleepDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalSleepDataset" endpoint { name: "data.experimental.SleepDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSlidingWindowDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSlidingWindowDataset.pbtxt index 045a2bc8a6a..aa0fe454722 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSlidingWindowDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSlidingWindowDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalSlidingWindowDataset" endpoint { name: "data.experimental.SlidingWindowDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSqlDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSqlDataset.pbtxt index b0937039065..f827b21b8b7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSqlDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalSqlDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalSqlDataset" endpoint { name: "data.experimental.SqlDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalStatsAggregatorHandle.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalStatsAggregatorHandle.pbtxt index 1910ed3140c..ec2f2aff7ca 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalStatsAggregatorHandle.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalStatsAggregatorHandle.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalStatsAggregatorHandle" endpoint { name: "data.experimental.StatsAggregatorHandle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalStatsAggregatorSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalStatsAggregatorSummary.pbtxt index 9773131e228..6f9b79ac777 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalStatsAggregatorSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalStatsAggregatorSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalStatsAggregatorSummary" endpoint { name: "data.experimental.StatsAggregatorSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalTakeWhileDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalTakeWhileDataset.pbtxt index 207875b0588..2b494bd04c4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalTakeWhileDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalTakeWhileDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalTakeWhileDataset" endpoint { name: "data.experimental.TakeWhileDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalThreadPoolDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalThreadPoolDataset.pbtxt index e9825ac34c7..55fc4665fd9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalThreadPoolDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalThreadPoolDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalThreadPoolDataset" endpoint { name: "data.experimental.ThreadPoolDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalThreadPoolHandle.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalThreadPoolHandle.pbtxt index 55d81428408..ecaa0ceb2c9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalThreadPoolHandle.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalThreadPoolHandle.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalThreadPoolHandle" endpoint { name: "data.experimental.ThreadPoolHandle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalUnbatchDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalUnbatchDataset.pbtxt index 137e99a60ad..c08a60749be 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalUnbatchDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalUnbatchDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalUnbatchDataset" endpoint { name: "data.experimental.UnbatchDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalUniqueDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalUniqueDataset.pbtxt index 108ad10a98f..d644078b402 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalUniqueDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExperimentalUniqueDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExperimentalUniqueDataset" endpoint { name: "data.experimental.UniqueDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Expint.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Expint.pbtxt index b2cd45a605e..64b09ca6ab7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Expint.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Expint.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Expint" endpoint { name: "math.special.Expint" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Expm1.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Expm1.pbtxt index 71a8fcf0225..df2ece3b9e8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Expm1.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Expm1.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Expm1" endpoint { name: "math.Expm1" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractGlimpseV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractGlimpseV2.pbtxt index 8fd84593339..e0491472fc4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractGlimpseV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractGlimpseV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExtractGlimpseV2" endpoint { name: "image.ExtractGlimpse" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractImagePatches.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractImagePatches.pbtxt index 7cdcfd75285..ab6177b5247 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractImagePatches.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractImagePatches.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExtractImagePatches" endpoint { name: "image.ExtractImagePatches" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractJpegShape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractJpegShape.pbtxt index c95fcc9cef4..da8258cc5b2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractJpegShape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractJpegShape.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ExtractJpegShape" endpoint { name: "image.ExtractJpegShape" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractVolumePatches.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractVolumePatches.pbtxt index 6f61c832109..5d5d80ce08f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractVolumePatches.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ExtractVolumePatches.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ExtractVolumePatches" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT.pbtxt index d9655903086..a50549a383e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FFT" endpoint { name: "signal.Fft" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT2D.pbtxt index 474103076b9..ffbf0a00050 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FFT2D" endpoint { name: "signal.Fft2d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT3D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT3D.pbtxt index 8e1606b8f9d..a7415cc5d03 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT3D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FFT3D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FFT3D" endpoint { name: "signal.Fft3d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FFTND.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FFTND.pbtxt index 6c40faf3436..753cdcb1997 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FFTND.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FFTND.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FFTND" endpoint { name: "signal.FftNd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FIFOQueueV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FIFOQueueV2.pbtxt index c0861a6e8dd..797fe75a0b5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FIFOQueueV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FIFOQueueV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FIFOQueueV2" endpoint { name: "io.FifoQueue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Fact.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Fact.pbtxt index 436664e554b..f60455d31a5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Fact.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Fact.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Fact" endpoint { name: "math.Fact" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeParam.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeParam.pbtxt index ac8f751442c..3310fb8af02 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeParam.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeParam.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "FakeParam" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxArgs.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxArgs.pbtxt index 809d231a55c..61723a6b616 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxArgs.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxArgs.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FakeQuantWithMinMaxArgs" endpoint { name: "quantization.FakeQuantWithMinMaxArgs" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxArgsGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxArgsGradient.pbtxt index 50d0f51a140..a995fff37e4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxArgsGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxArgsGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FakeQuantWithMinMaxArgsGradient" endpoint { name: "quantization.FakeQuantWithMinMaxArgsGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVars.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVars.pbtxt index b86258aab2b..7318899ee3f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVars.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVars.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FakeQuantWithMinMaxVars" endpoint { name: "quantization.FakeQuantWithMinMaxVars" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsGradient.pbtxt index 3c1343423c1..7738b510c20 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FakeQuantWithMinMaxVarsGradient" endpoint { name: "quantization.FakeQuantWithMinMaxVarsGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsPerChannel.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsPerChannel.pbtxt index afe45a290d3..270c2644610 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsPerChannel.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsPerChannel.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FakeQuantWithMinMaxVarsPerChannel" endpoint { name: "quantization.FakeQuantWithMinMaxVarsPerChannel" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsPerChannelGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsPerChannelGradient.pbtxt index 9dd62fdffd0..0cd372b0162 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsPerChannelGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQuantWithMinMaxVarsPerChannelGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FakeQuantWithMinMaxVarsPerChannelGradient" endpoint { name: "quantization.FakeQuantWithMinMaxVarsPerChannelGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQueue.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQueue.pbtxt index 8960966f084..b49719e8142 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQueue.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FakeQueue.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FakeQueue" endpoint { name: "io.FakeQueue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FileSystemSetConfiguration.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FileSystemSetConfiguration.pbtxt index a4699e1a487..48a9f01c087 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FileSystemSetConfiguration.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FileSystemSetConfiguration.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "FileSystemSetConfiguration" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Fill.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Fill.pbtxt index 3997328ed31..b0883b54e03 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Fill.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Fill.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Fill" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FinalizeTPUEmbedding.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FinalizeTPUEmbedding.pbtxt index 991b3c66b12..bb76cd12973 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FinalizeTPUEmbedding.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FinalizeTPUEmbedding.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FinalizeTPUEmbedding" endpoint { name: "tpu.FinalizeTPUEmbedding" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Fingerprint.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Fingerprint.pbtxt index 3a8379e516c..42f780314bb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Fingerprint.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Fingerprint.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Fingerprint" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FixedLengthRecordReaderV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FixedLengthRecordReaderV2.pbtxt index f897c21365b..c6acb018dc2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FixedLengthRecordReaderV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FixedLengthRecordReaderV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FixedLengthRecordReaderV2" endpoint { name: "io.FixedLengthRecordReader" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FixedUnigramCandidateSampler.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FixedUnigramCandidateSampler.pbtxt index eb9c68d4dba..b4e26238201 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FixedUnigramCandidateSampler.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FixedUnigramCandidateSampler.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FixedUnigramCandidateSampler" endpoint { name: "nn.FixedUnigramCandidateSampler" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Floor.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Floor.pbtxt index a2b80f97e05..9cbf0eb0e4e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Floor.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Floor.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Floor" endpoint { name: "math.Floor" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FloorDiv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FloorDiv.pbtxt index 054d85f55c9..693eed27e08 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FloorDiv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FloorDiv.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FloorDiv" endpoint { name: "math.FloorDiv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FloorMod.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FloorMod.pbtxt index ff2216a9357..c6c7ea42659 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FloorMod.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FloorMod.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FloorMod" endpoint { name: "math.FloorMod" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FlushSummaryWriter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FlushSummaryWriter.pbtxt index feaa3a6dc22..5731ce679d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FlushSummaryWriter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FlushSummaryWriter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FlushSummaryWriter" endpoint { name: "summary.FlushSummaryWriter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_For.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_For.pbtxt index 30363d1e963..4d01b94bd26 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_For.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_For.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "For" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalAvgPool.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalAvgPool.pbtxt index fc2e6ca54b8..1e2afb0ca3b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalAvgPool.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalAvgPool.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FractionalAvgPool" endpoint { name: "nn.FractionalAvgPool" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalAvgPoolGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalAvgPoolGrad.pbtxt index 4e11d5e3950..f51859f903e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalAvgPoolGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalAvgPoolGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FractionalAvgPoolGrad" endpoint { name: "nn.FractionalAvgPoolGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalMaxPool.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalMaxPool.pbtxt index 061b358ec27..ad0fddc2bc6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalMaxPool.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalMaxPool.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FractionalMaxPool" endpoint { name: "nn.FractionalMaxPool" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalMaxPoolGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalMaxPoolGrad.pbtxt index c70e6d721e5..00bf30c2b68 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalMaxPoolGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FractionalMaxPoolGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FractionalMaxPoolGrad" endpoint { name: "nn.FractionalMaxPoolGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FresnelCos.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FresnelCos.pbtxt index 88373aebe7c..239ea452c59 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FresnelCos.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FresnelCos.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FresnelCos" endpoint { name: "math.special.FresnelCos" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FresnelSin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FresnelSin.pbtxt index 294d07b18ce..01e64aa2368 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FresnelSin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FresnelSin.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FresnelSin" endpoint { name: "math.special.FresnelSin" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedBatchNormGradV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedBatchNormGradV3.pbtxt index ed5f0ebb827..bf2ae00fd7f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedBatchNormGradV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedBatchNormGradV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FusedBatchNormGradV3" endpoint { name: "nn.FusedBatchNormGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedBatchNormV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedBatchNormV3.pbtxt index 1fe73e286a0..e3cc882ca7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedBatchNormV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedBatchNormV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FusedBatchNormV3" endpoint { name: "nn.FusedBatchNorm" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedPadConv2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedPadConv2D.pbtxt index e7ee10e0c58..7e0d6eb913d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedPadConv2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedPadConv2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FusedPadConv2D" endpoint { name: "nn.FusedPadConv2d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedResizeAndPadConv2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedResizeAndPadConv2D.pbtxt index 6948fc1b87d..fc92f057104 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedResizeAndPadConv2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_FusedResizeAndPadConv2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "FusedResizeAndPadConv2D" endpoint { name: "nn.FusedResizeAndPadConv2d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GRUBlockCell.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GRUBlockCell.pbtxt index 1dffbe38475..0b5ab9c8b0b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GRUBlockCell.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GRUBlockCell.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "GRUBlockCell" endpoint { name: "nn.GRUBlockCell" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GRUBlockCellGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GRUBlockCellGrad.pbtxt index 33e7f4077f0..642a35b7945 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GRUBlockCellGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GRUBlockCellGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "GRUBlockCellGrad" endpoint { name: "nn.GRUBlockCellGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GatherNd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GatherNd.pbtxt index 257c0316ea0..80ed9a514c5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GatherNd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GatherNd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "GatherNd" endpoint { name: "GatherNd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GatherV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GatherV2.pbtxt index 0927e77a968..d27fd30efa0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GatherV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GatherV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "GatherV2" endpoint { name: "Gather" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GcsConfigureBlockCache.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GcsConfigureBlockCache.pbtxt index 1ba3044d4c3..0878563c93b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GcsConfigureBlockCache.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GcsConfigureBlockCache.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "GcsConfigureBlockCache" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GcsConfigureCredentials.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GcsConfigureCredentials.pbtxt index 98bd555fb87..1653b16c4ee 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GcsConfigureCredentials.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GcsConfigureCredentials.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "GcsConfigureCredentials" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateBigQueryReaderPartitions.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateBigQueryReaderPartitions.pbtxt index 956f40762d7..3b037ef31c6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateBigQueryReaderPartitions.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateBigQueryReaderPartitions.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "GenerateBigQueryReaderPartitions" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateBoundingBoxProposals.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateBoundingBoxProposals.pbtxt index 2a830eb572e..069e9b74fff 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateBoundingBoxProposals.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateBoundingBoxProposals.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "GenerateBoundingBoxProposals" endpoint { name: "image.GenerateBoundingBoxProposals" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateVocabRemapping.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateVocabRemapping.pbtxt index 9aac3b17f3c..02c132223ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateVocabRemapping.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GenerateVocabRemapping.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "GenerateVocabRemapping" endpoint { name: "train.GenerateVocabRemapping" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GetElementAtIndex.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GetElementAtIndex.pbtxt index 041f46e450e..9fac3335954 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GetElementAtIndex.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GetElementAtIndex.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "GetElementAtIndex" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GetOptions.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GetOptions.pbtxt index 382e395959f..eeb6d4c91d8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GetOptions.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GetOptions.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "GetOptions" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GetSessionHandleV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GetSessionHandleV2.pbtxt index ba89942d77f..3484fbcd5d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GetSessionHandleV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GetSessionHandleV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "GetSessionHandleV2" endpoint { name: "GetSessionHandle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GetSessionTensor.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GetSessionTensor.pbtxt index 34b6e627cdc..496b31c6ef0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GetSessionTensor.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GetSessionTensor.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "GetSessionTensor" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Greater.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Greater.pbtxt index 594f9276be1..a84b4c9bc6b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Greater.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Greater.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Greater" endpoint { name: "math.Greater" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GreaterEqual.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GreaterEqual.pbtxt index 17ea8696b0d..57f8c014728 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GreaterEqual.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GreaterEqual.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "GreaterEqual" endpoint { name: "math.GreaterEqual" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_GuaranteeConst.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_GuaranteeConst.pbtxt index 8cac25787da..56a115a0603 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_GuaranteeConst.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_GuaranteeConst.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "GuaranteeConst" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_HSVToRGB.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_HSVToRGB.pbtxt index 95b042d5d68..5689d054353 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_HSVToRGB.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_HSVToRGB.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "HSVToRGB" endpoint { name: "image.HsvToRgb" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_HashTableV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_HashTableV2.pbtxt index 38cc5818d3b..4cde617757a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_HashTableV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_HashTableV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "HashTableV2" endpoint { name: "HashTable" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_HistogramFixedWidth.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_HistogramFixedWidth.pbtxt index f64d9ae1d23..f3d2065032f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_HistogramFixedWidth.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_HistogramFixedWidth.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "HistogramFixedWidth" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_HistogramSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_HistogramSummary.pbtxt index 97f28335bb9..6c2c3b8254a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_HistogramSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_HistogramSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "HistogramSummary" endpoint { name: "summary.HistogramSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_HostConst.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_HostConst.pbtxt index ba589e73e71..f2a7160eccd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_HostConst.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_HostConst.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "HostConst" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT.pbtxt index 4a15ebec7f6..a84e2d6dd57 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IFFT" endpoint { name: "signal.Ifft" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT2D.pbtxt index 35d696ee739..3380f459463 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IFFT2D" endpoint { name: "signal.Ifft2d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT3D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT3D.pbtxt index 76a3164e6aa..02db3a66379 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT3D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFT3D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IFFT3D" endpoint { name: "signal.Ifft3d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFTND.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFTND.pbtxt index dcee8641384..214a8bfc0b8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFTND.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IFFTND.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IFFTND" endpoint { name: "signal.IfftNd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT.pbtxt index 7a68b01524e..ebd31423283 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IRFFT" endpoint { name: "signal.Irfft" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT2D.pbtxt index 239ec445d02..e73397a832f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IRFFT2D" endpoint { name: "signal.Irfft2d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT3D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT3D.pbtxt index 87969436b7f..e6a064cfa6c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT3D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFT3D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IRFFT3D" endpoint { name: "signal.Irfft3d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFTND.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFTND.pbtxt index bdab5f77932..848e444c33e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFTND.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IRFFTND.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IRFFTND" endpoint { name: "signal.IrfftNd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Identity.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Identity.pbtxt index b6df3c6cfec..f90a3e1b0f9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Identity.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Identity.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Identity" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IdentityN.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IdentityN.pbtxt index 827df10c65a..a39e00d4106 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IdentityN.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IdentityN.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "IdentityN" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IdentityReaderV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IdentityReaderV2.pbtxt index 8081ac26b3d..92bc4a10279 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IdentityReaderV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IdentityReaderV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IdentityReaderV2" endpoint { name: "io.IdentityReader" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_If.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_If.pbtxt index a3bc33ac2ce..292c093587e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_If.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_If.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "If" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Igamma.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Igamma.pbtxt index cbdd8b984c4..e0134f5acc4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Igamma.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Igamma.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Igamma" endpoint { name: "math.Igamma" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IgammaGradA.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IgammaGradA.pbtxt index 0659c80c39f..46eaba97345 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IgammaGradA.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IgammaGradA.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IgammaGradA" endpoint { name: "math.IgammaGradA" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Igammac.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Igammac.pbtxt index 94f6085e1a9..3114d90fd61 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Igammac.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Igammac.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Igammac" endpoint { name: "math.Igammac" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Imag.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Imag.pbtxt index 4227c7078fc..66427ed58bd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Imag.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Imag.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Imag" endpoint { name: "math.Imag" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageProjectiveTransformV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageProjectiveTransformV2.pbtxt index badcf0d63ca..ae6bb8507be 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageProjectiveTransformV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageProjectiveTransformV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ImageProjectiveTransformV2" endpoint { name: "image.ImageProjectiveTransformV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageProjectiveTransformV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageProjectiveTransformV3.pbtxt index 765ffe8107a..2f477c6d695 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageProjectiveTransformV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageProjectiveTransformV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ImageProjectiveTransformV3" endpoint { name: "image.ImageProjectiveTransformV3" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageSummary.pbtxt index 1871e6b6550..5c3bd5f5047 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ImageSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ImageSummary" endpoint { name: "summary.ImageSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ImmutableConst.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ImmutableConst.pbtxt index fd0384dc452..6f7a34c5a69 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ImmutableConst.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ImmutableConst.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ImmutableConst" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ImportEvent.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ImportEvent.pbtxt index c2d8d2eba35..630a8894724 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ImportEvent.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ImportEvent.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ImportEvent" endpoint { name: "summary.ImportEvent" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InTopKV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InTopKV2.pbtxt index 400ee714624..0fc46096895 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InTopKV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InTopKV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "InTopKV2" endpoint { name: "nn.InTopK" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedDequeue.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedDequeue.pbtxt index 3cc8a15b7c5..0d57c36ce27 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedDequeue.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedDequeue.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "InfeedDequeue" endpoint { name: "tpu.InfeedDequeue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedDequeueTuple.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedDequeueTuple.pbtxt index 6e1f824518f..3655572e592 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedDequeueTuple.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedDequeueTuple.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "InfeedDequeueTuple" endpoint { name: "tpu.InfeedDequeueTuple" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueue.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueue.pbtxt index 2e5f2cf8171..889c70cbfb1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueue.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueue.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "InfeedEnqueue" endpoint { name: "tpu.InfeedEnqueue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueuePrelinearizedBuffer.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueuePrelinearizedBuffer.pbtxt index 2d78a4c3f19..e37e5ed26cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueuePrelinearizedBuffer.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueuePrelinearizedBuffer.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "InfeedEnqueuePrelinearizedBuffer" endpoint { name: "tpu.InfeedEnqueuePrelinearizedBuffer" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueueTuple.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueueTuple.pbtxt index dc50c834e39..99c1cc4f0a1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueueTuple.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InfeedEnqueueTuple.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "InfeedEnqueueTuple" endpoint { name: "tpu.InfeedEnqueueTuple" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InitializeTableFromTextFileV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InitializeTableFromTextFileV2.pbtxt index 9a4f7022018..34712e89316 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InitializeTableFromTextFileV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InitializeTableFromTextFileV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "InitializeTableFromTextFileV2" endpoint { name: "InitializeTableFromTextFile" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InitializeTableV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InitializeTableV2.pbtxt index d7a9a813d07..efb93c75341 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InitializeTableV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InitializeTableV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "InitializeTableV2" endpoint { name: "InitializeTable" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceAdd.pbtxt index 3d157ab7f83..c5b62051abe 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceAdd.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "InplaceAdd" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceSub.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceSub.pbtxt index b2ed5496430..2b6359ab957 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceSub.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceSub.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "InplaceSub" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceUpdate.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceUpdate.pbtxt index 91041b43abd..8d3a0f9d699 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceUpdate.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InplaceUpdate.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "InplaceUpdate" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Inv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Inv.pbtxt index 49f3e6c0429..543e2c9bbe8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Inv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Inv.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Inv" endpoint { name: "linalg.Inv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InvGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InvGrad.pbtxt index d3bfa78e99c..560855ffcf2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InvGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InvGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "InvGrad" endpoint { name: "nn.InvGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Invert.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Invert.pbtxt index 9898bfa003d..6119fb19629 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Invert.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Invert.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Invert" endpoint { name: "bitwise.Invert" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_InvertPermutation.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_InvertPermutation.pbtxt index 9ee103f554e..3fa442de299 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_InvertPermutation.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_InvertPermutation.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "InvertPermutation" endpoint { name: "math.InvertPermutation" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsBoostedTreesEnsembleInitialized.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsBoostedTreesEnsembleInitialized.pbtxt index 5c4bdd5ddc4..9efd7cd8357 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsBoostedTreesEnsembleInitialized.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsBoostedTreesEnsembleInitialized.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "IsBoostedTreesEnsembleInitialized" endpoint { name: "estimator.IsBoostedTreesEnsembleInitialized" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsBoostedTreesQuantileStreamResourceInitialized.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsBoostedTreesQuantileStreamResourceInitialized.pbtxt index 8676e052712..630406cc2f0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsBoostedTreesQuantileStreamResourceInitialized.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsBoostedTreesQuantileStreamResourceInitialized.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "IsBoostedTreesQuantileStreamResourceInitialized" endpoint { name: "estimator.IsBoostedTreesQuantileStreamResourceInitialized" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsFinite.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsFinite.pbtxt index fce58903991..1c33eae3169 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsFinite.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsFinite.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IsFinite" endpoint { name: "math.IsFinite" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsInf.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsInf.pbtxt index 823c1d72812..dbe157edb81 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsInf.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsInf.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IsInf" endpoint { name: "math.IsInf" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsNan.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsNan.pbtxt index 58805bf99f6..a5575098082 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsNan.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsNan.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IsNan" endpoint { name: "math.IsNan" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsTPUEmbeddingInitialized.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsTPUEmbeddingInitialized.pbtxt index e0e66156b85..8f99e9b3e71 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsTPUEmbeddingInitialized.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsTPUEmbeddingInitialized.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IsTPUEmbeddingInitialized" endpoint { name: "tpu.IsTPUEmbeddingInitialized" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsVariableInitialized.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsVariableInitialized.pbtxt index 7bf51da2da9..b5f0f182125 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsVariableInitialized.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsVariableInitialized.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "IsVariableInitialized" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsotonicRegression.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsotonicRegression.pbtxt index ec88abda69e..e0d1edb67aa 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IsotonicRegression.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IsotonicRegression.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IsotonicRegression" endpoint { name: "nn.IsotonicRegression" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorFromStringHandleV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorFromStringHandleV2.pbtxt index 86745a3a564..214318f4aea 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorFromStringHandleV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorFromStringHandleV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IteratorFromStringHandleV2" endpoint { name: "data.IteratorFromStringHandle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetDevice.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetDevice.pbtxt index 8f829ce2787..9da26e5af9f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetDevice.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetDevice.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IteratorGetDevice" endpoint { name: "data.IteratorGetDevice" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetNextAsOptional.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetNextAsOptional.pbtxt index ae02a0e0171..95ea8dc4224 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetNextAsOptional.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetNextAsOptional.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IteratorGetNextAsOptional" endpoint { name: "data.IteratorGetNextAsOptional" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetNextSync.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetNextSync.pbtxt index 4aa7c07a774..5f74f24e12f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetNextSync.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorGetNextSync.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IteratorGetNextSync" endpoint { name: "data.IteratorGetNextSync" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorToStringHandle.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorToStringHandle.pbtxt index 7413ec846e7..0a7723ac676 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorToStringHandle.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_IteratorToStringHandle.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "IteratorToStringHandle" endpoint { name: "data.IteratorToStringHandle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_KMC2ChainInitialization.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_KMC2ChainInitialization.pbtxt index 5bf90f86d8c..5c2ed95566d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_KMC2ChainInitialization.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_KMC2ChainInitialization.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "KMC2ChainInitialization" endpoint { name: "cluster.KMC2ChainInitialization" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_KmeansPlusPlusInitialization.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_KmeansPlusPlusInitialization.pbtxt index f3f71d56162..b4cb77a981b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_KmeansPlusPlusInitialization.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_KmeansPlusPlusInitialization.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "KmeansPlusPlusInitialization" endpoint { name: "cluster.KmeansPlusPlusInitialization" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_KthOrderStatistic.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_KthOrderStatistic.pbtxt index 9e737cdf29c..98c602c5ebe 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_KthOrderStatistic.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_KthOrderStatistic.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "KthOrderStatistic" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_L2Loss.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_L2Loss.pbtxt index c348e0f0e0b..e00de2d3643 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_L2Loss.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_L2Loss.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "L2Loss" endpoint { name: "nn.L2Loss" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LMDBReader.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LMDBReader.pbtxt index b6cd7bdbb7e..cff04abb5f5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LMDBReader.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LMDBReader.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LMDBReader" endpoint { name: "io.LmdbReader" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LRN.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LRN.pbtxt index d16fea31843..5990d283ee7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LRN.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LRN.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LRN" endpoint { name: "nn.LocalResponseNormalization" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LRNGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LRNGrad.pbtxt index a50e738d785..f6c64ca6d04 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LRNGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LRNGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LRNGrad" endpoint { name: "nn.LocalResponseNormalizationGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LSTMBlockCell.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LSTMBlockCell.pbtxt index a1e1f52a119..dd8baae2a1e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LSTMBlockCell.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LSTMBlockCell.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LSTMBlockCell" endpoint { name: "nn.LSTMBlockCell" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LSTMBlockCellGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LSTMBlockCellGrad.pbtxt index 4d0c82555d2..518a29ef8b1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LSTMBlockCellGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LSTMBlockCellGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LSTMBlockCellGrad" endpoint { name: "nn.LSTMBlockCellGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LeakyReluGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LeakyReluGrad.pbtxt index 9899c64c13e..0b6ab5953da 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LeakyReluGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LeakyReluGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LeakyReluGrad" endpoint { name: "data.LeakyReluGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LearnedUnigramCandidateSampler.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LearnedUnigramCandidateSampler.pbtxt index 5f193da1be1..bc4ab82cdc2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LearnedUnigramCandidateSampler.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LearnedUnigramCandidateSampler.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LearnedUnigramCandidateSampler" endpoint { name: "nn.LearnedUnigramCandidateSampler" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LeftShift.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LeftShift.pbtxt index 44a8727e40f..e00c50f0d68 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LeftShift.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LeftShift.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LeftShift" endpoint { name: "bitwise.LeftShift" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Less.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Less.pbtxt index 577d2556b81..0fa328a0f94 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Less.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Less.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Less" endpoint { name: "math.Less" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LessEqual.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LessEqual.pbtxt index 6cad35c6226..7faf528185e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LessEqual.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LessEqual.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LessEqual" endpoint { name: "math.LessEqual" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Lgamma.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Lgamma.pbtxt index eb7bc9660c0..b6e817d9d69 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Lgamma.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Lgamma.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Lgamma" endpoint { name: "math.Lgamma" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LinSpace.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LinSpace.pbtxt index 599c310021a..09eb5212385 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LinSpace.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LinSpace.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LinSpace" endpoint { name: "LinSpace" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ListDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ListDataset.pbtxt index 8639c922a6b..434f9b2c332 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ListDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ListDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ListDataset" endpoint { name: "data.ListDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ListDiff.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ListDiff.pbtxt index aa94c958f17..4cac575906b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ListDiff.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ListDiff.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ListDiff" endpoint { name: "SetDiff1d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadAllTPUEmbeddingParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadAllTPUEmbeddingParameters.pbtxt index 0efa1b29cf2..1c5cf1f526e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadAllTPUEmbeddingParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadAllTPUEmbeddingParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadAllTPUEmbeddingParameters" endpoint { name: "tpu.LoadAllTPUEmbeddingParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadAndRemapMatrix.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadAndRemapMatrix.pbtxt index 54ee68fde44..3c8984a6cb6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadAndRemapMatrix.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadAndRemapMatrix.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadAndRemapMatrix" endpoint { name: "linalg.LoadAndRemapMatrix" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingADAMParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingADAMParameters.pbtxt index b5c87ea4f40..a591533cf42 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingADAMParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingADAMParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingADAMParameters" endpoint { name: "tpu.LoadTPUEmbeddingADAMParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingADAMParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingADAMParametersGradAccumDebug.pbtxt index 6702f33be77..cd5c2cf8a4d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingADAMParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingADAMParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingADAMParametersGradAccumDebug" endpoint { name: "tpu.LoadTPUEmbeddingADAMParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdadeltaParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdadeltaParameters.pbtxt index 630f6adadcd..4b4466b3ebe 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdadeltaParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdadeltaParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingAdadeltaParameters" endpoint { name: "tpu.LoadTPUEmbeddingAdadeltaParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.pbtxt index 3b4a4a8de5a..3e6fbd0cda8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingAdadeltaParametersGradAccumDebug" endpoint { name: "tpu.LoadTPUEmbeddingAdadeltaParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradMomentumParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradMomentumParameters.pbtxt index 4b81cfb1c65..4ab6b43fb23 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradMomentumParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradMomentumParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingAdagradMomentumParameters" endpoint { name: "tpu.LoadTPUEmbeddingAdagradMomentumParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradParameters.pbtxt index dccec09e382..53dc92a921c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingAdagradParameters" endpoint { name: "tpu.LoadTPUEmbeddingAdagradParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradParametersGradAccumDebug.pbtxt index bd6f676de12..fa8a345407c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingAdagradParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingAdagradParametersGradAccumDebug" endpoint { name: "tpu.LoadTPUEmbeddingAdagradParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingCenteredRMSPropParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingCenteredRMSPropParameters.pbtxt index 5e4356fb9a4..f1781c96808 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingCenteredRMSPropParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingCenteredRMSPropParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingCenteredRMSPropParameters" endpoint { name: "tpu.LoadTPUEmbeddingCenteredRMSPropParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFTRLParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFTRLParameters.pbtxt index e262d4fa79e..6b3377f0d72 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFTRLParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFTRLParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingFTRLParameters" endpoint { name: "tpu.LoadTPUEmbeddingFTRLParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFTRLParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFTRLParametersGradAccumDebug.pbtxt index 363d4f38bfa..f17d8fcc776 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFTRLParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFTRLParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingFTRLParametersGradAccumDebug" endpoint { name: "tpu.LoadTPUEmbeddingFTRLParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFrequencyEstimatorParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFrequencyEstimatorParameters.pbtxt index 97111705c86..33baa18848c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFrequencyEstimatorParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFrequencyEstimatorParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingFrequencyEstimatorParameters" endpoint { name: "tpu.LoadTPUEmbeddingFrequencyEstimatorParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.pbtxt index 1b4493b7fd5..fba39f852dc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug" endpoint { name: "tpu.LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMDLAdagradLightParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMDLAdagradLightParameters.pbtxt index 1336f14f472..d03ed8796c1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMDLAdagradLightParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMDLAdagradLightParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingMDLAdagradLightParameters" endpoint { name: "tpu.LoadTPUEmbeddingMDLAdagradLightParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMomentumParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMomentumParameters.pbtxt index 1e56e9c9000..2ea9c9cfc8c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMomentumParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMomentumParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingMomentumParameters" endpoint { name: "tpu.LoadTPUEmbeddingMomentumParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMomentumParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMomentumParametersGradAccumDebug.pbtxt index 99bdda45764..fccbff595f9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMomentumParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingMomentumParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingMomentumParametersGradAccumDebug" endpoint { name: "tpu.LoadTPUEmbeddingMomentumParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalAdagradParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalAdagradParameters.pbtxt index 6771025a7d8..1b6f3afd838 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalAdagradParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalAdagradParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingProximalAdagradParameters" endpoint { name: "tpu.LoadTPUEmbeddingProximalAdagradParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.pbtxt index 870549ab640..65fe6e27afe 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug" endpoint { name: "tpu.LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalYogiParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalYogiParameters.pbtxt index 2953882f543..f416308b609 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalYogiParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalYogiParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingProximalYogiParameters" endpoint { name: "tpu.LoadTPUEmbeddingProximalYogiParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.pbtxt index c55d7d84731..2656c561bbb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingProximalYogiParametersGradAccumDebug" endpoint { name: "tpu.LoadTPUEmbeddingProximalYogiParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingRMSPropParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingRMSPropParameters.pbtxt index 18a175c7668..bee7db0753b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingRMSPropParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingRMSPropParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingRMSPropParameters" endpoint { name: "tpu.LoadTPUEmbeddingRMSPropParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingRMSPropParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingRMSPropParametersGradAccumDebug.pbtxt index 1dbe67ff290..3becf4df5d3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingRMSPropParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingRMSPropParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingRMSPropParametersGradAccumDebug" endpoint { name: "tpu.LoadTPUEmbeddingRMSPropParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingStochasticGradientDescentParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingStochasticGradientDescentParameters.pbtxt index 846fb1a9c4f..57102d66657 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingStochasticGradientDescentParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingStochasticGradientDescentParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingStochasticGradientDescentParameters" endpoint { name: "tpu.LoadTPUEmbeddingStochasticGradientDescentParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.pbtxt index 86e15662c61..e6dae92e1f2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug" endpoint { name: "tpu.LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Log.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Log.pbtxt index 9d11c26c71a..79d5b27a477 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Log.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Log.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Log" endpoint { name: "math.Log" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Log1p.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Log1p.pbtxt index 6cc1d6e6c82..f91d9ec6a09 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Log1p.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Log1p.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Log1p" endpoint { name: "math.Log1p" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogMatrixDeterminant.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogMatrixDeterminant.pbtxt index 5e52d9ecedc..3828143fa9b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogMatrixDeterminant.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogMatrixDeterminant.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LogMatrixDeterminant" endpoint { name: "linalg.LogMatrixDeterminant" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogSoftmax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogSoftmax.pbtxt index 19518a71ea4..94186851ee9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogSoftmax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogSoftmax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LogSoftmax" endpoint { name: "nn.LogSoftmax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogUniformCandidateSampler.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogUniformCandidateSampler.pbtxt index bdcf01c20fd..c1f7c67d6e5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogUniformCandidateSampler.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogUniformCandidateSampler.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LogUniformCandidateSampler" endpoint { name: "random.LogUniformCandidateSampler" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalAnd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalAnd.pbtxt index 12921dd932c..ebd2c0f5d60 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalAnd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalAnd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LogicalAnd" endpoint { name: "math.LogicalAnd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalNot.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalNot.pbtxt index 9e0960958ed..3665727828c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalNot.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalNot.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LogicalNot" endpoint { name: "math.LogicalNot" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalOr.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalOr.pbtxt index 6c834e46414..e4a567d034e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalOr.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LogicalOr.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LogicalOr" endpoint { name: "math.LogicalOr" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableExportV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableExportV2.pbtxt index d780f2a21d7..0e0c4a4fac3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableExportV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableExportV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LookupTableExportV2" endpoint { name: "LookupTableExport" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableFindV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableFindV2.pbtxt index 2247547b62b..936dd7afecf 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableFindV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableFindV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LookupTableFindV2" endpoint { name: "LookupTableFind" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableImportV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableImportV2.pbtxt index a39cffa12d1..9e7925797da 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableImportV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableImportV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LookupTableImportV2" endpoint { name: "LookupTableImport" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableInsertV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableInsertV2.pbtxt index 037b743b6be..d5db4b3b535 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableInsertV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableInsertV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LookupTableInsertV2" endpoint { name: "LookupTableInsert" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableRemoveV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableRemoveV2.pbtxt index 61f6d8db36a..911df9ae719 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableRemoveV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableRemoveV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LookupTableRemoveV2" endpoint { name: "LookupTableRemove" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableSizeV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableSizeV2.pbtxt index b5526230d76..fafc1f8c910 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableSizeV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LookupTableSizeV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "LookupTableSizeV2" endpoint { name: "LookupTableSize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoopCond.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoopCond.pbtxt index 492f78f62ee..88907751f5e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LoopCond.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LoopCond.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "LoopCond" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_LowerBound.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_LowerBound.pbtxt index 31f1d3038ca..3e92dbec886 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_LowerBound.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_LowerBound.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "LowerBound" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Lu.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Lu.pbtxt index 45e3be1da5e..269a5fbb7d5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Lu.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Lu.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Lu" endpoint { name: "linalg.Lu" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MakeUnique.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MakeUnique.pbtxt index a485fd5b938..3c61d469fd4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MakeUnique.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MakeUnique.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "MakeUnique" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapClear.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapClear.pbtxt index 3ed9bf8a5d8..b5bba0a4941 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapClear.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapClear.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "MapClear" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapDefun.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapDefun.pbtxt index 43b1dc722c0..0e069d080fe 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapDefun.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapDefun.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "MapDefun" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapIncompleteSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapIncompleteSize.pbtxt index 2472209d20a..cd28e3f8a28 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapIncompleteSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapIncompleteSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MapIncompleteSize" out_arg { name: "size" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapPeek.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapPeek.pbtxt index eb1bd158f0d..3541cc96d63 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapPeek.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapPeek.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "MapPeek" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapSize.pbtxt index fe1d5701b4e..30bad2d2ccb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MapSize" out_arg { name: "size" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapStage.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapStage.pbtxt index 6d9f66cfc48..fab2cf93595 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapStage.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapStage.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "MapStage" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapUnstage.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapUnstage.pbtxt index bb118f0fcb9..82be22f5dc8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapUnstage.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapUnstage.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "MapUnstage" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapUnstageNoKey.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapUnstageNoKey.pbtxt index 1004e96482a..dac737d2486 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MapUnstageNoKey.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MapUnstageNoKey.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "MapUnstageNoKey" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatMul.pbtxt index fe4b8405b9c..bc7b2833b00 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatMul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatMul" endpoint { name: "linalg.MatMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatchingFiles.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatchingFiles.pbtxt index bb7b0968957..fea6f70d5cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatchingFiles.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatchingFiles.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatchingFiles" endpoint { name: "io.MatchingFiles" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixBandPart.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixBandPart.pbtxt index eaf426c00ef..24f955501f7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixBandPart.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixBandPart.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixBandPart" endpoint { name: "linalg.BandPart" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDeterminant.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDeterminant.pbtxt index b56d2dad3a8..933a41dddc0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDeterminant.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDeterminant.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixDeterminant" endpoint { name: "linalg.Det" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagPartV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagPartV2.pbtxt index 03f0b064272..c40b2c16745 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagPartV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagPartV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixDiagPartV2" endpoint { name: "linalg.MatrixDiagPart" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagPartV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagPartV3.pbtxt index fe362b5058d..05fadef2d8a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagPartV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagPartV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixDiagPartV3" endpoint { name: "linalg.MatrixDiagPartV3" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagV2.pbtxt index 44d6cc40593..b25ce946738 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixDiagV2" endpoint { name: "linalg.MatrixDiag" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagV3.pbtxt index cebc08f5e23..fdb19d77f1b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixDiagV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixDiagV3" endpoint { name: "linalg.MatrixDiagV3" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixExponential.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixExponential.pbtxt index fb232dab983..9db500e926a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixExponential.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixExponential.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixExponential" endpoint { name: "linalg.MatrixExponential" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixInverse.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixInverse.pbtxt index 68721fc78db..1792f6deef7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixInverse.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixInverse.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixInverse" endpoint { name: "linalg.Inv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixLogarithm.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixLogarithm.pbtxt index 04137ffae79..9fcb9badeb5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixLogarithm.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixLogarithm.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixLogarithm" endpoint { name: "linalg.MatrixLogarithm" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSetDiagV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSetDiagV3.pbtxt index a9d26f74c7f..28a0fb5b934 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSetDiagV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSetDiagV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixSetDiagV3" endpoint { name: "linalg.MatrixSetDiag" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSolve.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSolve.pbtxt index 02c21448bba..cf4bc9b04a6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSolve.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSolve.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixSolve" endpoint { name: "linalg.Solve" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSolveLs.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSolveLs.pbtxt index 9cee578ec28..c7fc10c5447 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSolveLs.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSolveLs.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixSolveLs" endpoint { name: "linalg.MatrixSolveLs" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSquareRoot.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSquareRoot.pbtxt index 14c7624fe37..42d3518b6e5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSquareRoot.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixSquareRoot.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixSquareRoot" endpoint { name: "linalg.Sqrtm" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixTriangularSolve.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixTriangularSolve.pbtxt index 1f61e99efe4..0e20889c3ae 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixTriangularSolve.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MatrixTriangularSolve.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MatrixTriangularSolve" endpoint { name: "linalg.TriangularSolve" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Max.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Max.pbtxt index 03868720edf..112ab3af60a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Max.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Max.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Max" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3D.pbtxt index 17aeb6a8c93..77232e6020a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MaxPool3D" endpoint { name: "nn.MaxPool3d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3DGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3DGrad.pbtxt index ca7a7be835b..bbdc2058f46 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3DGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3DGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MaxPool3DGrad" endpoint { name: "nn.MaxPool3dGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3DGradGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3DGradGrad.pbtxt index c70aa3fe30a..cd2d4d76817 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3DGradGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPool3DGradGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MaxPool3DGradGrad" endpoint { name: "nn.MaxPool3dGradGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradGradV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradGradV2.pbtxt index 2ca8a7b0221..68a36bb8b63 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradGradV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradGradV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MaxPoolGradGradV2" endpoint { name: "nn.MaxPoolGradGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradGradWithArgmax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradGradWithArgmax.pbtxt index d43cf7447cc..84f92c16fd9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradGradWithArgmax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradGradWithArgmax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MaxPoolGradGradWithArgmax" endpoint { name: "nn.MaxPoolGradGradWithArgmax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradV2.pbtxt index 556dd0be502..78c26e0d20a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MaxPoolGradV2" endpoint { name: "nn.MaxPoolGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradWithArgmax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradWithArgmax.pbtxt index c10701f555e..82d58e00566 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradWithArgmax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolGradWithArgmax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MaxPoolGradWithArgmax" endpoint { name: "nn.MaxPoolGradWithArgmax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolV2.pbtxt index 84634943553..a8ebcfea908 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MaxPoolV2" endpoint { name: "nn.MaxPool" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolWithArgmax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolWithArgmax.pbtxt index 43630534cb7..c0e9bb2aa29 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolWithArgmax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MaxPoolWithArgmax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MaxPoolWithArgmax" endpoint { name: "nn.MaxPoolWithArgmax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Maximum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Maximum.pbtxt index 1df9c605305..0a510be5947 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Maximum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Maximum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Maximum" endpoint { name: "math.Maximum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Mean.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Mean.pbtxt index 7bdcdc3d74e..707b1eddb86 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Mean.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Mean.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Mean" endpoint { name: "math.Mean" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Merge.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Merge.pbtxt index 954d5085adf..e04a5e7670d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Merge.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Merge.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Merge" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeDedupData.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeDedupData.pbtxt index f3ec72b9e9d..c62ad85f44d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeDedupData.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeDedupData.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MergeDedupData" endpoint { name: "tpu.MergeDedupData" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeSummary.pbtxt index f52c7c0996b..528399aaec1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MergeSummary" endpoint { name: "summary.MergeSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeV2Checkpoints.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeV2Checkpoints.pbtxt index 8899c8c4ed7..671ae3f9917 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeV2Checkpoints.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MergeV2Checkpoints.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MergeV2Checkpoints" endpoint { name: "train.MergeV2Checkpoints" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Mfcc.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Mfcc.pbtxt index 6cb04e73ff4..018361798cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Mfcc.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Mfcc.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Mfcc" endpoint { name: "audio.Mfcc" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Min.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Min.pbtxt index 72894c1ffd4..3355adfbdde 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Min.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Min.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Min" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Minimum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Minimum.pbtxt index 69f76a98299..cb33aa21fb4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Minimum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Minimum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Minimum" endpoint { name: "math.Minimum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MirrorPad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MirrorPad.pbtxt index e1cb766f8fd..5bc1ebdacbc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MirrorPad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MirrorPad.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "MirrorPad" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MirrorPadGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MirrorPadGrad.pbtxt index ddd8ab3ba1d..0a9c168e261 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MirrorPadGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MirrorPadGrad.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "MirrorPadGrad" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MlirPassthroughOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MlirPassthroughOp.pbtxt index 975ae9a6dee..bf4453b8f2d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MlirPassthroughOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MlirPassthroughOp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "MlirPassthroughOp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Mod.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Mod.pbtxt index 76fbbe97a89..e4003385089 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Mod.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Mod.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Mod" endpoint { name: "math.Mod" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Mul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Mul.pbtxt index 605e110931e..8d1f243721d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Mul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Mul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Mul" endpoint { name: "math.Mul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MulNoNan.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MulNoNan.pbtxt index 23e2247ddcf..e5af10eb9b4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MulNoNan.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MulNoNan.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MulNoNan" endpoint { name: "math.MulNoNan" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIterator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIterator.pbtxt index 81eabf9bdcf..b51de3ab1e5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIterator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIterator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MultiDeviceIterator" endpoint { name: "data.MultiDeviceIterator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorFromStringHandle.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorFromStringHandle.pbtxt index 4006f72d520..59f8a287dad 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorFromStringHandle.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorFromStringHandle.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MultiDeviceIteratorFromStringHandle" endpoint { name: "data.MultiDeviceIteratorFromStringHandle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorGetNextFromShard.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorGetNextFromShard.pbtxt index a7e6fc15082..f36e12598c4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorGetNextFromShard.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorGetNextFromShard.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MultiDeviceIteratorGetNextFromShard" endpoint { name: "data.MultiDeviceIteratorGetNextFromShard" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorInit.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorInit.pbtxt index 1663bc5c226..3b477b5b21f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorInit.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorInit.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MultiDeviceIteratorInit" endpoint { name: "data.MultiDeviceIteratorInit" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorToStringHandle.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorToStringHandle.pbtxt index ff061da390b..3d4958f4495 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorToStringHandle.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MultiDeviceIteratorToStringHandle.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MultiDeviceIteratorToStringHandle" endpoint { name: "data.MultiDeviceIteratorToStringHandle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Multinomial.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Multinomial.pbtxt index bd98bb10b4b..2b693f02801 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Multinomial.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Multinomial.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Multinomial" endpoint { name: "random.Multinomial" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableDenseHashTableV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableDenseHashTableV2.pbtxt index d1f7f26848b..8f20cfd93f5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableDenseHashTableV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableDenseHashTableV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MutableDenseHashTableV2" endpoint { name: "MutableDenseHashTable" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableHashTableOfTensorsV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableHashTableOfTensorsV2.pbtxt index f6d7451267b..d9c26fde9dc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableHashTableOfTensorsV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableHashTableOfTensorsV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MutableHashTableOfTensorsV2" endpoint { name: "MutableHashTableOfTensors" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableHashTableV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableHashTableV2.pbtxt index 45d619d6747..0cfaa1a226b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableHashTableV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MutableHashTableV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MutableHashTableV2" endpoint { name: "MutableHashTable" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MutexLock.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MutexLock.pbtxt index 75c7be5286a..99bf40ba1ed 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MutexLock.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MutexLock.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "MutexLock" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_MutexV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_MutexV2.pbtxt index f89cd106432..17198f4f38c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_MutexV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_MutexV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "MutexV2" endpoint { name: "Mutex" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclAllReduce.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclAllReduce.pbtxt index eec537f2059..cd7390fa15e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclAllReduce.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclAllReduce.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "NcclAllReduce" endpoint: { name: "distribute.NcclAllReduce" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclBroadcast.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclBroadcast.pbtxt index 42150e2853b..74abc5b82d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclBroadcast.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclBroadcast.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "NcclBroadcast" endpoint: { name: "distribute.NcclBroadcast" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclReduce.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclReduce.pbtxt index 77ce0f6d79a..a0ee5487b3e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclReduce.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NcclReduce.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "NcclReduce" endpoint: { name: "distribute.NcclReduce" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Ndtri.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Ndtri.pbtxt index 9e03d7c01f1..9394ba422af 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Ndtri.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Ndtri.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Ndtri" endpoint { name: "math.Ndtri" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NearestNeighbors.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NearestNeighbors.pbtxt index da67ad1992c..d50362c31fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NearestNeighbors.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NearestNeighbors.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "NearestNeighbors" endpoint { name: "image.NearestNeighbors" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Neg.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Neg.pbtxt index c7e9ede2a56..440db9e6414 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Neg.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Neg.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Neg" endpoint { name: "math.Neg" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NegTrain.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NegTrain.pbtxt index eb62186362a..5f381b6e034 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NegTrain.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NegTrain.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "NegTrain" endpoint { name: "train.NegTrain" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NextAfter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NextAfter.pbtxt index b5ab917ba84..c1c88706cb2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NextAfter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NextAfter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "NextAfter" endpoint { name: "math.NextAfter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NextIteration.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NextIteration.pbtxt index bc63f6ada14..63b551aad19 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NextIteration.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NextIteration.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "NextIteration" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NoOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NoOp.pbtxt index 337fb5da146..f3d89127156 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NoOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NoOp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "NoOp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NonDeterministicInts.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NonDeterministicInts.pbtxt index 564fbc5928d..aa8cc027cd6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NonDeterministicInts.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NonDeterministicInts.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "NonDeterministicInts" endpoint { name: "random.NonDeterministicInts" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NonMaxSuppressionV5.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NonMaxSuppressionV5.pbtxt index 31860e21e45..7821a13912d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NonMaxSuppressionV5.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NonMaxSuppressionV5.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "NonMaxSuppressionV5" endpoint { name: "image.NonMaxSuppression" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NonMaxSuppressionWithOverlaps.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NonMaxSuppressionWithOverlaps.pbtxt index 06fa52920d2..de5488bb255 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NonMaxSuppressionWithOverlaps.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NonMaxSuppressionWithOverlaps.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "NonMaxSuppressionWithOverlaps" endpoint { name: "image.NonMaxSuppressionWithOverlaps" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NotEqual.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NotEqual.pbtxt index 0d99af40b5d..5b587960e14 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NotEqual.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NotEqual.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "NotEqual" endpoint { name: "math.NotEqual" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_NthElement.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_NthElement.pbtxt index 57097e634aa..7930c041ad3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_NthElement.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_NthElement.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "NthElement" endpoint { name: "nn.NthElement" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OneHot.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OneHot.pbtxt index 66872d5eb8e..116d0272f16 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OneHot.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OneHot.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "OneHot" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OnesLike.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OnesLike.pbtxt index 97abe0814a9..06bdeacbd0e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OnesLike.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OnesLike.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "OnesLike" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalFromValue.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalFromValue.pbtxt index d251fd5d948..282d866f180 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalFromValue.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalFromValue.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OptionalFromValue" endpoint { name: "data.OptionalFromValue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalGetValue.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalGetValue.pbtxt index 7fcdb5ac694..b0be584ec90 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalGetValue.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalGetValue.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OptionalGetValue" endpoint { name: "data.OptionalGetValue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalHasValue.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalHasValue.pbtxt index 4ffa15b564c..f2778d04bb8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalHasValue.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalHasValue.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OptionalHasValue" endpoint { name: "data.OptionalHasValue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalNone.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalNone.pbtxt index cec29a42ae2..77ec49c964c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalNone.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OptionalNone.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OptionalNone" endpoint { name: "data.OptionalNone" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapClear.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapClear.pbtxt index e36b2aa3e4f..30c9cc626d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapClear.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapClear.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "OrderedMapClear" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapIncompleteSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapIncompleteSize.pbtxt index 27d68e2d99d..ef2e4e4f272 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapIncompleteSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapIncompleteSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OrderedMapIncompleteSize" out_arg { name: "size" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapPeek.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapPeek.pbtxt index 06fc2182773..2ac8b71d2fb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapPeek.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapPeek.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "OrderedMapPeek" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapSize.pbtxt index 30e6215a0ee..47e4f6188ce 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OrderedMapSize" out_arg { name: "size" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapStage.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapStage.pbtxt index 8b579d21a0e..22a96eaccb0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapStage.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapStage.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "OrderedMapStage" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapUnstage.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapUnstage.pbtxt index d3d6862fbe0..b617e0ad11e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapUnstage.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapUnstage.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "OrderedMapUnstage" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapUnstageNoKey.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapUnstageNoKey.pbtxt index 3d275c85d94..e5beaff8915 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapUnstageNoKey.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OrderedMapUnstageNoKey.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "OrderedMapUnstageNoKey" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeue.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeue.pbtxt index 15f5d29a2b6..6a8b5d5f562 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeue.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeue.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OutfeedDequeue" endpoint { name: "tpu.OutfeedDequeue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueTuple.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueTuple.pbtxt index 3e15876cfbf..9b8a9b7bebd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueTuple.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueTuple.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OutfeedDequeueTuple" endpoint { name: "tpu.OutfeedDequeueTuple" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueTupleV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueTupleV2.pbtxt index 7fdebc8a3bd..7fef814d5b3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueTupleV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueTupleV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OutfeedDequeueTupleV2" endpoint { name: "tpu.OutfeedDequeueTupleV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueV2.pbtxt index cc61dd2a047..02c947f23f8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedDequeueV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OutfeedDequeueV2" endpoint { name: "tpu.OutfeedDequeueV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedEnqueue.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedEnqueue.pbtxt index 02be5eee442..18694993470 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedEnqueue.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedEnqueue.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OutfeedEnqueue" endpoint { name: "tpu.OutfeedEnqueue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedEnqueueTuple.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedEnqueueTuple.pbtxt index 9b56fecc72c..e4347fe0bcd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedEnqueueTuple.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_OutfeedEnqueueTuple.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "OutfeedEnqueueTuple" endpoint { name: "tpu.OutfeedEnqueueTuple" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Pack.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Pack.pbtxt index d9e9897d77c..2be5e46f791 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Pack.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Pack.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Pack" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_PadV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_PadV2.pbtxt index ffc12164560..2462c556cf3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_PadV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_PadV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "PadV2" endpoint { name: "Pad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_PaddingFIFOQueueV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_PaddingFIFOQueueV2.pbtxt index 605025be791..d0b4a712ff8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_PaddingFIFOQueueV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_PaddingFIFOQueueV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "PaddingFIFOQueueV2" endpoint { name: "io.PaddingFifoQueue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelConcat.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelConcat.pbtxt index 0b17c7d2569..cead44173d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelConcat.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelConcat.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ParallelConcat" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelDynamicStitch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelDynamicStitch.pbtxt index 79a55b763f3..a8bebe9f4f5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelDynamicStitch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelDynamicStitch.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ParallelDynamicStitch" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelFilterDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelFilterDataset.pbtxt index 4536ba19dbc..32e189b0963 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelFilterDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParallelFilterDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ParallelFilterDataset" endpoint { name: "data.ParallelFilterDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParameterizedTruncatedNormal.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParameterizedTruncatedNormal.pbtxt index 26ca2fc86fa..e271245d703 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParameterizedTruncatedNormal.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParameterizedTruncatedNormal.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ParameterizedTruncatedNormal" endpoint { name: "random.ParameterizedTruncatedNormal" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseExampleV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseExampleV2.pbtxt index 62f76eb8ccc..c78eb77249c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseExampleV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseExampleV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ParseExampleV2" endpoint { name: "io.ParseExample" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSequenceExampleV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSequenceExampleV2.pbtxt index 6a873ad57c7..3ce6e01560f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSequenceExampleV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSequenceExampleV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ParseSequenceExampleV2" endpoint { name: "io.ParseSequenceExample" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSingleExample.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSingleExample.pbtxt index 7559957b35e..d2dfdc20ea2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSingleExample.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSingleExample.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ParseSingleExample" endpoint { name: "io.ParseSingleExample" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSingleSequenceExample.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSingleSequenceExample.pbtxt index 00eb325b2a4..2a83b9105e0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSingleSequenceExample.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseSingleSequenceExample.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ParseSingleSequenceExample" endpoint { name: "io.ParseSingleSequenceExample" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseTensor.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseTensor.pbtxt index a78cdc7f5c4..e8e5db934af 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseTensor.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ParseTensor.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ParseTensor" endpoint { name: "io.ParseTensor" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_PartitionedCall.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_PartitionedCall.pbtxt index 1ac10b60280..268c519a8a7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_PartitionedCall.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_PartitionedCall.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "PartitionedCall" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Placeholder.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Placeholder.pbtxt index 5e6daa2ae42..2e83fe4d8f8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Placeholder.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Placeholder.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Placeholder" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_PlaceholderWithDefault.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_PlaceholderWithDefault.pbtxt index 59067a9c688..d20aff9cb92 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_PlaceholderWithDefault.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_PlaceholderWithDefault.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "PlaceholderWithDefault" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Polygamma.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Polygamma.pbtxt index 746b3375a0f..f81d95924f1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Polygamma.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Polygamma.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Polygamma" endpoint { name: "math.Polygamma" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_PopulationCount.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_PopulationCount.pbtxt index 6aacdf4d121..840404a23d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_PopulationCount.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_PopulationCount.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "PopulationCount" endpoint { name: "math.PopulationCount" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Pow.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Pow.pbtxt index e7eaaed6952..8657e4afb98 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Pow.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Pow.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Pow" endpoint { name: "math.Pow" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Prelinearize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Prelinearize.pbtxt index 3d8c0d77a16..71d98c0868f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Prelinearize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Prelinearize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Prelinearize" endpoint { name: "tpu.Prelinearize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_PrelinearizeTuple.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_PrelinearizeTuple.pbtxt index e00a3b58b82..59751130dfc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_PrelinearizeTuple.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_PrelinearizeTuple.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "PrelinearizeTuple" endpoint { name: "tpu.PrelinearizeTuple" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_PreventGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_PreventGradient.pbtxt index 4731f21af4c..bafa0a5a739 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_PreventGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_PreventGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "PreventGradient" endpoint { name: "train.PreventGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_PrintV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_PrintV2.pbtxt index d1e4d74b1e3..573751c55b8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_PrintV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_PrintV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "PrintV2" endpoint { name: "Print" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_PriorityQueueV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_PriorityQueueV2.pbtxt index 1f6a6f2906b..8bd3b0a04dc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_PriorityQueueV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_PriorityQueueV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "PriorityQueueV2" endpoint { name: "io.PriorityQueue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Prod.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Prod.pbtxt index 7a98972a871..d1c62ee4c7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Prod.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Prod.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Prod" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Qr.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Qr.pbtxt index c0d31e79778..13b372131af 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Qr.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Qr.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Qr" endpoint { name: "linalg.Qr" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV3.pbtxt index ce20ef12386..49b0b0d4878 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizeAndDequantizeV3" endpoint { name: "quantization.QuantizeAndDequantizeV3" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV4.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV4.pbtxt index 9920bcd06a4..4e9780f470c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV4.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV4.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizeAndDequantizeV4" endpoint { name: "quantization.QuantizeAndDequantizeV4" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV4Grad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV4Grad.pbtxt index 49ee4ab14a7..3c86a135f58 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV4Grad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeAndDequantizeV4Grad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizeAndDequantizeV4Grad" endpoint { name: "quantization.QuantizeAndDequantizeV4Grad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeDownAndShrinkRange.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeDownAndShrinkRange.pbtxt index 7119f53cb29..ac2dc64b29b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeDownAndShrinkRange.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeDownAndShrinkRange.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizeDownAndShrinkRange" endpoint { name: "quantization.QuantizeDownAndShrinkRange" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeV2.pbtxt index 25c9c3bdce4..8dd0155b0cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizeV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizeV2" endpoint { name: "quantization.Quantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedAdd.pbtxt index 1a2bfa36ed7..409160600a2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedAdd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedAdd" endpoint { name: "math.QuantizedAdd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedAvgPool.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedAvgPool.pbtxt index 7f16fb046dd..4f6112fd2d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedAvgPool.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedAvgPool.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedAvgPool" endpoint { name: "nn.QuantizedAvgPool" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedBatchNormWithGlobalNormalization.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedBatchNormWithGlobalNormalization.pbtxt index 2101ce5d690..f83d5c2433a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedBatchNormWithGlobalNormalization.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedBatchNormWithGlobalNormalization.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedBatchNormWithGlobalNormalization" endpoint { name: "nn.QuantizedBatchNormWithGlobalNormalization" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedBiasAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedBiasAdd.pbtxt index c18fb1c5741..42af03225d9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedBiasAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedBiasAdd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedBiasAdd" endpoint { name: "nn.QuantizedBiasAdd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConcat.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConcat.pbtxt index 98ad8f92ac6..6f494b440b1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConcat.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConcat.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConcat" endpoint { name: "quantization.QuantizedConcat" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2D.pbtxt index 9909157603e..a6e20f4585d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2D" endpoint { name: "nn.QuantizedConv2d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndRelu.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndRelu.pbtxt index 6f808e6ec3d..11babc82e64 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndRelu.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndRelu.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2DAndRelu" endpoint { name: "nn.QuantizedConv2DAndRelu" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndReluAndRequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndReluAndRequantize.pbtxt index 78f9b567a8c..69598eb29e7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndReluAndRequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndReluAndRequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2DAndReluAndRequantize" endpoint { name: "nn.QuantizedConv2DAndReluAndRequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndRequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndRequantize.pbtxt index d28e8d45855..074c8bb81dc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndRequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DAndRequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2DAndRequantize" endpoint { name: "nn.QuantizedConv2DAndRequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DPerChannel.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DPerChannel.pbtxt index c9af2597c01..8e0ad23bd42 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DPerChannel.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DPerChannel.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2DPerChannel" endpoint { name: "nn.QuantizedConv2DPerChannel" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBias.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBias.pbtxt index 3eb73496eb3..bfb35fd99ee 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBias.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBias.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2DWithBias" endpoint { name: "nn.QuantizedConv2DWithBias" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndRelu.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndRelu.pbtxt index c889787cf35..094b5484db9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndRelu.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndRelu.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2DWithBiasAndRelu" endpoint { name: "nn.QuantizedConv2DWithBiasAndRelu" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndReluAndRequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndReluAndRequantize.pbtxt index e24261bb2ba..45a9ae59f11 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndReluAndRequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndReluAndRequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2DWithBiasAndReluAndRequantize" endpoint { name: "nn.QuantizedConv2DWithBiasAndReluAndRequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndRequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndRequantize.pbtxt index f280bd25ff0..e2360686b4a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndRequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasAndRequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2DWithBiasAndRequantize" endpoint { name: "nn.QuantizedConv2DWithBiasAndRequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.pbtxt index ef2f2a9da36..16c15d1bcbb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2DWithBiasSignedSumAndReluAndRequantize" endpoint { name: "nn.QuantizedConv2DWithBiasSignedSumAndReluAndRequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSumAndRelu.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSumAndRelu.pbtxt index c1fa4efcdf6..210d5287924 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSumAndRelu.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSumAndRelu.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2DWithBiasSumAndRelu" endpoint { name: "nn.QuantizedConv2DWithBiasSumAndRelu" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSumAndReluAndRequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSumAndReluAndRequantize.pbtxt index 08c7dc22698..910800ac4f0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSumAndReluAndRequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedConv2DWithBiasSumAndReluAndRequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedConv2DWithBiasSumAndReluAndRequantize" endpoint { name: "nn.QuantizedConv2DWithBiasSumAndReluAndRequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2D.pbtxt index 8b77dac4062..cfcc863566b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedDepthwiseConv2D" endpoint { name: "nn.QuantizedDepthwiseConv2D" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBias.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBias.pbtxt index e69ceccf887..961de7a11f7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBias.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBias.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedDepthwiseConv2DWithBias" endpoint { name: "nn.QuantizedDepthwiseConv2DWithBias" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt index 72fac5d7da9..4470675b660 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedDepthwiseConv2DWithBiasAndRelu" endpoint { name: "nn.QuantizedDepthwiseConv2DWithBiasAndRelu" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt index 823e41213a5..e2673935a16 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize" endpoint { name: "nn.QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedInstanceNorm.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedInstanceNorm.pbtxt index bbd2e7fc5ec..52620d0f998 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedInstanceNorm.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedInstanceNorm.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedInstanceNorm" endpoint { name: "nn.QuantizedInstanceNorm" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMul.pbtxt index 7962cbade6f..40f0a5e788c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedMatMul" endpoint { name: "linalg.QuantizedMatMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBias.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBias.pbtxt index 3642c953987..65cd7780258 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBias.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBias.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedMatMulWithBias" endpoint { name: "linalg.QuantizedMatMulWithBias" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndDequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndDequantize.pbtxt index 45af20c078c..2c47dfba1b0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndDequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndDequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedMatMulWithBiasAndDequantize" endpoint { name: "quantization.QuantizedMatMulWithBiasAndDequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndRelu.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndRelu.pbtxt index eba6765ac5d..9f7d19c4203 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndRelu.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndRelu.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedMatMulWithBiasAndRelu" endpoint { name: "linalg.QuantizedMatMulWithBiasAndRelu" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndReluAndRequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndReluAndRequantize.pbtxt index 76c88313eaa..548eeb7b9ef 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndReluAndRequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndReluAndRequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedMatMulWithBiasAndReluAndRequantize" endpoint { name: "linalg.QuantizedMatMulWithBiasAndReluAndRequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndRequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndRequantize.pbtxt index 6540e8a93fe..24994b5662f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndRequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMatMulWithBiasAndRequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedMatMulWithBiasAndRequantize" endpoint { name: "quantization.QuantizedMatMulWithBiasAndRequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMaxPool.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMaxPool.pbtxt index 57e900494e9..40f6f65c9b1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMaxPool.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMaxPool.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedMaxPool" endpoint { name: "nn.QuantizedMaxPool" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMul.pbtxt index be23ef706e2..6b14b69beb5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedMul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedMul" endpoint { name: "math.QuantizedMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedRelu.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedRelu.pbtxt index 7b9a11640ba..8e1b314e688 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedRelu.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedRelu.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedRelu" endpoint { name: "nn.QuantizedRelu" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedRelu6.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedRelu6.pbtxt index 6a60e2112ee..f5230201707 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedRelu6.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedRelu6.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedRelu6" endpoint { name: "nn.QuantizedRelu6" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedReluX.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedReluX.pbtxt index cc47d322b27..a52915868d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedReluX.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedReluX.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedReluX" endpoint { name: "nn.QuantizedReluX" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedReshape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedReshape.pbtxt index 4557853d94f..f2049b9f380 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedReshape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedReshape.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedReshape" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedResizeBilinear.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedResizeBilinear.pbtxt index 81dca490944..28191a12de9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedResizeBilinear.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QuantizedResizeBilinear.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QuantizedResizeBilinear" endpoint { name: "image.QuantizedResizeBilinear" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueCloseV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueCloseV2.pbtxt index ce779650e5b..08c2af13ab5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueCloseV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueCloseV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QueueCloseV2" endpoint { name: "io.QueueClose" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueManyV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueManyV2.pbtxt index 10fe198ff26..e0cdf5ce764 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueManyV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueManyV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QueueDequeueManyV2" endpoint { name: "io.QueueDequeueMany" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueUpToV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueUpToV2.pbtxt index fadea0926b6..715b614ccda 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueUpToV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueUpToV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QueueDequeueUpToV2" endpoint { name: "io.QueueDequeueUpTo" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueV2.pbtxt index 7ba03afbfeb..670a81b09c6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueDequeueV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QueueDequeueV2" endpoint { name: "io.QueueDequeue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueEnqueueManyV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueEnqueueManyV2.pbtxt index be3fed47896..8f08727b990 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueEnqueueManyV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueEnqueueManyV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QueueEnqueueManyV2" endpoint { name: "io.QueueEnqueueMany" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueEnqueueV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueEnqueueV2.pbtxt index e71a2211e1e..56700dbe62d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueEnqueueV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueEnqueueV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QueueEnqueueV2" endpoint { name: "io.QueueEnqueue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueIsClosedV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueIsClosedV2.pbtxt index 148d313a6d3..e3c27b82fe8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueIsClosedV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueIsClosedV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QueueIsClosedV2" endpoint { name: "io.QueueIsClosed" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueSizeV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueSizeV2.pbtxt index bc17c8daf96..f352e15e0c7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueSizeV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_QueueSizeV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "QueueSizeV2" endpoint { name: "io.QueueSize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT.pbtxt index 9576600e756..708de1951ae 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RFFT" endpoint { name: "signal.Rfft" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT2D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT2D.pbtxt index 41d638b26a8..8488c65b5d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT2D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT2D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RFFT2D" endpoint { name: "signal.Rfft2d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT3D.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT3D.pbtxt index 7a762d22e5c..09218cd6296 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT3D.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFT3D.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RFFT3D" endpoint { name: "signal.Rfft3d" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFTND.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFTND.pbtxt index 51476b79b65..4b46e0f0d31 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFTND.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RFFTND.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RFFTND" endpoint { name: "signal.RfftNd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RGBToHSV.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RGBToHSV.pbtxt index 1b35891ae2c..2172f52405b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RGBToHSV.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RGBToHSV.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RGBToHSV" endpoint { name: "image.RgbToHsv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedBincount.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedBincount.pbtxt index b52b8ff650e..632be33d30d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedBincount.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedBincount.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedBincount" endpoint { name: "ragged.RaggedBincount" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedCountSparseOutput.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedCountSparseOutput.pbtxt index 74597fd7b71..e74b9bd9d0a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedCountSparseOutput.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedCountSparseOutput.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedCountSparseOutput" endpoint { name: "ragged.RaggedCountSparseOutput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedCross.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedCross.pbtxt index bebf9f47f78..7da3096c7ef 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedCross.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedCross.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedCross" endpoint { name: "ragged.RaggedCross" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedFillEmptyRows.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedFillEmptyRows.pbtxt index 83fc6eb549b..5f135f87e74 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedFillEmptyRows.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedFillEmptyRows.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedFillEmptyRows" endpoint { name: "ragged.RaggedFillEmptyRows" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedFillEmptyRowsGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedFillEmptyRowsGrad.pbtxt index fd56e4c69cf..5f8f1790f32 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedFillEmptyRowsGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedFillEmptyRowsGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedFillEmptyRowsGrad" endpoint { name: "ragged.RaggedFillEmptyRowsGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedGather.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedGather.pbtxt index ac066f5d68c..10da3a31954 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedGather.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedGather.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedGather" endpoint { name: "ragged.RaggedGather" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedRange.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedRange.pbtxt index 8853575bda6..6ec658b61fe 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedRange.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedRange.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedRange" endpoint { name: "ragged.RaggedRange" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorFromVariant.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorFromVariant.pbtxt index 9594d2d6ac6..2067148bde1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorFromVariant.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorFromVariant.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedTensorFromVariant" endpoint { name: "ragged.RaggedTensorFromVariant" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToSparse.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToSparse.pbtxt index 1a658898238..c6d61a22606 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToSparse.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToSparse.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedTensorToSparse" endpoint { name: "ragged.RaggedTensorToSparse" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToTensor.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToTensor.pbtxt index 25d8f380e2f..2bae8ad3de2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToTensor.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToTensor.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedTensorToTensor" endpoint { name: "ragged.RaggedTensorToTensor" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToVariant.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToVariant.pbtxt index 5da45895a35..3e4b2029a1b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToVariant.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToVariant.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedTensorToVariant" endpoint { name: "ragged.RaggedTensorToVariant" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToVariantGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToVariantGradient.pbtxt index 53abb4a0d2f..a09acd4debd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToVariantGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RaggedTensorToVariantGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RaggedTensorToVariantGradient" endpoint { name: "ragged.RaggedTensorToVariantGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomCrop.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomCrop.pbtxt index a3b8a3cecda..be299f2ed38 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomCrop.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomCrop.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RandomCrop" endpoint { name: "image.RandomCrop" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomGamma.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomGamma.pbtxt index 927f2c56937..c8fbfbf0134 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomGamma.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomGamma.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RandomGamma" endpoint { name: "random.RandomGamma" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomGammaGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomGammaGrad.pbtxt index 9257495c9bd..6d0b3466690 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomGammaGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomGammaGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RandomGammaGrad" endpoint { name: "random.RandomGammaGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomIndexShuffle.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomIndexShuffle.pbtxt index 93c95a499cd..0af6f3e5b1e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomIndexShuffle.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomIndexShuffle.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "RandomIndexShuffle" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomPoissonV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomPoissonV2.pbtxt index 5efe01bf401..09bdecdaa10 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomPoissonV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomPoissonV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RandomPoissonV2" endpoint { name: "random.RandomPoisson" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomShuffle.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomShuffle.pbtxt index 6dcd12fd375..5d0d7a3b680 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomShuffle.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomShuffle.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RandomShuffle" endpoint { name: "random.RandomShuffle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomShuffleQueueV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomShuffleQueueV2.pbtxt index 779363303ca..4dd84fac74d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomShuffleQueueV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomShuffleQueueV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RandomShuffleQueueV2" endpoint { name: "io.RandomShuffleQueue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomStandardNormal.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomStandardNormal.pbtxt index 413fc87bdf0..5ac99b9005b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomStandardNormal.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomStandardNormal.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RandomStandardNormal" endpoint { name: "random.RandomStandardNormal" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomUniform.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomUniform.pbtxt index 2a93df83df6..bdec5ac99d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomUniform.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomUniform.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RandomUniform" endpoint { name: "random.RandomUniform" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomUniformInt.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomUniformInt.pbtxt index a1383f406a3..4102517f3de 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomUniformInt.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RandomUniformInt.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RandomUniformInt" endpoint { name: "random.RandomUniformInt" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Range.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Range.pbtxt index 24f3787a8e3..dbda35b7374 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Range.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Range.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Range" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Rank.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Rank.pbtxt index baa84aab10d..dc306a7ae56 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Rank.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Rank.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Rank" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadFile.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadFile.pbtxt index f74250d42f7..8d2c022f428 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadFile.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadFile.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReadFile" endpoint { name: "io.ReadFile" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadVariableOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadVariableOp.pbtxt index 018886d5b82..7f053e301a6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadVariableOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadVariableOp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ReadVariableOp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadVariableXlaSplitND.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadVariableXlaSplitND.pbtxt index 81374fce42a..5d6bfb45373 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadVariableXlaSplitND.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReadVariableXlaSplitND.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReadVariableXlaSplitND" endpoint { name: "xla.ReadVariableSplitND" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderNumRecordsProducedV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderNumRecordsProducedV2.pbtxt index 54a30abe187..13578bcba83 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderNumRecordsProducedV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderNumRecordsProducedV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReaderNumRecordsProducedV2" endpoint { name: "io.ReaderNumRecordsProduced" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderNumWorkUnitsCompletedV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderNumWorkUnitsCompletedV2.pbtxt index 0904ba19e53..1a72c3be10a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderNumWorkUnitsCompletedV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderNumWorkUnitsCompletedV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReaderNumWorkUnitsCompletedV2" endpoint { name: "io.ReaderNumWorkUnitsCompleted" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderReadUpToV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderReadUpToV2.pbtxt index 777d09fa2c6..06a316fbb70 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderReadUpToV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderReadUpToV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReaderReadUpToV2" endpoint { name: "io.ReaderReadUpTo" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderReadV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderReadV2.pbtxt index a5d45bd1db9..64bd40cdde8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderReadV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderReadV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReaderReadV2" endpoint { name: "io.ReaderRead" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderResetV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderResetV2.pbtxt index 265a3442f53..05bd5c48bc3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderResetV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderResetV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReaderResetV2" endpoint { name: "io.ReaderReset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderRestoreStateV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderRestoreStateV2.pbtxt index 4728ce7796b..c53c47ff372 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderRestoreStateV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderRestoreStateV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReaderRestoreStateV2" endpoint { name: "io.ReaderRestoreState" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderSerializeStateV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderSerializeStateV2.pbtxt index aa396095b17..ec18d3c71b6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderSerializeStateV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReaderSerializeStateV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReaderSerializeStateV2" endpoint { name: "io.ReaderSerializeState" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Real.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Real.pbtxt index 3aaea928dec..3ddd3bc902a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Real.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Real.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Real" endpoint { name: "math.Real" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RealDiv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RealDiv.pbtxt index 415bd29da04..366c95f2566 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RealDiv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RealDiv.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RealDiv" endpoint { name: "math.RealDiv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Reciprocal.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Reciprocal.pbtxt index 1c0d787c24e..bb6956bbe3c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Reciprocal.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Reciprocal.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Reciprocal" endpoint { name: "math.Reciprocal" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReciprocalGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReciprocalGrad.pbtxt index 68879669b5f..57cc8c630e1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReciprocalGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReciprocalGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReciprocalGrad" endpoint { name: "math.ReciprocalGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RecordInput.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RecordInput.pbtxt index c4807c68dee..bf8836b3d81 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RecordInput.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RecordInput.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RecordInput" endpoint { name: "random.RecordInput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Recv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Recv.pbtxt index f0d5d9f0705..6ba56fa3392 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Recv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Recv.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Recv" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RecvTPUEmbeddingActivations.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RecvTPUEmbeddingActivations.pbtxt index 61e46884b36..05ce63f87aa 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RecvTPUEmbeddingActivations.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RecvTPUEmbeddingActivations.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RecvTPUEmbeddingActivations" endpoint { name: "tpu.RecvTPUEmbeddingActivations" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReduceJoin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReduceJoin.pbtxt index 79193222018..bb2b90169a1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReduceJoin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReduceJoin.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReduceJoin" endpoint { name: "strings.ReduceJoin" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefEnter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefEnter.pbtxt index 9cd2281bc6a..886f9cc3436 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefEnter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefEnter.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "RefEnter" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefExit.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefExit.pbtxt index 67e8d39c9af..1495c957912 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefExit.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefExit.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "RefExit" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefIdentity.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefIdentity.pbtxt index 53483bd1bb7..013b3bcce61 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefIdentity.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefIdentity.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "RefIdentity" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefMerge.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefMerge.pbtxt index 6ea3145841b..97599f361be 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefMerge.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefMerge.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "RefMerge" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefNextIteration.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefNextIteration.pbtxt index 5d008204b7a..0b94ec2d5d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefNextIteration.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefNextIteration.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "RefNextIteration" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefSelect.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefSelect.pbtxt index d7cda2d5b30..dc135b3cc98 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefSelect.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefSelect.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "RefSelect" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefSwitch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefSwitch.pbtxt index 78261d8b7e6..abccabbf444 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RefSwitch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RefSwitch.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "RefSwitch" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RegexFullMatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RegexFullMatch.pbtxt index 7f88e24eac6..ed0a0765b5c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RegexFullMatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RegexFullMatch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RegexFullMatch" endpoint { name: "strings.RegexFullMatch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RegexReplace.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RegexReplace.pbtxt index 01c9e93cab7..a2987dba302 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RegexReplace.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RegexReplace.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RegexReplace" endpoint { name: "strings.RegexReplace" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Relayout.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Relayout.pbtxt index 6eb3853eb8c..50f3036cc78 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Relayout.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Relayout.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Relayout" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RelayoutLike.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RelayoutLike.pbtxt index 2abfdc15f19..b83aaf0cf61 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RelayoutLike.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RelayoutLike.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RelayoutLike" endpoint { name: "RelayoutLike" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu.pbtxt index 39d7fec4526..87e110a0739 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Relu" endpoint { name: "nn.Relu" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu6.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu6.pbtxt index fcc012b5033..c1dc6c6d205 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu6.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu6.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Relu6" endpoint { name: "nn.Relu6" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu6Grad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu6Grad.pbtxt index 33e959cc7b3..bb4621ffb5b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu6Grad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Relu6Grad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Relu6Grad" endpoint { name: "nn.Relu6Grad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReluGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReluGrad.pbtxt index ec4a8b5f972..7830ad371d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReluGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReluGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReluGrad" endpoint { name: "nn.ReluGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RemoteCall.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RemoteCall.pbtxt index a0ddb017847..b2f13cc48b9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RemoteCall.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RemoteCall.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "RemoteCall" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RemoteFusedGraphExecute.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RemoteFusedGraphExecute.pbtxt index b73e633ef2c..c30673aa76e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RemoteFusedGraphExecute.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RemoteFusedGraphExecute.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "RemoteFusedGraphExecute" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizationRange.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizationRange.pbtxt index d365e899255..81e17cf420d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizationRange.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizationRange.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RequantizationRange" endpoint { name: "quantization.RequantizationRange" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizationRangePerChannel.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizationRangePerChannel.pbtxt index 7630f0d58be..2073052bfe9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizationRangePerChannel.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizationRangePerChannel.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RequantizationRangePerChannel" endpoint { name: "math.RequantizationRangePerChannel" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Requantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Requantize.pbtxt index d397cde4db9..c771cef0746 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Requantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Requantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Requantize" endpoint { name: "quantization.Requantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizePerChannel.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizePerChannel.pbtxt index 3cbf3ed48f7..2539fbe9528 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizePerChannel.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RequantizePerChannel.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RequantizePerChannel" endpoint { name: "math.RequantizePerChannel" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Reshape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Reshape.pbtxt index 4bf3a409d1a..bd628df6d68 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Reshape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Reshape.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Reshape" endpoint { name: "Reshape" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeArea.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeArea.pbtxt index 5358c18d4b4..2514478bc1e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeArea.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeArea.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResizeArea" endpoint { name: "image.ResizeArea" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBicubic.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBicubic.pbtxt index 0d0942e2662..669b0889911 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBicubic.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBicubic.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResizeBicubic" endpoint { name: "image.ResizeBicubic" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBicubicGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBicubicGrad.pbtxt index 12e61dc8238..63478567394 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBicubicGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBicubicGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResizeBicubicGrad" endpoint { name: "image.ResizeBicubicGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBilinear.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBilinear.pbtxt index ad123744a92..42bc9578c0b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBilinear.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBilinear.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResizeBilinear" endpoint { name: "image.ResizeBilinear" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBilinearGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBilinearGrad.pbtxt index 04f3e9f19ef..88bccdf83ca 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBilinearGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeBilinearGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResizeBilinearGrad" endpoint { name: "image.ResizeBilinearGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeNearestNeighbor.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeNearestNeighbor.pbtxt index 86ad39a5173..84f8e26218d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeNearestNeighbor.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeNearestNeighbor.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResizeNearestNeighbor" endpoint { name: "image.ResizeNearestNeighbor" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeNearestNeighborGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeNearestNeighborGrad.pbtxt index 70eeb906fab..2b5ce61b1cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeNearestNeighborGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResizeNearestNeighborGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResizeNearestNeighborGrad" endpoint { name: "image.ResizeNearestNeighborGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorApplyGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorApplyGradient.pbtxt index 032baafbf7e..2463e311f36 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorApplyGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorApplyGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceAccumulatorApplyGradient" endpoint { name: "train.ResourceAccumulatorApplyGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorNumAccumulated.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorNumAccumulated.pbtxt index 9bbd06f6e07..414247dc55b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorNumAccumulated.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorNumAccumulated.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceAccumulatorNumAccumulated" endpoint { name: "train.ResourceAccumulatorNumAccumulated" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorSetGlobalStep.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorSetGlobalStep.pbtxt index 171c7651371..02083395b15 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorSetGlobalStep.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorSetGlobalStep.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceAccumulatorSetGlobalStep" endpoint { name: "train.ResourceAccumulatorSetGlobalStep" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorTakeGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorTakeGradient.pbtxt index 518a496b88a..7d7fbd9c9ba 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorTakeGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceAccumulatorTakeGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceAccumulatorTakeGradient" endpoint { name: "train.ResourceAccumulatorTakeGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdaMax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdaMax.pbtxt index ff57bd5849a..cbe0abd7fd2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdaMax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdaMax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyAdaMax" endpoint { name: "train.ResourceApplyAdaMax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdadelta.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdadelta.pbtxt index d4369f0eade..11ea32f0474 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdadelta.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdadelta.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyAdadelta" endpoint { name: "train.ResourceApplyAdadelta" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdagradDA.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdagradDA.pbtxt index afddaaff573..7de1a78a3e7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdagradDA.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdagradDA.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyAdagradDA" endpoint { name: "train.ResourceApplyAdagradDa" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdagradV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdagradV2.pbtxt index 71def75670a..4b2cdf69a9b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdagradV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdagradV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyAdagradV2" endpoint { name: "train.ResourceApplyAdagrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdam.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdam.pbtxt index 390bd999c45..13b9b145b78 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdam.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdam.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyAdam" endpoint { name: "train.ResourceApplyAdam" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdamWithAmsgrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdamWithAmsgrad.pbtxt index 503e3b142b3..3afb7a28c5c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdamWithAmsgrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAdamWithAmsgrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyAdamWithAmsgrad" endpoint { name: "train.ResourceApplyAdamWithAmsgrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAddSign.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAddSign.pbtxt index bf944477be3..8e57cf8d4c9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAddSign.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyAddSign.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyAddSign" endpoint { name: "train.ResourceApplyAddSign" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyCenteredRMSProp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyCenteredRMSProp.pbtxt index 85c97b430a8..5bc55386fb3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyCenteredRMSProp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyCenteredRMSProp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyCenteredRMSProp" endpoint { name: "train.ResourceApplyCenteredRmsProp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyFtrlV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyFtrlV2.pbtxt index 8209fd607e1..db4e93ed80e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyFtrlV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyFtrlV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyFtrlV2" endpoint { name: "train.ResourceApplyFtrl" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyGradientDescent.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyGradientDescent.pbtxt index a54fed14d17..48a55a96cc1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyGradientDescent.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyGradientDescent.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyGradientDescent" endpoint { name: "train.ResourceApplyGradientDescent" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyKerasMomentum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyKerasMomentum.pbtxt index 2d76993dc47..35b88fc8869 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyKerasMomentum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyKerasMomentum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyKerasMomentum" endpoint { name: "train.ResourceApplyKerasMomentum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyMomentum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyMomentum.pbtxt index debb0a8131e..ea88f416f0f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyMomentum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyMomentum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyMomentum" endpoint { name: "train.ResourceApplyMomentum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyPowerSign.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyPowerSign.pbtxt index 96df22c81f0..c2a67f1fee3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyPowerSign.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyPowerSign.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyPowerSign" endpoint { name: "train.ResourceApplyPowerSign" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyProximalAdagrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyProximalAdagrad.pbtxt index 809b8b3af3b..c022658a317 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyProximalAdagrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyProximalAdagrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyProximalAdagrad" endpoint { name: "train.ResourceApplyProximalAdagrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyProximalGradientDescent.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyProximalGradientDescent.pbtxt index c9ff5a499d5..a209ab6a065 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyProximalGradientDescent.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyProximalGradientDescent.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyProximalGradientDescent" endpoint { name: "train.ResourceApplyProximalGradientDescent" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyRMSProp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyRMSProp.pbtxt index fa3adf759e0..7e5a287fbdf 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyRMSProp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceApplyRMSProp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceApplyRMSProp" endpoint { name: "train.ResourceApplyRmsProp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceConditionalAccumulator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceConditionalAccumulator.pbtxt index 9289bdefc55..9b23eb1891c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceConditionalAccumulator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceConditionalAccumulator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceConditionalAccumulator" endpoint { name: "train.ResourceConditionalAccumulator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceCountUpTo.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceCountUpTo.pbtxt index 439c1f17557..4c1309f160d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceCountUpTo.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceCountUpTo.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceCountUpTo" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceGather.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceGather.pbtxt index 79c6e8abbcb..a9b829ebd04 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceGather.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceGather.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceGather" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceGatherNd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceGatherNd.pbtxt index 339a7234c3f..ec282febc2b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceGatherNd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceGatherNd.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceGatherNd" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterAdd.pbtxt index e4184e33bf0..33b6d9c67d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterAdd.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterAdd" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterDiv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterDiv.pbtxt index 3e21c24a588..b32181fde11 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterDiv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterDiv.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterDiv" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMax.pbtxt index d25b14272d6..e758222d6ed 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMax.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterMax" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMin.pbtxt index 6243cc1ae3e..bce335396b4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMin.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterMin" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMul.pbtxt index 393e5556c0b..4740ed6669c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterMul.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterMul" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdAdd.pbtxt index 2fd38f7be87..29e9541aac8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdAdd.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterNdAdd" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdMax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdMax.pbtxt index b042ac6c22b..2b2382e88b7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdMax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdMax.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterNdMax" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdMin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdMin.pbtxt index c3c617112b2..bad7c7741b5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdMin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdMin.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterNdMin" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdSub.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdSub.pbtxt index 7733debc1f9..5dad023a56a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdSub.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdSub.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterNdSub" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdUpdate.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdUpdate.pbtxt index 732de5f1cc2..72d079bef41 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdUpdate.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterNdUpdate.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterNdUpdate" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterSub.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterSub.pbtxt index 77081dda4d5..ca9e5fa6a25 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterSub.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterSub.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterSub" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterUpdate.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterUpdate.pbtxt index 9c2cc0ec210..bd850c7bcd2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterUpdate.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceScatterUpdate.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceScatterUpdate" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdadelta.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdadelta.pbtxt index c6e3ae2219c..7614ee61566 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdadelta.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdadelta.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceSparseApplyAdadelta" endpoint { name: "train.ResourceSparseApplyAdadelta" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagrad.pbtxt index 5be4d6199b5..3acd27409f1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceSparseApplyAdagrad" endpoint { name: "train.ResourceSparseApplyAdagrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagradDA.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagradDA.pbtxt index 0547687d640..dff8e161d04 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagradDA.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagradDA.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceSparseApplyAdagradDA" endpoint { name: "train.ResourceSparseApplyAdagradDa" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagradV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagradV2.pbtxt index 65d6ac18ecb..f86922242c7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagradV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyAdagradV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceSparseApplyAdagradV2" endpoint { name: "train.ResourceSparseApplyAdagradV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyCenteredRMSProp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyCenteredRMSProp.pbtxt index 632b0ab4c20..0f402d6bb96 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyCenteredRMSProp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyCenteredRMSProp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceSparseApplyCenteredRMSProp" endpoint { name: "train.ResourceSparseApplyCenteredRmsProp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyFtrlV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyFtrlV2.pbtxt index cd126d78ab6..553da2bcb6f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyFtrlV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyFtrlV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceSparseApplyFtrlV2" endpoint { name: "train.ResourceSparseApplyFtrl" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyKerasMomentum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyKerasMomentum.pbtxt index 78ba4775304..8c39775ba83 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyKerasMomentum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyKerasMomentum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceSparseApplyKerasMomentum" endpoint { name: "train.ResourceSparseApplyKerasMomentum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyMomentum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyMomentum.pbtxt index 7e00039e014..d165cf2f94e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyMomentum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyMomentum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceSparseApplyMomentum" endpoint { name: "train.ResourceSparseApplyMomentum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyProximalAdagrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyProximalAdagrad.pbtxt index 04fe8504e5d..a97d3c5d608 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyProximalAdagrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyProximalAdagrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceSparseApplyProximalAdagrad" endpoint { name: "train.ResourceSparseApplyProximalAdagrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyProximalGradientDescent.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyProximalGradientDescent.pbtxt index 27df43c9c50..69db57fbc14 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyProximalGradientDescent.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyProximalGradientDescent.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceSparseApplyProximalGradientDescent" endpoint { name: "train.ResourceSparseApplyProximalGradientDescent" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyRMSProp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyRMSProp.pbtxt index ec8910a88a7..3cac8411190 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyRMSProp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceSparseApplyRMSProp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ResourceSparseApplyRMSProp" endpoint { name: "train.ResourceSparseApplyRmsProp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceStridedSliceAssign.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceStridedSliceAssign.pbtxt index 83805389b98..bf142658402 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceStridedSliceAssign.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ResourceStridedSliceAssign.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ResourceStridedSliceAssign" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RestoreSlice.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RestoreSlice.pbtxt index 0728f5908b2..d49abdc2abf 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RestoreSlice.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RestoreSlice.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RestoreSlice" endpoint { name: "train.RestoreSlice" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RestoreV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RestoreV2.pbtxt index 909968873f2..f73221177e7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RestoreV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RestoreV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RestoreV2" endpoint { name: "train.Restore" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveAllTPUEmbeddingParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveAllTPUEmbeddingParameters.pbtxt index d0c6b42fafd..0918a4bbd71 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveAllTPUEmbeddingParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveAllTPUEmbeddingParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveAllTPUEmbeddingParameters" endpoint { name: "tpu.RetrieveAllTPUEmbeddingParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingADAMParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingADAMParameters.pbtxt index 4ba2dcfd019..6fa45ac4709 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingADAMParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingADAMParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingADAMParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingADAMParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingADAMParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingADAMParametersGradAccumDebug.pbtxt index c185287ab80..19024de237a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingADAMParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingADAMParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingADAMParametersGradAccumDebug" endpoint { name: "tpu.RetrieveTPUEmbeddingADAMParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdadeltaParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdadeltaParameters.pbtxt index b520470d94a..608071b458b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdadeltaParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdadeltaParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingAdadeltaParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingAdadeltaParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.pbtxt index 3e4226d1e29..ce7f843c0d3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug" endpoint { name: "tpu.RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradMomentumParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradMomentumParameters.pbtxt index 506117ac2b7..c086360d4fb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradMomentumParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradMomentumParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingAdagradMomentumParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingAdagradMomentumParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradParameters.pbtxt index 8260e0b2cc2..2829ab63f30 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingAdagradParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingAdagradParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.pbtxt index 5c0d7d42f0e..08a26da1fc2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingAdagradParametersGradAccumDebug" endpoint { name: "tpu.RetrieveTPUEmbeddingAdagradParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingCenteredRMSPropParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingCenteredRMSPropParameters.pbtxt index 722fde799ae..b339631e163 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingCenteredRMSPropParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingCenteredRMSPropParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingCenteredRMSPropParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingCenteredRMSPropParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFTRLParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFTRLParameters.pbtxt index 8d0c6ee79da..de9f9931616 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFTRLParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFTRLParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingFTRLParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingFTRLParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.pbtxt index b5ce64d483d..57b3e0e2e28 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingFTRLParametersGradAccumDebug" endpoint { name: "tpu.RetrieveTPUEmbeddingFTRLParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFrequencyEstimatorParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFrequencyEstimatorParameters.pbtxt index 02da67a33d4..a30b2e979d9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFrequencyEstimatorParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFrequencyEstimatorParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingFrequencyEstimatorParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingFrequencyEstimatorParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.pbtxt index 5f54c810a4f..eff5462872f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug" endpoint { name: "tpu.RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMDLAdagradLightParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMDLAdagradLightParameters.pbtxt index 040018699c7..c4320af9050 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMDLAdagradLightParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMDLAdagradLightParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingMDLAdagradLightParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingMDLAdagradLightParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMomentumParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMomentumParameters.pbtxt index 017d38ce98c..cae7612c2b2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMomentumParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMomentumParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingMomentumParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingMomentumParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.pbtxt index fdcd930b1c9..c3d1eea0d1d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingMomentumParametersGradAccumDebug" endpoint { name: "tpu.RetrieveTPUEmbeddingMomentumParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalAdagradParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalAdagradParameters.pbtxt index 7e88ab316d8..a6a7b7d8582 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalAdagradParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalAdagradParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingProximalAdagradParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingProximalAdagradParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.pbtxt index 16b7e25975d..8f0cba646fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug" endpoint { name: "tpu.RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalYogiParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalYogiParameters.pbtxt index 618d0cda5e9..3e516888ec3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalYogiParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalYogiParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingProximalYogiParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingProximalYogiParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.pbtxt index a04bed75d87..26a810e8794 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug" endpoint { name: "tpu.RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingRMSPropParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingRMSPropParameters.pbtxt index cb56f096e8b..03b991ee6b4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingRMSPropParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingRMSPropParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingRMSPropParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingRMSPropParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.pbtxt index 5eb9017d4d2..2a873e27fc3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug" endpoint { name: "tpu.RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingStochasticGradientDescentParameters.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingStochasticGradientDescentParameters.pbtxt index 755ffdd69e5..a0377103d49 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingStochasticGradientDescentParameters.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingStochasticGradientDescentParameters.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingStochasticGradientDescentParameters" endpoint { name: "tpu.RetrieveTPUEmbeddingStochasticGradientDescentParameters" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.pbtxt index 6d476f70bd5..71758e43589 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug" endpoint { name: "tpu.RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReverseSequence.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReverseSequence.pbtxt index 87638c0dcc9..f0e6bd4a1cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReverseSequence.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReverseSequence.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ReverseSequence" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReverseV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReverseV2.pbtxt index 71efbe1892e..c286316354f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ReverseV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ReverseV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ReverseV2" endpoint { name: "Reverse" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RewriteDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RewriteDataset.pbtxt index fc093decb92..cd73223372b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RewriteDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RewriteDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RewriteDataset" endpoint { name: "data.RewriteDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RightShift.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RightShift.pbtxt index 68fab3e8cf8..8f6889fd4d5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RightShift.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RightShift.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RightShift" endpoint { name: "bitwise.RightShift" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Rint.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Rint.pbtxt index 48fbcc7c346..0bf2aa48f28 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Rint.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Rint.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Rint" endpoint { name: "math.Rint" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscAbs.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscAbs.pbtxt index c2ab94f053b..16a02df71a8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscAbs.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscAbs.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscAbs" endpoint { name: "risc.RiscAbs" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscAdd.pbtxt index 5694b59c62f..db1cafd86b1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscAdd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscAdd" endpoint { name: "risc.RiscAdd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBinaryArithmetic.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBinaryArithmetic.pbtxt index 910399fa401..a6b0d3849d9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBinaryArithmetic.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBinaryArithmetic.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscBinaryArithmetic" endpoint { name: "risc.RiscBinaryArithmetic" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBinaryComparison.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBinaryComparison.pbtxt index 014e43b1444..b278cdb19b5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBinaryComparison.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBinaryComparison.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscBinaryComparison" endpoint { name: "risc.RiscBinaryComparison" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBitcast.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBitcast.pbtxt index 3393f70a8b5..3576ea43316 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBitcast.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBitcast.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscBitcast" endpoint { name: "risc.RiscBitcast" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBroadcast.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBroadcast.pbtxt index 755892ca968..70f651c5595 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBroadcast.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscBroadcast.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscBroadcast" endpoint { name: "risc.RiscBroadcast" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCast.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCast.pbtxt index d1bffc26bff..03d2dddb2a7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCast.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCast.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscCast" endpoint { name: "risc.RiscCast" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCeil.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCeil.pbtxt index 286b8298d51..7cc1796e649 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCeil.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCeil.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscCeil" endpoint { name: "risc.RiscCeil" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCholesky.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCholesky.pbtxt index cdb5975e035..f58e0969b02 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCholesky.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCholesky.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscCholesky" endpoint { name: "risc.RiscCholesky" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscConcat.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscConcat.pbtxt index 670cb46be04..e5aad1d665c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscConcat.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscConcat.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscConcat" endpoint { name: "risc.RiscConcat" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCondition.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCondition.pbtxt index 2284aeed689..20b4043192e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCondition.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCondition.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscCondition" endpoint { name: "risc.RiscCondition" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscConv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscConv.pbtxt index 4e2342a8da9..3b85466aad8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscConv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscConv.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscConv" endpoint { name: "risc.RiscConv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCos.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCos.pbtxt index d9905d7e1b0..bd0bd4faa20 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCos.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscCos.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscCos" endpoint { name: "risc.RiscCos" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscDiv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscDiv.pbtxt index 651d569b479..62752229c2b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscDiv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscDiv.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscDiv" endpoint { name: "risc.RiscDiv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscDot.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscDot.pbtxt index 4eac65da4f8..884d0093f49 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscDot.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscDot.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscDot" endpoint { name: "risc.RiscDot" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscExp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscExp.pbtxt index 35bb77b83c6..a0f735e9812 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscExp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscExp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscExp" endpoint { name: "risc.RiscExp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscFft.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscFft.pbtxt index a3dcbe69337..7939ade66d4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscFft.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscFft.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscFft" endpoint { name: "risc.RiscFft" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscFloor.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscFloor.pbtxt index 9f5d762d1a4..4bbf58f30be 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscFloor.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscFloor.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscFloor" endpoint { name: "risc.RiscFloor" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscGather.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscGather.pbtxt index c4fe724889d..65e03eabd05 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscGather.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscGather.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscGather" endpoint { name: "risc.RiscGather" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscImag.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscImag.pbtxt index 70d8136856b..c8473b54de1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscImag.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscImag.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscImag" endpoint { name: "risc.RiscImag" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscIsFinite.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscIsFinite.pbtxt index 5418f7a9906..9155259eba0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscIsFinite.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscIsFinite.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscIsFinite" endpoint { name: "risc.RiscIsFinite" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLog.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLog.pbtxt index b0bb8f3aaed..c8e1afe2a75 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLog.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLog.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscLog" endpoint { name: "risc.RiscLog" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalAnd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalAnd.pbtxt index 1ccb0264901..bc2d5b1f9eb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalAnd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalAnd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscLogicalAnd" endpoint { name: "risc.RiscLogicalAnd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalNot.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalNot.pbtxt index 6f97af1c7b6..c4743d410b3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalNot.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalNot.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscLogicalNot" endpoint { name: "risc.RiscLogicalNot" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalOr.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalOr.pbtxt index 97e37710419..f23f059b514 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalOr.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscLogicalOr.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscLogicalOr" endpoint { name: "risc.RiscLogicalOr" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMax.pbtxt index 240f8119a9e..06d25cbc86a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscMax" endpoint { name: "risc.RiscMax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMin.pbtxt index a8ccba66ae1..309d515d6fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMin.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscMin" endpoint { name: "risc.RiscMin" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMul.pbtxt index 21fc1e0e336..51927d3a135 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscMul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscMul" endpoint { name: "risc.RiscMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscNeg.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscNeg.pbtxt index 894b769a72a..0e0dd0ea4b0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscNeg.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscNeg.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscNeg" endpoint { name: "risc.RiscNeg" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPad.pbtxt index 729bba07740..0e3d478d02b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscPad" endpoint { name: "risc.RiscPad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPool.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPool.pbtxt index 9ed6a55dd07..74cd28a15fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPool.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPool.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscPool" endpoint { name: "risc.RiscPool" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPow.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPow.pbtxt index 3eac196376f..2565bd11555 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPow.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscPow.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscPow" endpoint { name: "risc.RiscPow" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscRandomUniform.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscRandomUniform.pbtxt index ef96f0a2796..942c4bec622 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscRandomUniform.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscRandomUniform.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscRandomUniform" endpoint { name: "risc.RiscRandomUniform" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReal.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReal.pbtxt index 5b9691512fc..5d24d2ad837 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReal.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReal.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscReal" endpoint { name: "risc.RiscReal" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReduce.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReduce.pbtxt index d5d614c828e..bc9b20496e5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReduce.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReduce.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscReduce" endpoint { name: "risc.RiscReduce" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscRem.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscRem.pbtxt index 0bb38f8de55..22de8c713ea 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscRem.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscRem.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscRem" endpoint { name: "risc.RiscRem" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReshape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReshape.pbtxt index b2ab27447a3..fd3bacbd2d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReshape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReshape.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscReshape" endpoint { name: "risc.RiscReshape" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReverse.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReverse.pbtxt index ccb027a8859..ee8e646e4b9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReverse.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscReverse.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscReverse" endpoint { name: "risc.RiscReverse" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscScatter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscScatter.pbtxt index 0eea45dcf04..dabe270375d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscScatter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscScatter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscScatter" endpoint { name: "risc.RiscScatter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscShape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscShape.pbtxt index dab7319a922..83666efcdf6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscShape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscShape.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscShape" endpoint { name: "risc.RiscShape" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSign.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSign.pbtxt index a157b69acbb..2cc5dfc378c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSign.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSign.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscSign" endpoint { name: "risc.RiscSign" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSlice.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSlice.pbtxt index fc630149b69..ecb7b991196 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSlice.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSlice.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscSlice" endpoint { name: "risc.RiscSlice" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSort.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSort.pbtxt index 2048ca7aab1..3361401d336 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSort.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSort.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscSort" endpoint { name: "risc.RiscSort" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSqueeze.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSqueeze.pbtxt index f09b55721f9..5b9b50d209e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSqueeze.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSqueeze.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscSqueeze" endpoint { name: "risc.RiscSqueeze" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSub.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSub.pbtxt index 924d3f38189..ea48b182d22 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSub.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscSub.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscSub" endpoint { name: "risc.RiscSub" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscTranspose.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscTranspose.pbtxt index 877ee6d6570..f2d3e739b50 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscTranspose.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscTranspose.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscTranspose" endpoint { name: "risc.RiscTranspose" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscTriangularSolve.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscTriangularSolve.pbtxt index f74b9a88a86..70b4fbdeed4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscTriangularSolve.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscTriangularSolve.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscTriangularSolve" endpoint { name: "risc.RiscTriangularSolve" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscUnary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscUnary.pbtxt index 429c93bff49..d1d03367c05 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscUnary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscUnary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscUnary" endpoint { name: "risc.RiscUnary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscWhile.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscWhile.pbtxt index e4810438b46..745b47cdfab 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscWhile.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RiscWhile.pbtxt @@ -1,4 +1,5 @@ op { + visibility: SKIP graph_op_name: "RiscWhile" endpoint { name: "risc.RiscWhile" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RngReadAndSkip.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RngReadAndSkip.pbtxt index 92057c8e041..8603fa95988 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RngReadAndSkip.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RngReadAndSkip.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RngReadAndSkip" endpoint { name: "random.RngReadAndSkip" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RngSkip.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RngSkip.pbtxt index d9d3c6a1e73..9074f38c5da 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RngSkip.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RngSkip.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RngSkip" endpoint { name: "random.RngSkip" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Roll.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Roll.pbtxt index 50f7915a65a..fe4eed9ab13 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Roll.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Roll.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Roll" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Round.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Round.pbtxt index dd612a33d63..960ffba508f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Round.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Round.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Round" endpoint { name: "math.Round" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Rpc.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Rpc.pbtxt index 0d1e2b90e6a..528afe26709 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Rpc.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Rpc.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Rpc" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Rsqrt.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Rsqrt.pbtxt index 06b1b81ecd4..97165e2d758 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Rsqrt.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Rsqrt.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Rsqrt" endpoint { name: "math.Rsqrt" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_RsqrtGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_RsqrtGrad.pbtxt index 88073b6f254..8aa9f02b9bc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_RsqrtGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_RsqrtGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "RsqrtGrad" endpoint { name: "math.RsqrtGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SampleDistortedBoundingBoxV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SampleDistortedBoundingBoxV2.pbtxt index 65573141378..0aef133b9e6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SampleDistortedBoundingBoxV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SampleDistortedBoundingBoxV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SampleDistortedBoundingBoxV2" endpoint { name: "image.SampleDistortedBoundingBox" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SaveSlices.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SaveSlices.pbtxt index b3360356800..33af2108dd0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SaveSlices.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SaveSlices.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SaveSlices" endpoint { name: "train.SaveSlices" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SaveV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SaveV2.pbtxt index 644d1824aa1..0fc943f3540 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SaveV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SaveV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SaveV2" endpoint { name: "train.Save" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScalarSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScalarSummary.pbtxt index c339ce0a7a5..7b6f6129353 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScalarSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScalarSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ScalarSummary" endpoint { name: "summary.ScalarSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScaleAndTranslate.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScaleAndTranslate.pbtxt index 850f8effbe1..25364907a30 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScaleAndTranslate.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScaleAndTranslate.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ScaleAndTranslate" endpoint { name: "image.ScaleAndTranslate" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScaleAndTranslateGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScaleAndTranslateGrad.pbtxt index 99bc3a92bb7..e3256e0f704 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScaleAndTranslateGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScaleAndTranslateGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ScaleAndTranslateGrad" endpoint { name: "image.ScaleAndTranslateGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterAdd.pbtxt index 41c63dc0a40..74492ab813b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterAdd.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterAdd" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterDiv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterDiv.pbtxt index 5754249eafc..97252d64db3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterDiv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterDiv.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterDiv" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMax.pbtxt index aa6375cbd76..5217cb1f668 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMax.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterMax" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMin.pbtxt index ea007120c36..c082832265c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMin.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterMin" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMul.pbtxt index f1d91258e4b..4d284a527c2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterMul.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterMul" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNd.pbtxt index 8ef01b2fcac..5d5308a7444 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ScatterNd" endpoint { name: "ScatterNd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdAdd.pbtxt index bea152a9da5..61d9acdd48c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdAdd.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterNdAdd" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdMax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdMax.pbtxt index 392f99b3bec..617c639add6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdMax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdMax.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterNdMax" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdMin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdMin.pbtxt index 9ced2ef3ba3..53d6754e0e3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdMin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdMin.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterNdMin" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdNonAliasingAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdNonAliasingAdd.pbtxt index 4dd756bfc0c..98baca56e19 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdNonAliasingAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdNonAliasingAdd.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterNdNonAliasingAdd" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdSub.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdSub.pbtxt index 384e79d64ef..867227b1507 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdSub.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdSub.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterNdSub" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdUpdate.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdUpdate.pbtxt index 92fce7f0ac3..2c4432c9ed0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdUpdate.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterNdUpdate.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterNdUpdate" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterSub.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterSub.pbtxt index 5baaa4f6045..25a2e9519fa 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterSub.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterSub.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterSub" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterUpdate.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterUpdate.pbtxt index 83ac128ed60..cfcff646652 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterUpdate.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ScatterUpdate.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ScatterUpdate" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaFprint.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaFprint.pbtxt index ce179918cd0..19725ee76d2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaFprint.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaFprint.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SdcaFprint" endpoint { name: "train.SdcaFprint" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaOptimizerV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaOptimizerV2.pbtxt index b72ee64e501..b67c06a7069 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaOptimizerV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaOptimizerV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SdcaOptimizerV2" endpoint { name: "train.SdcaOptimizer" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaShrinkL1.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaShrinkL1.pbtxt index 83993bcf149..b65cd2a92c0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaShrinkL1.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SdcaShrinkL1.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SdcaShrinkL1" endpoint { name: "train.SdcaShrinkL1" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMaxV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMaxV2.pbtxt index 7b8d476dc67..58d1cce4a47 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMaxV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMaxV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SegmentMaxV2" endpoint { name: "math.SegmentMax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMean.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMean.pbtxt index 982db87bf09..78a64153e26 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMean.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMean.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SegmentMean" endpoint { name: "math.SegmentMean" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMinV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMinV2.pbtxt index 84cbd30843d..8d3c1ea4bdd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMinV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentMinV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SegmentMinV2" endpoint { name: "math.SegmentMin" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentProdV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentProdV2.pbtxt index 53276c470cc..6ed9d2bf402 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentProdV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentProdV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SegmentProdV2" endpoint { name: "math.SegmentProd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentSumV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentSumV2.pbtxt index aa200316606..0895c391f2b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentSumV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SegmentSumV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SegmentSumV2" endpoint { name: "math.SegmentSum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SelectV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SelectV2.pbtxt index 01721ea1f68..815f4b6ddde 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SelectV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SelectV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SelectV2" endpoint { name: "Select" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SelfAdjointEigV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SelfAdjointEigV2.pbtxt index c79f08ac323..6fa79916a1d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SelfAdjointEigV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SelfAdjointEigV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SelfAdjointEigV2" endpoint { name: "linalg.SelfAdjointEig" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Selu.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Selu.pbtxt index 7002d5be122..f53e34f3107 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Selu.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Selu.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Selu" endpoint { name: "nn.Selu" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SeluGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SeluGrad.pbtxt index b34e2f22393..b66494ca140 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SeluGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SeluGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SeluGrad" endpoint { name: "nn.SeluGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Send.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Send.pbtxt index 01f52451dfd..223838e2998 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Send.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Send.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Send" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SendTPUEmbeddingGradients.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SendTPUEmbeddingGradients.pbtxt index fc4e90d96dc..c51956d45cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SendTPUEmbeddingGradients.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SendTPUEmbeddingGradients.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SendTPUEmbeddingGradients" endpoint { name: "tpu.SendTPUEmbeddingGradients" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeIterator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeIterator.pbtxt index 37789c753b4..f2184edf40a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeIterator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeIterator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SerializeIterator" endpoint { name: "data.SerializeIterator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeManySparse.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeManySparse.pbtxt index 567a8e4b7f0..b49092d898f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeManySparse.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeManySparse.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SerializeManySparse" endpoint { name: "io.SerializeManySparse" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeSparse.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeSparse.pbtxt index ad86c7095c3..6182dd1c4fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeSparse.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeSparse.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SerializeSparse" endpoint { name: "io.SerializeSparse" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeTensor.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeTensor.pbtxt index 169120a0635..63b220fe4a9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeTensor.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SerializeTensor.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SerializeTensor" endpoint { name: "io.SerializeTensor" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SetSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SetSize.pbtxt index 78c43275762..c7c0b9784e4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SetSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SetSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SetSize" out_arg { name: "size" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Shape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Shape.pbtxt index 5074000b537..5c14c619e12 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Shape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Shape.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Shape" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ShapeN.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ShapeN.pbtxt index b4bfb91118a..0bae14f919a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ShapeN.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ShapeN.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ShapeN" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ShardedFilename.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ShardedFilename.pbtxt index 8efd0afb8b3..1163fc341c0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ShardedFilename.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ShardedFilename.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ShardedFilename" endpoint { name: "io.ShardedFilename" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ShardedFilespec.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ShardedFilespec.pbtxt index e31cac1040c..521e1abbe26 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ShardedFilespec.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ShardedFilespec.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ShardedFilespec" endpoint { name: "io.ShardedFilespec" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ShutdownDistributedTPU.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ShutdownDistributedTPU.pbtxt index bb34a733ac2..b5a98673065 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ShutdownDistributedTPU.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ShutdownDistributedTPU.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ShutdownDistributedTPU" endpoint { name: "tpu.ShutdownDistributedTPU" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ShutdownTPUSystem.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ShutdownTPUSystem.pbtxt index 8c0707b60f5..6e3aa7fe8f3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ShutdownTPUSystem.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ShutdownTPUSystem.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ShutdownTPUSystem" endpoint { name: "tpu.ShutdownTPUSystem" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sigmoid.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sigmoid.pbtxt index b6e16a41104..d96d55bd1a4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sigmoid.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sigmoid.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Sigmoid" endpoint { name: "math.Sigmoid" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SigmoidGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SigmoidGrad.pbtxt index bb141abe3ba..b22b87eb6ee 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SigmoidGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SigmoidGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SigmoidGrad" endpoint { name: "math.SigmoidGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sign.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sign.pbtxt index 435fb9e825d..c94099d9d0a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sign.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sign.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Sign" endpoint { name: "math.Sign" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sin.pbtxt index 2fc95755bac..b28cb476fdf 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sin.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Sin" endpoint { name: "math.Sin" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sinh.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sinh.pbtxt index f343685e80b..6eb717431b8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sinh.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sinh.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Sinh" endpoint { name: "math.Sinh" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Size.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Size.pbtxt index a41cddd8ac7..4ff9bd964d4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Size.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Size.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Size" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Skipgram.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Skipgram.pbtxt index d095c7b61b9..2bd4b0592b1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Skipgram.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Skipgram.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Skipgram" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Slice.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Slice.pbtxt index adfe6fa4fd9..0440df64566 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Slice.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Slice.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Slice" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Snapshot.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Snapshot.pbtxt index 6e49c1a5431..2c1746d189d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Snapshot.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Snapshot.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Snapshot" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotChunkDataset.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotChunkDataset.pbtxt index 4d3b33c84b8..e851579d2e8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotChunkDataset.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotChunkDataset.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SnapshotChunkDataset" endpoint { name: "data.SnapshotChunkDataset" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotDatasetReader.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotDatasetReader.pbtxt index 36a1a08fee1..82de05ad6b3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotDatasetReader.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotDatasetReader.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SnapshotDatasetReader" endpoint { name: "data.SnapshotDatasetReader" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotNestedDatasetReader.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotNestedDatasetReader.pbtxt index 8cd7d40eb47..ca3d1ca3fc5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotNestedDatasetReader.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SnapshotNestedDatasetReader.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SnapshotNestedDatasetReader" endpoint { name: "data.SnapshotNestedDatasetReader" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SobolSample.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SobolSample.pbtxt index 4a3daebb2b8..f6a7390ccf2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SobolSample.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SobolSample.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SobolSample" endpoint { name: "math.SobolSample" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Softmax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Softmax.pbtxt index cb27a04d217..0c8e7080f4b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Softmax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Softmax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Softmax" endpoint { name: "nn.Softmax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftmaxCrossEntropyWithLogits.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftmaxCrossEntropyWithLogits.pbtxt index e064562c0f2..b6e28d83c31 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftmaxCrossEntropyWithLogits.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftmaxCrossEntropyWithLogits.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SoftmaxCrossEntropyWithLogits" endpoint { name: "nn.SoftmaxCrossEntropyWithLogits" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Softplus.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Softplus.pbtxt index 671656c2873..6f24c635b0f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Softplus.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Softplus.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Softplus" endpoint { name: "math.Softplus" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftplusGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftplusGrad.pbtxt index d1b074a9b7a..fe4b790d539 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftplusGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftplusGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SoftplusGrad" endpoint { name: "math.SoftplusGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Softsign.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Softsign.pbtxt index 23aa1e3c58b..13f1a13bc77 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Softsign.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Softsign.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Softsign" endpoint { name: "nn.Softsign" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftsignGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftsignGrad.pbtxt index 73faf74511f..42dc05ceb71 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftsignGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SoftsignGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SoftsignGrad" endpoint { name: "nn.SoftsignGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToBatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToBatch.pbtxt index 4fe7232e544..92ee4f45f77 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToBatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToBatch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SpaceToBatch" endpoint { name: "nn.SpaceToBatch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToBatchND.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToBatchND.pbtxt index 6612b48286c..febe2f4b5ab 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToBatchND.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToBatchND.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SpaceToBatchND" endpoint { name: "SpaceToBatchNd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToDepth.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToDepth.pbtxt index cb421c75db0..980fade691f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToDepth.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SpaceToDepth.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SpaceToDepth" endpoint { name: "nn.SpaceToDepth" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAccumulatorApplyGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAccumulatorApplyGradient.pbtxt index cdb0b14b671..c631827feac 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAccumulatorApplyGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAccumulatorApplyGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseAccumulatorApplyGradient" endpoint { name: "sparse.SparseAccumulatorApplyGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAccumulatorTakeGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAccumulatorTakeGradient.pbtxt index 8fc1e709596..563a0a0b3ea 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAccumulatorTakeGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAccumulatorTakeGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseAccumulatorTakeGradient" endpoint { name: "sparse.SparseAccumulatorTakeGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAdd.pbtxt index 0761f2ed162..ff7a9529137 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAdd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseAdd" endpoint { name: "sparse.SparseAdd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAddGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAddGrad.pbtxt index 6529c46a17e..24ac26d72d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAddGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseAddGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseAddGrad" endpoint { name: "sparse.SparseAddGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdadelta.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdadelta.pbtxt index 7122f210a43..f98e86d163d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdadelta.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdadelta.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseApplyAdadelta" endpoint { name: "train.SparseApplyAdadelta" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdagradDA.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdagradDA.pbtxt index 950dc00dd35..dd565aea78e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdagradDA.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdagradDA.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseApplyAdagradDA" endpoint { name: "train.SparseApplyAdagradDa" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdagradV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdagradV2.pbtxt index 4b959f79648..5f5fed2f326 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdagradV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyAdagradV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseApplyAdagradV2" endpoint { name: "train.SparseApplyAdagrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyCenteredRMSProp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyCenteredRMSProp.pbtxt index 090536f5eb9..040e66f749a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyCenteredRMSProp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyCenteredRMSProp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseApplyCenteredRMSProp" endpoint { name: "train.SparseApplyCenteredRmsProp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyFtrlV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyFtrlV2.pbtxt index 43b9833451a..d4fcd069cc4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyFtrlV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyFtrlV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseApplyFtrlV2" endpoint { name: "train.SparseApplyFtrl" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyMomentum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyMomentum.pbtxt index f7e79c5e7e4..649e41093eb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyMomentum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyMomentum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseApplyMomentum" endpoint { name: "train.SparseApplyMomentum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyProximalAdagrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyProximalAdagrad.pbtxt index 8ac6cf771cb..4327f9c91b7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyProximalAdagrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyProximalAdagrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseApplyProximalAdagrad" endpoint { name: "train.SparseApplyProximalAdagrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyProximalGradientDescent.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyProximalGradientDescent.pbtxt index 926ed2c1d4d..5df170d2046 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyProximalGradientDescent.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyProximalGradientDescent.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseApplyProximalGradientDescent" endpoint { name: "train.SparseApplyProximalGradientDescent" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyRMSProp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyRMSProp.pbtxt index 3e39f4ffa58..b40e17fd759 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyRMSProp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseApplyRMSProp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseApplyRMSProp" endpoint { name: "train.SparseApplyRmsProp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseBincount.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseBincount.pbtxt index 422904962de..b51ce0c901e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseBincount.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseBincount.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseBincount" endpoint { name: "sparse.SparseBincount" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseConcat.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseConcat.pbtxt index 8ceb600a42c..cfaa82f05bc 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseConcat.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseConcat.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseConcat" endpoint { name: "sparse.SparseConcat" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseConditionalAccumulator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseConditionalAccumulator.pbtxt index 3dc2c1ea8a0..ae7b32e6363 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseConditionalAccumulator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseConditionalAccumulator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseConditionalAccumulator" endpoint { name: "sparse.SparseConditionalAccumulator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCountSparseOutput.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCountSparseOutput.pbtxt index 93b11727dca..6f2f6ca4b67 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCountSparseOutput.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCountSparseOutput.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseCountSparseOutput" endpoint { name: "sparse.SparseCountSparseOutput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCrossHashed.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCrossHashed.pbtxt index 7c2f41b6f68..9d91c960992 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCrossHashed.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCrossHashed.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseCrossHashed" endpoint { name: "sparse.SparseCrossHashed" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCrossV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCrossV2.pbtxt index 477bc3aff25..a34de03d384 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCrossV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseCrossV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseCrossV2" endpoint { name: "sparse.SparseCross" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseAdd.pbtxt index 013b7eede94..73291dc92ed 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseAdd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseDenseCwiseAdd" endpoint { name: "sparse.SparseDenseCwiseAdd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseDiv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseDiv.pbtxt index 8cf56d7b418..02c3e96fe97 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseDiv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseDiv.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseDenseCwiseDiv" endpoint { name: "sparse.SparseDenseCwiseDiv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseMul.pbtxt index 37c50f23276..3f2db8ac9cd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseDenseCwiseMul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseDenseCwiseMul" endpoint { name: "sparse.SparseDenseCwiseMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseFillEmptyRows.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseFillEmptyRows.pbtxt index 3adddbd34bf..30c8781f95e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseFillEmptyRows.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseFillEmptyRows.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseFillEmptyRows" endpoint { name: "sparse.SparseFillEmptyRows" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseFillEmptyRowsGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseFillEmptyRowsGrad.pbtxt index 708069d028d..599fd18d639 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseFillEmptyRowsGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseFillEmptyRowsGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseFillEmptyRowsGrad" endpoint { name: "sparse.SparseFillEmptyRowsGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatMul.pbtxt index 445d53b0236..e860faa19c5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatMul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatMul" endpoint { name: "sparse.SparseMatMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixAdd.pbtxt index 6f6b6443e85..b01ba2badcb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixAdd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatrixAdd" endpoint { name: "linalg.sparse.SparseMatrixAdd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixMatMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixMatMul.pbtxt index 47dddcb4bce..9d09c711dd3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixMatMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixMatMul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatrixMatMul" endpoint { name: "linalg.sparse.SparseMatrixMatMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixMul.pbtxt index 16bf766d1bf..309c305b811 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixMul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatrixMul" endpoint { name: "linalg.sparse.SparseMatrixMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixNNZ.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixNNZ.pbtxt index 1a9adbd9f4d..b5139613595 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixNNZ.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixNNZ.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatrixNNZ" endpoint { name: "linalg.sparse.SparseMatrixNNZ" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixOrderingAMD.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixOrderingAMD.pbtxt index 7b15db6fe32..955993d0ca9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixOrderingAMD.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixOrderingAMD.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatrixOrderingAMD" endpoint { name: "linalg.sparse.SparseMatrixOrderingAMD" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSoftmax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSoftmax.pbtxt index 85418910d8e..38f3eee2100 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSoftmax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSoftmax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatrixSoftmax" endpoint { name: "linalg.sparse.SparseMatrixSoftmax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSoftmaxGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSoftmaxGrad.pbtxt index 79f71005c27..235a82ecba9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSoftmaxGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSoftmaxGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatrixSoftmaxGrad" endpoint { name: "linalg.sparse.SparseMatrixSoftmaxGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSparseCholesky.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSparseCholesky.pbtxt index 6913a9f26bd..3da4a8c1085 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSparseCholesky.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSparseCholesky.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatrixSparseCholesky" endpoint { name: "linalg.sparse.SparseMatrixSparseCholesky" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSparseMatMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSparseMatMul.pbtxt index d35bdf49e25..2cc059cfe04 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSparseMatMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixSparseMatMul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatrixSparseMatMul" endpoint { name: "linalg.sparse.SparseMatrixSparseMatMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixTranspose.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixTranspose.pbtxt index c5a7425febd..32e7154bb5b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixTranspose.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixTranspose.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatrixTranspose" endpoint { name: "linalg.sparse.SparseMatrixTranspose" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixZeros.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixZeros.pbtxt index 37b5fc48ecf..17db640c5dd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixZeros.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseMatrixZeros.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseMatrixZeros" endpoint { name: "linalg.sparse.SparseMatrixZeros" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceMax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceMax.pbtxt index a7467b9b474..e6791f6a5a9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceMax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceMax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseReduceMax" endpoint { name: "sparse.SparseReduceMax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceMaxSparse.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceMaxSparse.pbtxt index 987e98467ed..28916074692 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceMaxSparse.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceMaxSparse.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseReduceMaxSparse" endpoint { name: "sparse.SparseReduceMaxSparse" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceSum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceSum.pbtxt index 739fb5fb952..f26df04d991 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceSum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceSum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseReduceSum" endpoint { name: "sparse.SparseReduceSum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceSumSparse.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceSumSparse.pbtxt index 64d6d45f1a2..466d862a573 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceSumSparse.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReduceSumSparse.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseReduceSumSparse" endpoint { name: "sparse.SparseReduceSumSparse" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReorder.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReorder.pbtxt index 202066e76f5..7698a5ad576 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReorder.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReorder.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseReorder" endpoint { name: "sparse.SparseReorder" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReshape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReshape.pbtxt index 0a393a6105f..50979bbc49f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReshape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseReshape.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseReshape" endpoint { name: "sparse.SparseReshape" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMean.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMean.pbtxt index 9187dbed670..b0d0394bb93 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMean.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMean.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSegmentMean" endpoint { name: "sparse.SparseSegmentMean" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMeanGradV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMeanGradV2.pbtxt index e80adf6c81a..f2cae7f512c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMeanGradV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMeanGradV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSegmentMeanGradV2" endpoint { name: "sparse.SparseSegmentMeanGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMeanWithNumSegments.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMeanWithNumSegments.pbtxt index bb3ac2256d1..94edd7dbfb0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMeanWithNumSegments.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentMeanWithNumSegments.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSegmentMeanWithNumSegments" endpoint { name: "sparse.SparseSegmentMeanWithNumSegments" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtN.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtN.pbtxt index 2cecff503ff..70808796ccd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtN.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtN.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSegmentSqrtN" endpoint { name: "sparse.SparseSegmentSqrtN" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtNGradV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtNGradV2.pbtxt index a8b577eca01..6984f2b771d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtNGradV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtNGradV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSegmentSqrtNGradV2" endpoint { name: "sparse.SparseSegmentSqrtNGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtNWithNumSegments.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtNWithNumSegments.pbtxt index 78aed85f0a9..8e26330994e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtNWithNumSegments.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSqrtNWithNumSegments.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSegmentSqrtNWithNumSegments" endpoint { name: "sparse.SparseSegmentSqrtNWithNumSegments" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSum.pbtxt index f18f3516670..fa46eeaf785 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSegmentSum" endpoint { name: "sparse.SparseSegmentSum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSumGradV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSumGradV2.pbtxt index 333163ac6a3..90956748160 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSumGradV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSumGradV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSegmentSumGradV2" endpoint { name: "sparse.SparseSegmentSumGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSumWithNumSegments.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSumWithNumSegments.pbtxt index 8e384857cc7..3e9b7c8ec2c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSumWithNumSegments.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSegmentSumWithNumSegments.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSegmentSumWithNumSegments" endpoint { name: "sparse.SparseSegmentSumWithNumSegments" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSlice.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSlice.pbtxt index 1d794df9252..aff3e800d7c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSlice.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSlice.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSlice" endpoint { name: "sparse.SparseSlice" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSliceGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSliceGrad.pbtxt index 979326c0fc7..30e7c6d9ab5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSliceGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSliceGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSliceGrad" endpoint { name: "sparse.SparseSliceGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSoftmax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSoftmax.pbtxt index 56c96640cb4..f36e6e886c1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSoftmax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSoftmax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSoftmax" endpoint { name: "sparse.SparseSoftmax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSoftmaxCrossEntropyWithLogits.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSoftmaxCrossEntropyWithLogits.pbtxt index 7627d5f6074..145a804d5c0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSoftmaxCrossEntropyWithLogits.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSoftmaxCrossEntropyWithLogits.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSoftmaxCrossEntropyWithLogits" endpoint { name: "nn.SparseSoftmaxCrossEntropyWithLogits" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSparseMaximum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSparseMaximum.pbtxt index 31430962138..aa09d6e07ea 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSparseMaximum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSparseMaximum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSparseMaximum" endpoint { name: "sparse.SparseSparseMaximum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSparseMinimum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSparseMinimum.pbtxt index fc04bb4fed5..519822e9652 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSparseMinimum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSparseMinimum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSparseMinimum" endpoint { name: "sparse.SparseSparseMinimum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSplit.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSplit.pbtxt index 0afc95199aa..d2145cc62bd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSplit.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseSplit.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseSplit" endpoint { name: "sparse.SparseSplit" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorDenseAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorDenseAdd.pbtxt index fc5c882f834..26b82985d68 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorDenseAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorDenseAdd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseTensorDenseAdd" endpoint { name: "sparse.SparseTensorDenseAdd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorDenseMatMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorDenseMatMul.pbtxt index cd1f1e09e94..49bc58b9216 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorDenseMatMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorDenseMatMul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseTensorDenseMatMul" endpoint { name: "sparse.SparseTensorDenseMatMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorToCSRSparseMatrix.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorToCSRSparseMatrix.pbtxt index cbbe0cc0e7a..b0805568811 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorToCSRSparseMatrix.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseTensorToCSRSparseMatrix.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseTensorToCSRSparseMatrix" endpoint { name: "linalg.sparse.SparseTensorToCSRSparseMatrix" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseToDense.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseToDense.pbtxt index 68df155e461..7d5755622d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseToDense.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseToDense.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseToDense" endpoint { name: "sparse.SparseToDense" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseToSparseSetOperation.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseToSparseSetOperation.pbtxt index fb04366fead..31ca110cf54 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseToSparseSetOperation.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SparseToSparseSetOperation.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SparseToSparseSetOperation" endpoint { name: "sparse.SparseToSparseSetOperation" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Spence.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Spence.pbtxt index f03a6820aae..3d5de7d65c1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Spence.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Spence.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Spence" endpoint { name: "math.special.Spence" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Split.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Split.pbtxt index ffb7b52e091..802055bce54 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Split.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Split.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Split" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SplitDedupData.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SplitDedupData.pbtxt index 7597facf9ce..7870a4dc4b0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SplitDedupData.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SplitDedupData.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SplitDedupData" endpoint { name: "tpu.SplitDedupData" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SplitV.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SplitV.pbtxt index 94f4a08d70d..c9b3de491bd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SplitV.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SplitV.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "SplitV" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sqrt.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sqrt.pbtxt index daa2a1ee86a..d532c02fdfa 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sqrt.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sqrt.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Sqrt" endpoint { name: "math.Sqrt" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SqrtGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SqrtGrad.pbtxt index 8eca1b89b31..ceb42850b43 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SqrtGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SqrtGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SqrtGrad" endpoint { name: "math.SqrtGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Square.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Square.pbtxt index 561e10e020b..5599e333861 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Square.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Square.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Square" endpoint { name: "math.Square" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SquaredDifference.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SquaredDifference.pbtxt index 752dbcce7b8..a7083707231 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SquaredDifference.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SquaredDifference.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SquaredDifference" endpoint { name: "math.SquaredDifference" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Squeeze.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Squeeze.pbtxt index 264c3846036..2a4e7c84491 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Squeeze.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Squeeze.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Squeeze" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StackCloseV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StackCloseV2.pbtxt index 28aff9e1913..a8bfea03223 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StackCloseV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StackCloseV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StackCloseV2" endpoint { name: "StackClose" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StackPopV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StackPopV2.pbtxt index d2ecf4e5a87..2e4105bb750 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StackPopV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StackPopV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StackPopV2" endpoint { name: "StackPop" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StackPushV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StackPushV2.pbtxt index 519fd6c6b28..148a8063d32 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StackPushV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StackPushV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StackPushV2" endpoint { name: "StackPush" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StackV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StackV2.pbtxt index 725e469a031..b3bd175aa48 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StackV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StackV2.pbtxt @@ -1,6 +1,7 @@ op { + visibility: VISIBLE graph_op_name: "StackV2" endpoint { - name: "Stack" + name: "StackCreate" } } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Stage.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Stage.pbtxt index 87e0c7d9811..c723e60e880 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Stage.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Stage.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Stage" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StageClear.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StageClear.pbtxt index 26890a55b39..eac43754533 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StageClear.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StageClear.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "StageClear" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StagePeek.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StagePeek.pbtxt index 7c3ed3dc91c..04c8d374290 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StagePeek.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StagePeek.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "StagePeek" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StageSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StageSize.pbtxt index a697b775571..45c3929f9c0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StageSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StageSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StageSize" out_arg { name: "size" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulPartitionedCall.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulPartitionedCall.pbtxt index 2e6decf19ad..a4c76c42690 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulPartitionedCall.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulPartitionedCall.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "StatefulPartitionedCall" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulRandomBinomial.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulRandomBinomial.pbtxt index 20822a13398..73da6104d73 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulRandomBinomial.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulRandomBinomial.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatefulRandomBinomial" endpoint { name: "random.StatefulRandomBinomial" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulStandardNormalV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulStandardNormalV2.pbtxt index 29ecca63d71..ba4faf0d240 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulStandardNormalV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulStandardNormalV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatefulStandardNormalV2" endpoint { name: "random.StatefulStandardNormal" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulTruncatedNormal.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulTruncatedNormal.pbtxt index 56182fdc77d..d65ac72ba24 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulTruncatedNormal.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulTruncatedNormal.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatefulTruncatedNormal" endpoint { name: "random.StatefulTruncatedNormal" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniform.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniform.pbtxt index 9b5a3dc027d..d0a04cb7c8f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniform.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniform.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatefulUniform" endpoint { name: "random.StatefulUniform" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniformFullInt.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniformFullInt.pbtxt index 33f2029a665..ce31187b355 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniformFullInt.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniformFullInt.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatefulUniformFullInt" endpoint { name: "random.StatefulUniformFullInt" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniformInt.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniformInt.pbtxt index bdf7f00e459..c265945b378 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniformInt.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatefulUniformInt.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatefulUniformInt" endpoint { name: "random.StatefulUniformInt" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessCase.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessCase.pbtxt index ded4c85ca32..3a9fdec0052 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessCase.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessCase.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "StatelessCase" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessIf.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessIf.pbtxt index 37c7b9a9629..3aea3f14382 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessIf.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessIf.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "StatelessIf" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessMultinomial.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessMultinomial.pbtxt index 8cfbbfb2c25..2ff4b3b8e73 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessMultinomial.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessMultinomial.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessMultinomial" endpoint { name: "random.StatelessMultinomial" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessParameterizedTruncatedNormal.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessParameterizedTruncatedNormal.pbtxt index 11d620e74ed..adb32eec836 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessParameterizedTruncatedNormal.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessParameterizedTruncatedNormal.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessParameterizedTruncatedNormal" endpoint { name: "random.StatelessParameterizedTruncatedNormal" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomBinomial.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomBinomial.pbtxt index b955ecbb4ae..943a6947de3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomBinomial.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomBinomial.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomBinomial" endpoint { name: "random.StatelessRandomBinomial" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGammaV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGammaV3.pbtxt index c3c752a1c46..ee8c161b4f1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGammaV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGammaV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomGammaV3" endpoint { name: "random.StatelessRandomGamma" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetAlg.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetAlg.pbtxt index af3865f9438..fd0f93593e2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetAlg.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetAlg.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomGetAlg" endpoint { name: "random.StatelessRandomGetAlg" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetKeyCounter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetKeyCounter.pbtxt index 24ae0b187e4..07e2eff35e7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetKeyCounter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetKeyCounter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomGetKeyCounter" endpoint { name: "random.StatelessRandomGetKeyCounter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetKeyCounterAlg.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetKeyCounterAlg.pbtxt index f38cd9e7d55..5e7bef6aa6d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetKeyCounterAlg.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomGetKeyCounterAlg.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomGetKeyCounterAlg" endpoint { name: "random.StatelessRandomGetKeyCounterAlg" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomNormal.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomNormal.pbtxt index 4e648cfa691..3e301bd2723 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomNormal.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomNormal.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomNormal" endpoint { name: "random.StatelessRandomNormal" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomNormalV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomNormalV2.pbtxt index db288c2cc74..da12a69c0c1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomNormalV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomNormalV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomNormalV2" endpoint { name: "random.StatelessRandomNormalV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomPoisson.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomPoisson.pbtxt index 6fc96330192..f4d175a5cd8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomPoisson.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomPoisson.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomPoisson" endpoint { name: "random.StatelessRandomPoisson" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniform.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniform.pbtxt index 6f8f328b41f..e8de1d2d787 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniform.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniform.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomUniform" endpoint { name: "random.StatelessRandomUniform" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformFullInt.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformFullInt.pbtxt index 9349b1a79c7..627d8bc0e76 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformFullInt.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformFullInt.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomUniformFullInt" endpoint { name: "random.StatelessRandomUniformFullInt" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformFullIntV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformFullIntV2.pbtxt index f09a676ba26..5d53e1a9f5d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformFullIntV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformFullIntV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomUniformFullIntV2" endpoint { name: "random.StatelessRandomUniformFullIntV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformInt.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformInt.pbtxt index 2fec4a7cc62..584235bfb8b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformInt.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformInt.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomUniformInt" endpoint { name: "random.StatelessRandomUniformInt" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformIntV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformIntV2.pbtxt index 6d1578464cf..e97edc51a40 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformIntV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformIntV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomUniformIntV2" endpoint { name: "random.StatelessRandomUniformIntV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformV2.pbtxt index 006d0ce695d..6f67b221bb4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessRandomUniformV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessRandomUniformV2" endpoint { name: "random.StatelessRandomUniformV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessSampleDistortedBoundingBox.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessSampleDistortedBoundingBox.pbtxt index fb5828c0016..c36668abf17 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessSampleDistortedBoundingBox.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessSampleDistortedBoundingBox.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessSampleDistortedBoundingBox" endpoint { name: "image.StatelessSampleDistortedBoundingBox" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessShuffle.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessShuffle.pbtxt index 45ea52e0e3f..33a0530bae0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessShuffle.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessShuffle.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessShuffle" endpoint { name: "random.experimental.StatelessShuffle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessTruncatedNormal.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessTruncatedNormal.pbtxt index c05071dd737..474bdc4e4bb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessTruncatedNormal.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessTruncatedNormal.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessTruncatedNormal" endpoint { name: "random.StatelessTruncatedNormal" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessTruncatedNormalV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessTruncatedNormalV2.pbtxt index 61448d89e57..e890ecd3039 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessTruncatedNormalV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessTruncatedNormalV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatelessTruncatedNormalV2" endpoint { name: "random.StatelessTruncatedNormalV2" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessWhile.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessWhile.pbtxt index 1a4252c90ae..41bb7a7961b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessWhile.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatelessWhile.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "StatelessWhile" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StaticRegexFullMatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StaticRegexFullMatch.pbtxt index 66b841ad74b..07151390a8a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StaticRegexFullMatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StaticRegexFullMatch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StaticRegexFullMatch" endpoint { name: "strings.StaticRegexFullMatch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StaticRegexReplace.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StaticRegexReplace.pbtxt index 305bc8a3caf..e5007d4cfc1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StaticRegexReplace.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StaticRegexReplace.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StaticRegexReplace" endpoint { name: "strings.StaticRegexReplace" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorHandle.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorHandle.pbtxt index 301dc982e32..c84fc446134 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorHandle.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorHandle.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatsAggregatorHandle" endpoint { name: "data.StatsAggregatorHandle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorHandleV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorHandleV2.pbtxt index 358f535edb9..e45249ee2a1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorHandleV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorHandleV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatsAggregatorHandleV2" endpoint { name: "data.StatsAggregatorHandle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorSetSummaryWriter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorSetSummaryWriter.pbtxt index a0c6b1f833d..a60b4cc2816 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorSetSummaryWriter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorSetSummaryWriter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatsAggregatorSetSummaryWriter" endpoint { name: "data.StatsAggregatorSetSummaryWriter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorSummary.pbtxt index f83c8f5a4f6..684b8f7279c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StatsAggregatorSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StatsAggregatorSummary" endpoint { name: "summary.StatsAggregatorSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StochasticCastToInt.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StochasticCastToInt.pbtxt index 2fec43f9fa6..8db44ea4796 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StochasticCastToInt.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StochasticCastToInt.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StochasticCastToInt" endpoint { name: "StochasticCastToInt" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StopGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StopGradient.pbtxt index ad8473e8161..7c98d160fe9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StopGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StopGradient.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "StopGradient" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSlice.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSlice.pbtxt index b0c301dab13..6279266e73d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSlice.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSlice.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "StridedSlice" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSliceAssign.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSliceAssign.pbtxt index 6850dc2d1b2..db34ad3718a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSliceAssign.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSliceAssign.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "StridedSliceAssign" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSliceGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSliceGrad.pbtxt index b03204bcc4d..08d32c7c2ae 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSliceGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StridedSliceGrad.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "StridedSliceGrad" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringFormat.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringFormat.pbtxt index cdd03139966..1ad4b8a7c52 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringFormat.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringFormat.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringFormat" endpoint { name: "strings.StringFormat" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringJoin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringJoin.pbtxt index b43ff157cd5..3d69094d0d5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringJoin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringJoin.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringJoin" endpoint { name: "strings.Join" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringLength.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringLength.pbtxt index c8eb48cc3c7..d7662c7bd91 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringLength.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringLength.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringLength" endpoint { name: "strings.StringLength" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringLower.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringLower.pbtxt index 2fcba3db83e..a97797c82a6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringLower.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringLower.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringLower" endpoint { name: "strings.Lower" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringNGrams.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringNGrams.pbtxt index 27fc41b0a66..6e309754448 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringNGrams.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringNGrams.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringNGrams" endpoint { name: "strings.StringNGrams" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringSplitV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringSplitV2.pbtxt index 18c71d6bd7c..920c1897b75 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringSplitV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringSplitV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringSplitV2" endpoint { name: "strings.StringSplit" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringStrip.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringStrip.pbtxt index 01691211ff6..2e86e88d952 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringStrip.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringStrip.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringStrip" endpoint { name: "strings.Strip" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucket.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucket.pbtxt index 765517578d0..bcdd86c50bf 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucket.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucket.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringToHashBucket" endpoint { name: "strings.ToHashBucket" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucketFast.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucketFast.pbtxt index de08bc2d36f..22644b50359 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucketFast.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucketFast.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringToHashBucketFast" endpoint { name: "strings.ToHashBucketFast" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucketStrong.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucketStrong.pbtxt index 15b9138238a..85fa0bebbc1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucketStrong.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToHashBucketStrong.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringToHashBucketStrong" endpoint { name: "strings.ToHashBucketStrong" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToNumber.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToNumber.pbtxt index 196f694da5c..67a3b66b5f8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToNumber.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringToNumber.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringToNumber" endpoint { name: "strings.ToNumber" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringUpper.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringUpper.pbtxt index 0f08b83bfea..36e84c28477 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_StringUpper.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_StringUpper.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "StringUpper" endpoint { name: "strings.Upper" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sub.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sub.pbtxt index e79c6a0036b..3eb4886b046 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sub.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sub.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Sub" endpoint { name: "math.Sub" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Substr.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Substr.pbtxt index 78f34550a5b..2fbaf49f310 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Substr.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Substr.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Substr" endpoint { name: "strings.Substr" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sum.pbtxt index 3999fa6ed12..5b96b0eeabe 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Sum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Sum.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Sum" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SummaryWriter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SummaryWriter.pbtxt index 8338c0fa188..2f92a7a91a2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SummaryWriter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SummaryWriter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SummaryWriter" endpoint { name: "summary.SummaryWriter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Svd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Svd.pbtxt index 889d4162818..b63de9d7fb1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Svd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Svd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Svd" endpoint { name: "linalg.Svd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Switch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Switch.pbtxt index edd9255452d..1d1f34bc277 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Switch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Switch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Switch" endpoint { name: "SwitchCond" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SymbolicGradient.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SymbolicGradient.pbtxt index 6c6e68ae740..3bcd0816017 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SymbolicGradient.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SymbolicGradient.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SymbolicGradient" endpoint { name: "train.SymbolicGradient" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_SyncDevice.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_SyncDevice.pbtxt index a9af5f7b981..5d52baac9da 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_SyncDevice.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_SyncDevice.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "SyncDevice" endpoint { name: "SyncDevice" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompilationResult.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompilationResult.pbtxt index 52d5f4ef40c..708624d94dd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompilationResult.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompilationResult.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUCompilationResult" endpoint { name: "tpu.CompilationResult" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompile.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompile.pbtxt index 78361b1e6ab..aa0aff59cf3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompile.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompile.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUCompile" endpoint: { name: "tpu.Compile" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompileSucceededAssert.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompileSucceededAssert.pbtxt index de403819900..6d174965861 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompileSucceededAssert.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUCompileSucceededAssert.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUCompileSucceededAssert" endpoint: { name: "tpu.CompileSucceededAssert" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUEmbeddingActivations.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUEmbeddingActivations.pbtxt index 9e71e00a371..12f246fe851 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUEmbeddingActivations.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUEmbeddingActivations.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUEmbeddingActivations" endpoint { name: "tpu.EmbeddingActivations" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUExecute.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUExecute.pbtxt index 8d6be612503..875ec3b8535 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUExecute.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUExecute.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUExecute" endpoint: { name: "tpu.Execute" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUExecuteAndUpdateVariables.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUExecuteAndUpdateVariables.pbtxt index d51c7b9b39e..9948941a2db 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUExecuteAndUpdateVariables.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUExecuteAndUpdateVariables.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUExecuteAndUpdateVariables" endpoint: { name: "tpu.ExecuteAndUpdateVariables" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUOrdinalSelector.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUOrdinalSelector.pbtxt index 45fd42ed30b..63e235c38d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUOrdinalSelector.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUOrdinalSelector.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUOrdinalSelector" endpoint { name: "tpu.OrdinalSelector" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedCall.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedCall.pbtxt index dc133981735..8fb5545a812 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedCall.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedCall.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUPartitionedCall" endpoint { name: "tpu.PartitionedCall" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedInputV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedInputV2.pbtxt index 3f94d151be6..3fb6954d8ee 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedInputV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedInputV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUPartitionedInputV2" endpoint { name: "tpu.PartitionedInput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedOutputV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedOutputV2.pbtxt index b710b7f6f32..154f14a12ee 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedOutputV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUPartitionedOutputV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUPartitionedOutputV2" endpoint { name: "tpu.PartitionedOutput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicateMetadata.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicateMetadata.pbtxt index b6a02c9970c..defa58c6584 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicateMetadata.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicateMetadata.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUReplicateMetadata" endpoint { name: "tpu.ReplicateMetadata" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicatedInput.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicatedInput.pbtxt index 436fd057719..a260c1c5d1c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicatedInput.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicatedInput.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUReplicatedInput" endpoint { name: "tpu.ReplicatedInput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicatedOutput.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicatedOutput.pbtxt index 077eaf01c3f..e49d1e28d49 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicatedOutput.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReplicatedOutput.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUReplicatedOutput" endpoint { name: "tpu.ReplicatedOutput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReshardVariables.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReshardVariables.pbtxt index 7a6a824b2bd..127ac6f4d2b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReshardVariables.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPUReshardVariables.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPUReshardVariables" endpoint { name: "tpu.TPUReshardVariables" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPURoundRobin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPURoundRobin.pbtxt index aef3e5e47d1..8ea56aa4a33 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TPURoundRobin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TPURoundRobin.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TPURoundRobin" endpoint { name: "tpu.TPURoundRobin" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TakeManySparseFromTensorsMap.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TakeManySparseFromTensorsMap.pbtxt index 620e9fb0120..9313748bcd2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TakeManySparseFromTensorsMap.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TakeManySparseFromTensorsMap.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TakeManySparseFromTensorsMap" endpoint { name: "sparse.TakeManySparseFromTensorsMap" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Tan.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Tan.pbtxt index 560ca546b76..68a602f1922 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Tan.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Tan.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Tan" endpoint { name: "math.Tan" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Tanh.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Tanh.pbtxt index 90e441808f9..755eeb92c64 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Tanh.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Tanh.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Tanh" endpoint { name: "math.Tanh" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TanhGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TanhGrad.pbtxt index c0837d04874..4bd1bda2171 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TanhGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TanhGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TanhGrad" endpoint { name: "math.TanhGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TemporaryVariable.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TemporaryVariable.pbtxt index 8ce760f9c94..c8ff04b87e7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TemporaryVariable.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TemporaryVariable.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TemporaryVariable" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayCloseV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayCloseV3.pbtxt index 4a065779790..e2ba5d44e62 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayCloseV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayCloseV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorArrayCloseV3" endpoint { name: "TensorArrayClose" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayConcatV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayConcatV3.pbtxt index 34d09c901a0..64a203b0393 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayConcatV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayConcatV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorArrayConcatV3" endpoint { name: "TensorArrayConcat" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGatherV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGatherV3.pbtxt index b792ee98826..e6a188ed9a3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGatherV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGatherV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorArrayGatherV3" endpoint { name: "TensorArrayGather" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGradV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGradV3.pbtxt index 2c9adebd044..f763672814a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGradV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGradV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorArrayGradV3" endpoint { name: "TensorArrayGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGradWithShape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGradWithShape.pbtxt index d6804fb697b..bba4947f438 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGradWithShape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayGradWithShape.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorArrayGradWithShape" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayPack.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayPack.pbtxt index 030950b06fc..869d0f43b07 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayPack.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayPack.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorArrayPack" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayReadV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayReadV3.pbtxt index e6d38d64df9..9569a3439a9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayReadV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayReadV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorArrayReadV3" endpoint { name: "TensorArrayRead" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayScatterV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayScatterV3.pbtxt index 179c9611f58..c707d895d88 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayScatterV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayScatterV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorArrayScatterV3" endpoint { name: "TensorArrayScatter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArraySizeV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArraySizeV3.pbtxt index 55fe2ae46e9..31821dacaa9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArraySizeV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArraySizeV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorArraySizeV3" endpoint { name: "TensorArraySize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArraySplitV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArraySplitV3.pbtxt index 721af074d0d..428e4e76eaa 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArraySplitV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArraySplitV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorArraySplitV3" endpoint { name: "TensorArraySplit" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayUnpack.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayUnpack.pbtxt index a9011de23ea..fceaf068dae 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayUnpack.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayUnpack.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorArrayUnpack" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayV3.pbtxt index 2b87617a1f1..0b90427bb75 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorArrayV3" endpoint { name: "TensorArray" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayWriteV3.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayWriteV3.pbtxt index 7321057b2f0..fa206bbe0d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayWriteV3.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorArrayWriteV3.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorArrayWriteV3" endpoint { name: "TensorArrayWrite" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestCreateTreeVariable.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestCreateTreeVariable.pbtxt index 8e2410a0dd7..93b3906d2b5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestCreateTreeVariable.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestCreateTreeVariable.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorForestCreateTreeVariable" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeDeserialize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeDeserialize.pbtxt index 724bdb282d0..fab088263eb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeDeserialize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeDeserialize.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorForestTreeDeserialize" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeIsInitializedOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeIsInitializedOp.pbtxt index 7e93af8508f..78637d8f24b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeIsInitializedOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeIsInitializedOp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorForestTreeIsInitializedOp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreePredict.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreePredict.pbtxt index 73770fa2913..a405516ca3a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreePredict.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreePredict.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorForestTreePredict" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeResourceHandleOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeResourceHandleOp.pbtxt index c2ef0ee5dd3..7d5f15c2e88 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeResourceHandleOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeResourceHandleOp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorForestTreeResourceHandleOp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeSerialize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeSerialize.pbtxt index d10f9e632b6..bd9c21ad402 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeSerialize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeSerialize.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorForestTreeSerialize" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeSize.pbtxt index 9d81f1ea8c0..ccd2e9b8359 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorForestTreeSize.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorForestTreeSize" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListConcatLists.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListConcatLists.pbtxt index 3fa6265e108..67bb5a42add 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListConcatLists.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListConcatLists.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListConcatLists" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListConcatV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListConcatV2.pbtxt index 556b8bec5c2..9eb92f762b7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListConcatV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListConcatV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorListConcatV2" endpoint { name: "TensorListConcat" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListElementShape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListElementShape.pbtxt index 396a0cfa8fe..8b5bf4ab823 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListElementShape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListElementShape.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListElementShape" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListFromTensor.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListFromTensor.pbtxt index 3179feddd60..6e39c3603f9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListFromTensor.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListFromTensor.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListFromTensor" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListGather.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListGather.pbtxt index 580d34b68f5..fbdf95b8495 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListGather.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListGather.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListGather" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListGetItem.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListGetItem.pbtxt index 2c47208fa05..2126f008508 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListGetItem.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListGetItem.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListGetItem" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListLength.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListLength.pbtxt index 2ec689d71c8..5cbb2949b39 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListLength.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListLength.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListLength" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPopBack.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPopBack.pbtxt index 8d0d6ed55b3..a202f951410 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPopBack.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPopBack.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListPopBack" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPushBack.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPushBack.pbtxt index 285351cf4f3..bc1976329ad 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPushBack.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPushBack.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListPushBack" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPushBackBatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPushBackBatch.pbtxt index 1f33d492601..cb2429698a5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPushBackBatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListPushBackBatch.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListPushBackBatch" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListReserve.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListReserve.pbtxt index 53366006887..2dbc3158483 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListReserve.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListReserve.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListReserve" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListResize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListResize.pbtxt index b01c92d87ef..8cf4113e0d9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListResize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListResize.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListResize" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListScatterIntoExistingList.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListScatterIntoExistingList.pbtxt index 39f976a0c0a..74874214e28 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListScatterIntoExistingList.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListScatterIntoExistingList.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListScatterIntoExistingList" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListScatterV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListScatterV2.pbtxt index 8dd7e6a24e5..3c6b7f87127 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListScatterV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListScatterV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorListScatterV2" endpoint { name: "TensorListScatter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListSetItem.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListSetItem.pbtxt index 002e2a9bd37..5f1ce23c76f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListSetItem.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListSetItem.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListSetItem" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListSplit.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListSplit.pbtxt index 6189864b9e5..cdec66c4141 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListSplit.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListSplit.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListSplit" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListStack.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListStack.pbtxt index b4b03c86e57..a90c00bb32a 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListStack.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorListStack.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorListStack" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapErase.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapErase.pbtxt index ccdc25f3e23..31775e2bf11 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapErase.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapErase.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorMapErase" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapHasKey.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapHasKey.pbtxt index 1b7d4dc5187..36ecd0afbcb 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapHasKey.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapHasKey.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorMapHasKey" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapInsert.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapInsert.pbtxt index e11af0f9f9d..54bcbca6275 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapInsert.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapInsert.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorMapInsert" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapLookup.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapLookup.pbtxt index 6082a8a716a..aeeddf74c7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapLookup.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapLookup.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorMapLookup" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapSize.pbtxt index 4404fe87304..4ae63e845e8 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorMapSize" out_arg { name: "size" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapStackKeys.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapStackKeys.pbtxt index db5ed2c5d7d..85020446c47 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapStackKeys.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorMapStackKeys.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorMapStackKeys" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterAdd.pbtxt index 5a8c079b423..1025a81a744 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterAdd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorScatterAdd" endpoint { name: "TensorScatterNdAdd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterMax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterMax.pbtxt index 5d2bba85997..675e259e5e6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterMax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterMax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorScatterMax" endpoint { name: "TensorScatterNdMax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterMin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterMin.pbtxt index 13db4206343..0717cac52cd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterMin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterMin.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorScatterMin" endpoint { name: "TensorScatterNdMin" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterSub.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterSub.pbtxt index cc8edd680d6..c84b08421d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterSub.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterSub.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorScatterSub" endpoint { name: "TensorScatterNdSub" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterUpdate.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterUpdate.pbtxt index 610854ce204..6db3e35ade7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterUpdate.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorScatterUpdate.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorScatterUpdate" endpoint { name: "TensorScatterNdUpdate" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorStridedSliceUpdate.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorStridedSliceUpdate.pbtxt index 25ad5444c22..929566022cd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorStridedSliceUpdate.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorStridedSliceUpdate.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TensorStridedSliceUpdate" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorSummaryV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorSummaryV2.pbtxt index c285ada0128..b24a1a7a2b2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorSummaryV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TensorSummaryV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TensorSummaryV2" endpoint { name: "summary.TensorSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ThreadPoolHandle.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ThreadPoolHandle.pbtxt index aa1f830c454..d16386ed57c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ThreadPoolHandle.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ThreadPoolHandle.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ThreadPoolHandle" endpoint { name: "data.ThreadPoolHandle" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ThreadUnsafeUnigramCandidateSampler.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ThreadUnsafeUnigramCandidateSampler.pbtxt index 2ef4a834781..2b516280843 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ThreadUnsafeUnigramCandidateSampler.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ThreadUnsafeUnigramCandidateSampler.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ThreadUnsafeUnigramCandidateSampler" endpoint { name: "random.ThreadUnsafeUnigramCandidateSampler" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Tile.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Tile.pbtxt index 8e4e63a2228..0fa56fa1fdd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Tile.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Tile.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Tile" endpoint { name: "Tile" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TileGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TileGrad.pbtxt index 7cd975c02f6..62e42428e0b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TileGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TileGrad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TileGrad" endpoint { name: "train.TileGrad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Timestamp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Timestamp.pbtxt index 9ebc664ae15..f736a090070 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Timestamp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Timestamp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Timestamp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ToBool.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ToBool.pbtxt index 0f4f25bb522..22484c5d7f6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ToBool.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ToBool.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "ToBool" endpoint { name: "dtypes.ToBool" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKUnique.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKUnique.pbtxt index 4f73b1d2114..f27ff77a499 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKUnique.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKUnique.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TopKUnique" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKV2.pbtxt index 2b0dcf7c2a2..f419a1efef9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TopKV2" endpoint { name: "nn.TopK" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKWithUnique.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKWithUnique.pbtxt index 2db3b19ab22..e93caf331af 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKWithUnique.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TopKWithUnique.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TopKWithUnique" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TpuHandleToProtoKey.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TpuHandleToProtoKey.pbtxt index 3377fd9f87e..abfebedf34c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TpuHandleToProtoKey.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TpuHandleToProtoKey.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TpuHandleToProtoKey" endpoint { name: "tpu.TpuHandleToProtoKey" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Transpose.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Transpose.pbtxt index ad7900c00b2..89b4e99ecd4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Transpose.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Transpose.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Transpose" endpoint { name: "linalg.Transpose" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TridiagonalMatMul.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TridiagonalMatMul.pbtxt index 2c167ce6dc0..240ba5876e6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TridiagonalMatMul.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TridiagonalMatMul.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TridiagonalMatMul" endpoint { name: "linalg.TridiagonalMatMul" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TridiagonalSolve.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TridiagonalSolve.pbtxt index a15b050cb36..af3b6168e89 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TridiagonalSolve.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TridiagonalSolve.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TridiagonalSolve" endpoint { name: "linalg.TridiagonalSolve" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncateDiv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncateDiv.pbtxt index 4fbd6823401..2f8a4b52027 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncateDiv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncateDiv.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TruncateDiv" endpoint { name: "math.TruncateDiv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncateMod.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncateMod.pbtxt index 7d1ae9a14fa..24a731560a2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncateMod.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncateMod.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TruncateMod" endpoint { name: "math.TruncateMod" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncatedNormal.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncatedNormal.pbtxt index b2dd52c9558..3acaf2fafe6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncatedNormal.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TruncatedNormal.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "TruncatedNormal" endpoint { name: "random.TruncatedNormal" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_TryRpc.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_TryRpc.pbtxt index 7ca476086a8..27172fb0c67 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_TryRpc.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_TryRpc.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "TryRpc" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Unbatch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Unbatch.pbtxt index 513b05593a1..3de5706fca5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Unbatch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Unbatch.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Unbatch" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnbatchGrad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnbatchGrad.pbtxt index ce612f84f6f..5f2ea03307b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnbatchGrad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnbatchGrad.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "UnbatchGrad" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UncompressElement.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UncompressElement.pbtxt index 46225e53cdd..bc21a9c9e69 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UncompressElement.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UncompressElement.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UncompressElement" endpoint { name: "data.UncompressElement" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeDecode.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeDecode.pbtxt index c59cf22e4a0..dca1489846e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeDecode.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeDecode.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UnicodeDecode" endpoint { name: "strings.UnicodeDecode" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeDecodeWithOffsets.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeDecodeWithOffsets.pbtxt index 16cc033140c..7c17a9fe57d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeDecodeWithOffsets.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeDecodeWithOffsets.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UnicodeDecodeWithOffsets" endpoint { name: "strings.UnicodeDecodeWithOffsets" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeEncode.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeEncode.pbtxt index dd573e59822..9b8c17587a0 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeEncode.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeEncode.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UnicodeEncode" endpoint { name: "strings.UnicodeEncode" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeScript.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeScript.pbtxt index a873151d5fc..8b2d291ccce 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeScript.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeScript.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UnicodeScript" endpoint { name: "strings.UnicodeScript" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeTranscode.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeTranscode.pbtxt index 131cc6169c7..4b93c5d5f2d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeTranscode.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnicodeTranscode.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UnicodeTranscode" endpoint { name: "strings.UnicodeTranscode" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformCandidateSampler.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformCandidateSampler.pbtxt index d7a56c2a6eb..a96dd29dda5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformCandidateSampler.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformCandidateSampler.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniformCandidateSampler" endpoint { name: "random.UniformCandidateSampler" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformDequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformDequantize.pbtxt index b232938dba0..08124f39044 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformDequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformDequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniformDequantize" endpoint { name: "quantization.UniformDequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantize.pbtxt index 221a797635b..22c248ea8d4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniformQuantize" endpoint { name: "quantization.UniformQuantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedAdd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedAdd.pbtxt index a729f14137f..51931fc9707 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedAdd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedAdd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniformQuantizedAdd" endpoint { name: "math.UniformQuantizedAdd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedClipByValue.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedClipByValue.pbtxt index e5501bb7600..e768a707755 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedClipByValue.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedClipByValue.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniformQuantizedClipByValue" endpoint { name: "UniformQuantizedClipByValue" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedConvolution.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedConvolution.pbtxt index 8609b12ec77..3d672fac47b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedConvolution.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedConvolution.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniformQuantizedConvolution" endpoint { name: "nn.UniformQuantizedConvolution" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedConvolutionHybrid.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedConvolutionHybrid.pbtxt index 57071557bef..d5f04a646d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedConvolutionHybrid.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedConvolutionHybrid.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniformQuantizedConvolutionHybrid" endpoint { name: "nn.UniformQuantizedConvolutionHybrid" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedDot.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedDot.pbtxt index 9d06936eaa3..e238c3e49b9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedDot.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedDot.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniformQuantizedDot" endpoint { name: "quantization.UniformQuantizedDot" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedDotHybrid.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedDotHybrid.pbtxt index 73d7c46484c..2959a22a291 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedDotHybrid.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformQuantizedDotHybrid.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniformQuantizedDotHybrid" endpoint { name: "quantization.UniformQuantizedDotHybrid" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformRequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformRequantize.pbtxt index 016707d2f20..a575ed04e2d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformRequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniformRequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniformRequantize" endpoint { name: "quantization.UniformRequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniqueV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniqueV2.pbtxt index 6fe22cb1020..4fea9563886 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniqueV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniqueV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniqueV2" endpoint { name: "Unique" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniqueWithCountsV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniqueWithCountsV2.pbtxt index eb157451143..95924435d18 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UniqueWithCountsV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UniqueWithCountsV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UniqueWithCountsV2" endpoint { name: "UniqueWithCounts" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Unpack.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Unpack.pbtxt index c3ad63e8f80..6f7324be33f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Unpack.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Unpack.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Unpack" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnravelIndex.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnravelIndex.pbtxt index 1a08dc8f1e7..ae94b4777c5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnravelIndex.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnravelIndex.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "UnravelIndex" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentJoin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentJoin.pbtxt index 71c587c4f80..5121ea5571d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentJoin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentJoin.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UnsortedSegmentJoin" endpoint { name: "strings.UnsortedSegmentJoin" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentMax.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentMax.pbtxt index a58d8740d56..45871a02aa1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentMax.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentMax.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UnsortedSegmentMax" endpoint { name: "math.UnsortedSegmentMax" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentMin.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentMin.pbtxt index 2f17ab4624e..9eb1a6fa994 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentMin.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentMin.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UnsortedSegmentMin" endpoint { name: "math.UnsortedSegmentMin" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentProd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentProd.pbtxt index c594941bcc0..3affa46bbcf 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentProd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentProd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UnsortedSegmentProd" endpoint { name: "math.UnsortedSegmentProd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentSum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentSum.pbtxt index e10cf8a6c2c..5240db87509 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentSum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UnsortedSegmentSum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "UnsortedSegmentSum" endpoint { name: "math.UnsortedSegmentSum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Unstage.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Unstage.pbtxt index 31a97cf84db..6356f2a7acd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Unstage.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Unstage.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Unstage" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_UpperBound.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_UpperBound.pbtxt index 229a6ddfc36..aca7ec550d1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_UpperBound.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_UpperBound.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "UpperBound" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_VarHandleOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_VarHandleOp.pbtxt index 30bdace0e6e..bfad5a52c69 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_VarHandleOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_VarHandleOp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "VarHandleOp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_VarIsInitializedOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_VarIsInitializedOp.pbtxt index a3992019bc7..7de17ce5f50 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_VarIsInitializedOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_VarIsInitializedOp.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "VarIsInitializedOp" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_VariableShape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_VariableShape.pbtxt index 38c63b5b701..c79f4558ab5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_VariableShape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_VariableShape.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "VariableShape" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_VariableV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_VariableV2.pbtxt index c566dd1e79f..fddc7ef0723 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_VariableV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_VariableV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "VariableV2" endpoint { name: "Variable" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Where.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Where.pbtxt index f130181a6e3..7b3fbab8a9d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Where.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Where.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Where" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_While.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_While.pbtxt index 9d0f3b07633..00208efecfd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_While.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_While.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "While" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WholeFileReaderV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WholeFileReaderV2.pbtxt index e031d705fb2..debdb0a57f4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_WholeFileReaderV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_WholeFileReaderV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "WholeFileReaderV2" endpoint { name: "io.WholeFileReader" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Window.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Window.pbtxt index 95a92a29def..91fc235a976 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Window.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Window.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "Window" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WindowOp.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WindowOp.pbtxt index d5683040e7b..ab887423288 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_WindowOp.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_WindowOp.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "WindowOp" endpoint { name: "data.WindowOp" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WorkerHeartbeat.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WorkerHeartbeat.pbtxt index c2996f8aab8..f856ebae552 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_WorkerHeartbeat.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_WorkerHeartbeat.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "WorkerHeartbeat" endpoint { name: "tpu.WorkerHeartbeat" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteAudioSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteAudioSummary.pbtxt index fcd0df85c2c..e7b98098016 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteAudioSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteAudioSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "WriteAudioSummary" endpoint { name: "summary.WriteAudioSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteFile.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteFile.pbtxt index a2d6a5bace4..c2d2b67699f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteFile.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteFile.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "WriteFile" endpoint { name: "io.WriteFile" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteGraphSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteGraphSummary.pbtxt index 8e461bbba33..7e7005e0220 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteGraphSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteGraphSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "WriteGraphSummary" endpoint { name: "summary.WriteGraphSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteHistogramSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteHistogramSummary.pbtxt index c9910790325..a0c1701f6d1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteHistogramSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteHistogramSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "WriteHistogramSummary" endpoint { name: "summary.WriteHistogramSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteImageSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteImageSummary.pbtxt index 08bf0adb2f2..eddb644aa5b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteImageSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteImageSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "WriteImageSummary" endpoint { name: "summary.WriteImageSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteRawProtoSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteRawProtoSummary.pbtxt index 7f040b25166..f78bb0f262b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteRawProtoSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteRawProtoSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "WriteRawProtoSummary" endpoint { name: "summary.WriteRawProtoSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteScalarSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteScalarSummary.pbtxt index 7bc7c16a0c7..2d2b33865fa 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteScalarSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteScalarSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "WriteScalarSummary" endpoint { name: "summary.WriteScalarSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteSummary.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteSummary.pbtxt index 1141cb6dbb1..09c610f8076 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteSummary.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_WriteSummary.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "WriteSummary" endpoint { name: "summary.WriteSummary" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Xdivy.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Xdivy.pbtxt index 13a94b8a56e..7cd3819fb68 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Xdivy.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Xdivy.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Xdivy" endpoint { name: "math.Xdivy" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaAllReduce.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaAllReduce.pbtxt index 3bd3a004edb..eb7532c357c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaAllReduce.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaAllReduce.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaAllReduce" endpoint { name: "xla.AllReduce" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaBroadcastHelper.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaBroadcastHelper.pbtxt index b8ec724d793..6543eb71201 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaBroadcastHelper.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaBroadcastHelper.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaBroadcastHelper" endpoint { name: "xla.BroadcastHelper" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaClusterOutput.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaClusterOutput.pbtxt index e3cf4086fda..51cbb513ad4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaClusterOutput.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaClusterOutput.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaClusterOutput" endpoint { name: "xla.ClusterOutput" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaConcatND.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaConcatND.pbtxt index e85cc4435f1..80d1adcad85 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaConcatND.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaConcatND.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaConcatND" endpoint { name: "xla.ConcatND" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaConvV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaConvV2.pbtxt index bd999901f47..035dbf19dca 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaConvV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaConvV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaConvV2" endpoint { name: "xla.Conv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaCustomCall.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaCustomCall.pbtxt index babedf68248..dd8cbf1c9e6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaCustomCall.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaCustomCall.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaCustomCall" endpoint { name: "xla.CustomCall" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDequantize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDequantize.pbtxt index 89878147289..4ca3fe413d9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDequantize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDequantize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaDequantize" endpoint { name: "xla.Dequantize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDotV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDotV2.pbtxt index 77e09ad7491..59acfa58fc6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDotV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDotV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaDotV2" endpoint { name: "xla.Dot" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDynamicSlice.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDynamicSlice.pbtxt index 239997bf7aa..c313550903c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDynamicSlice.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDynamicSlice.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaDynamicSlice" endpoint { name: "xla.DynamicSlice" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDynamicUpdateSlice.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDynamicUpdateSlice.pbtxt index eaa6e117c04..bf1560b6ffd 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDynamicUpdateSlice.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaDynamicUpdateSlice.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaDynamicUpdateSlice" endpoint { name: "xla.DynamicUpdateSlice" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaEinsum.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaEinsum.pbtxt index 61c19999605..38b9549fd0f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaEinsum.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaEinsum.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaEinsum" endpoint { name: "xla.Einsum" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaGather.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaGather.pbtxt index 26a9e76c8d3..3f164e5a2f9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaGather.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaGather.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaGather" endpoint { name: "xla.Gather" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaHostCompute.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaHostCompute.pbtxt index 2738b74c205..58b42ca8972 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaHostCompute.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaHostCompute.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaHostCompute" endpoint: { name: "xla.XlaHostCompute" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaIf.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaIf.pbtxt index f52101e0933..76c3de53279 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaIf.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaIf.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaIf" endpoint { name: "xla.If" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaKeyValueSort.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaKeyValueSort.pbtxt index 9712e4c3009..197fff37685 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaKeyValueSort.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaKeyValueSort.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaKeyValueSort" endpoint { name: "xla.KeyValueSort" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaLaunch.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaLaunch.pbtxt index 2e7b1a9c79c..ad52fe53185 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaLaunch.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaLaunch.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaLaunch" endpoint { name: "xla.XlaLaunch" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaOptimizationBarrier.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaOptimizationBarrier.pbtxt index 36ab9b369b6..ac03403c4c2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaOptimizationBarrier.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaOptimizationBarrier.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaOptimizationBarrier" endpoint { name: "xla.OptimizationBarrier" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaPad.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaPad.pbtxt index c2b93e0592b..65a3e105ac3 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaPad.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaPad.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaPad" endpoint { name: "xla.Pad" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecv.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecv.pbtxt index 0c2af3a2242..34a99dc1c4d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecv.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecv.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaRecv" endpoint { name: "xla.Recv" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvFromHost.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvFromHost.pbtxt index c5bbf50280d..a391f48281d 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvFromHost.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvFromHost.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaRecvFromHost" endpoint: { name: "xla.XlaRecvFromHost" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingActivations.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingActivations.pbtxt index 14228752647..5022c15fd1c 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingActivations.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingActivations.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaRecvTPUEmbeddingActivations" endpoint { name: "xla.XlaRecvTPUEmbeddingActivations" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingDeduplicationData.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingDeduplicationData.pbtxt index c091acb4012..0cf0987c1c1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingDeduplicationData.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRecvTPUEmbeddingDeduplicationData.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaRecvTPUEmbeddingDeduplicationData" endpoint { name: "xla.XlaRecvTPUEmbeddingDeduplicationData" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduce.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduce.pbtxt index a674f767417..1aa5cb3d8e7 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduce.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduce.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaReduce" endpoint { name: "xla.Reduce" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduceScatter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduceScatter.pbtxt index 3d84f44b6cd..e01f50568b6 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduceScatter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduceScatter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaReduceScatter" endpoint { name: "xla.ReduceScatter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduceWindow.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduceWindow.pbtxt index c6a830cbfa0..b373f224025 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduceWindow.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReduceWindow.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaReduceWindow" endpoint { name: "xla.ReduceWindow" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRemoveDynamicDimensionSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRemoveDynamicDimensionSize.pbtxt index 9681ddd0420..da437cbdd85 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRemoveDynamicDimensionSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRemoveDynamicDimensionSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaRemoveDynamicDimensionSize" endpoint { name: "xla.RemoveDynamicDimensionSize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReplicaId.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReplicaId.pbtxt index c3b3d772b97..c05740d7ec5 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReplicaId.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaReplicaId.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaReplicaId" endpoint { name: "xla.ReplicaId" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRngBitGenerator.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRngBitGenerator.pbtxt index 7246acf05fb..ad43c3e0e4b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRngBitGenerator.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaRngBitGenerator.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaRngBitGenerator" endpoint { name: "xla.RngBitGenerator" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaScatter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaScatter.pbtxt index 546449833d1..ada3cad6a1e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaScatter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaScatter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaScatter" endpoint { name: "xla.Scatter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSelectAndScatter.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSelectAndScatter.pbtxt index 7295da3c344..398d5c252d4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSelectAndScatter.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSelectAndScatter.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSelectAndScatter" endpoint { name: "xla.SelectAndScatter" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSelfAdjointEig.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSelfAdjointEig.pbtxt index e2ee8a70d44..ba0d65274f4 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSelfAdjointEig.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSelfAdjointEig.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSelfAdjointEig" endpoint { name: "xla.SelfAdjointEig" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSend.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSend.pbtxt index 18eccc92cb2..449f03ec210 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSend.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSend.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSend" endpoint { name: "xla.Send" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendTPUEmbeddingGradients.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendTPUEmbeddingGradients.pbtxt index a402635fa73..b8fd705c59e 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendTPUEmbeddingGradients.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendTPUEmbeddingGradients.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSendTPUEmbeddingGradients" endpoint { name: "xla.XlaSendTPUEmbeddingGradients" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendToHost.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendToHost.pbtxt index 0168ae42782..7a1ee292205 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendToHost.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSendToHost.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSendToHost" endpoint: { name: "xla.XlaSendToHost" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSetBound.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSetBound.pbtxt index 3a53a0e79fe..15259595aca 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSetBound.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSetBound.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSetBound" endpoint: { name: "xla.XlaSetBound" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSetDynamicDimensionSize.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSetDynamicDimensionSize.pbtxt index 724d91eab72..991251b9223 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSetDynamicDimensionSize.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSetDynamicDimensionSize.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSetDynamicDimensionSize" endpoint { name: "xla.SetDynamicDimensionSize" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSharding.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSharding.pbtxt index 6b9cc6c772b..d4eb48632a2 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSharding.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSharding.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSharding" endpoint { name: "xla.Sharding" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSort.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSort.pbtxt index 509f7d96161..be14e6a1df1 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSort.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSort.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSort" endpoint { name: "xla.Sort" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSplitND.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSplitND.pbtxt index 850412bf28c..88842cee4f9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSplitND.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSplitND.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSplitND" endpoint { name: "xla.SplitND" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSpmdFullToShardShape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSpmdFullToShardShape.pbtxt index 560b18cebcb..8b01f372d84 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSpmdFullToShardShape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSpmdFullToShardShape.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSpmdFullToShardShape" endpoint { name: "xla.SpmdFullToShardShape" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSpmdShardToFullShape.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSpmdShardToFullShape.pbtxt index e960c39d326..8fbd831b214 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSpmdShardToFullShape.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSpmdShardToFullShape.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSpmdShardToFullShape" endpoint { name: "xla.SpmdShardToFullShape" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSvd.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSvd.pbtxt index f999da59c39..4256d071274 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSvd.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaSvd.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaSvd" endpoint { name: "xla.Svd" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaVariadicReduceV2.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaVariadicReduceV2.pbtxt index 0be02caac52..77a89498e7f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaVariadicReduceV2.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaVariadicReduceV2.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaVariadicReduceV2" endpoint: { name: "xla.XlaVariadicReduce" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaVariadicSort.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaVariadicSort.pbtxt index 3fa661cd3c6..e1c3ff1e93b 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaVariadicSort.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaVariadicSort.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaVariadicSort" endpoint { name: "xla.XlaVariadicSort" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaWhile.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaWhile.pbtxt index e1a42e446ed..8803344ae31 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaWhile.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_XlaWhile.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "XlaWhile" endpoint { name: "xla.While" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Xlog1py.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Xlog1py.pbtxt index 2eb32801d59..c0b480889c9 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Xlog1py.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Xlog1py.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Xlog1py" endpoint { name: "math.Xlog1py" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Xlogy.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Xlogy.pbtxt index 52f457d6458..737955bc3af 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Xlogy.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Xlogy.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Xlogy" endpoint { name: "math.Xlogy" diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_ZerosLike.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_ZerosLike.pbtxt index 8ef17aba9b4..af1d205a31f 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_ZerosLike.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_ZerosLike.pbtxt @@ -1,3 +1,4 @@ op { + visibility: VISIBLE graph_op_name: "ZerosLike" } diff --git a/tensorflow-core/tensorflow-core-api/src/api/api_def_Zeta.pbtxt b/tensorflow-core/tensorflow-core-api/src/api/api_def_Zeta.pbtxt index b602bbc7e1f..74a62881e01 100644 --- a/tensorflow-core/tensorflow-core-api/src/api/api_def_Zeta.pbtxt +++ b/tensorflow-core/tensorflow-core-api/src/api/api_def_Zeta.pbtxt @@ -1,4 +1,5 @@ op { + visibility: VISIBLE graph_op_name: "Zeta" endpoint { name: "math.Zeta" diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/ClusterOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/ClusterOps.java new file mode 100644 index 00000000000..869fb25f23a --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/ClusterOps.java @@ -0,0 +1,85 @@ +// Copyright 2020-2022 The TensorFlow Authors. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// ============================================================================== +// +// This class has been generated, DO NOT EDIT! +// +package org.tensorflow.op; + +import org.tensorflow.Operand; +import org.tensorflow.op.cluster.KMC2ChainInitialization; +import org.tensorflow.op.cluster.KmeansPlusPlusInitialization; +import org.tensorflow.types.TFloat32; +import org.tensorflow.types.TInt64; + +/** + * An API for building {@code cluster} operations as {@link Op Op}s + * + * @see {@link Ops} + */ +public final class ClusterOps { + private final Scope scope; + + private final Ops ops; + + ClusterOps(Ops ops) { + this.scope = ops.scope(); + this.ops = ops; + } + + /** + * Returns the index of a data point that should be added to the seed set. + * Entries in distances are assumed to be squared distances of candidate points to + * the already sampled centers in the seed set. The op constructs one Markov chain + * of the k-MC^2 algorithm and returns the index of one candidate point to be added + * as an additional cluster center. + * + * @param distances Vector with squared distances to the closest previously sampled cluster center + * for each candidate point. + * @param seed Scalar. Seed for initializing the random number generator. + * @return a new instance of KMC2ChainInitialization + */ + public KMC2ChainInitialization kMC2ChainInitialization(Operand distances, + Operand seed) { + return KMC2ChainInitialization.create(scope, distances, seed); + } + + /** + * Selects num_to_sample rows of input using the KMeans++ criterion. + * Rows of points are assumed to be input points. One row is selected at random. + * Subsequent rows are sampled with probability proportional to the squared L2 + * distance from the nearest row selected thus far till num_to_sample rows have + * been sampled. + * + * @param points Matrix of shape (n, d). Rows are assumed to be input points. + * @param numToSample Scalar. The number of rows to sample. This value must not be larger than n. + * @param seed Scalar. Seed for initializing the random number generator. + * @param numRetriesPerSample Scalar. For each row that is sampled, this parameter + * specifies the number of additional points to draw from the current + * distribution before selecting the best. If a negative value is specified, a + * heuristic is used to sample O(log(num_to_sample)) additional points. + * @return a new instance of KmeansPlusPlusInitialization + */ + public KmeansPlusPlusInitialization kmeansPlusPlusInitialization(Operand points, + Operand numToSample, Operand seed, Operand numRetriesPerSample) { + return KmeansPlusPlusInitialization.create(scope, points, numToSample, seed, numRetriesPerSample); + } + + /** + * Get the parent {@link Ops} object. + */ + public final Ops ops() { + return ops; + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/CollectiveOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/CollectiveOps.java new file mode 100644 index 00000000000..e5b580bd068 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/CollectiveOps.java @@ -0,0 +1,221 @@ +// Copyright 2020-2022 The TensorFlow Authors. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// ============================================================================== +// +// This class has been generated, DO NOT EDIT! +// +package org.tensorflow.op; + +import org.tensorflow.Operand; +import org.tensorflow.op.collective.CollectiveAllToAll; +import org.tensorflow.op.collective.CollectiveAssignGroup; +import org.tensorflow.op.collective.CollectiveBcastRecv; +import org.tensorflow.op.collective.CollectiveBcastSend; +import org.tensorflow.op.collective.CollectiveGather; +import org.tensorflow.op.collective.CollectiveInitializeCommunicator; +import org.tensorflow.op.collective.CollectivePermute; +import org.tensorflow.op.collective.CollectiveReduce; +import org.tensorflow.op.collective.CollectiveReduceScatter; +import org.tensorflow.types.TInt32; +import org.tensorflow.types.family.TNumber; +import org.tensorflow.types.family.TType; + +/** + * An API for building {@code collective} operations as {@link Op Op}s + * + * @see {@link Ops} + */ +public final class CollectiveOps { + private final Scope scope; + + private final Ops ops; + + CollectiveOps(Ops ops) { + this.scope = ops.scope(); + this.ops = ops; + } + + /** + * Mutually exchanges multiple tensors of identical type and shape. + * + * @param data type for {@code data} output + * @param input The input value + * @param communicator The communicator value + * @param groupAssignment The groupAssignment value + * @param options carries optional attribute values + * @param data type for {@code CollectiveAllToAllV3} output and operands + * @return a new instance of CollectiveAllToAll + */ + public CollectiveAllToAll collectiveAllToAll(Operand input, + Operand communicator, Operand groupAssignment, + CollectiveAllToAll.Options... options) { + return CollectiveAllToAll.create(scope, input, communicator, groupAssignment, options); + } + + /** + * Assign group keys based on group assignment. + * + * @param groupAssignment The groupAssignment value + * @param deviceIndex The deviceIndex value + * @param baseKey The baseKey value + * @return a new instance of CollectiveAssignGroup + */ + public CollectiveAssignGroup collectiveAssignGroup(Operand groupAssignment, + Operand deviceIndex, Operand baseKey) { + return CollectiveAssignGroup.create(scope, groupAssignment, deviceIndex, baseKey); + } + + /** + * Receives a tensor value broadcast from another device. + * + * @param data type for {@code data} output + * @param groupSize The groupSize value + * @param groupKey The groupKey value + * @param instanceKey The instanceKey value + * @param shape The shape value + * @param T The value of the T attribute + * @param options carries optional attribute values + * @param data type for {@code CollectiveBcastRecvV2} output and operands + * @return a new instance of CollectiveBcastRecv + */ + public CollectiveBcastRecv collectiveBcastRecv(Operand groupSize, + Operand groupKey, Operand instanceKey, Operand shape, + Class T, CollectiveBcastRecv.Options... options) { + return CollectiveBcastRecv.create(scope, groupSize, groupKey, instanceKey, shape, T, options); + } + + /** + * Broadcasts a tensor value to one or more other devices. + * + * @param data type for {@code data} output + * @param input The input value + * @param groupSize The groupSize value + * @param groupKey The groupKey value + * @param instanceKey The instanceKey value + * @param options carries optional attribute values + * @param data type for {@code CollectiveBcastSendV2} output and operands + * @return a new instance of CollectiveBcastSend + */ + public CollectiveBcastSend collectiveBcastSend(Operand input, + Operand groupSize, Operand groupKey, Operand instanceKey, + CollectiveBcastSend.Options... options) { + return CollectiveBcastSend.create(scope, input, groupSize, groupKey, instanceKey, options); + } + + /** + * Mutually accumulates multiple tensors of identical type and shape. + * {@code is_stateless} means each op does not need control dependencies to other + * collective ops. In this case, keys that are unique at runtime + * (e.g. {@code instance_key}) should be used to distinguish collective groups. + * + * @param data type for {@code data} output + * @param input The input value + * @param groupSize The groupSize value + * @param groupKey The groupKey value + * @param instanceKey The instanceKey value + * @param orderingToken The orderingToken value + * @param options carries optional attribute values + * @param data type for {@code CollectiveGatherV2} output and operands + * @return a new instance of CollectiveGather + */ + public CollectiveGather collectiveGather(Operand input, + Operand groupSize, Operand groupKey, Operand instanceKey, + Iterable> orderingToken, CollectiveGather.Options... options) { + return CollectiveGather.create(scope, input, groupSize, groupKey, instanceKey, orderingToken, options); + } + + /** + * Initializes a group for collective operations. + * + * @param groupKey The groupKey value + * @param rank The rank value + * @param groupSize The groupSize value + * @param options carries optional attribute values + * @return a new instance of CollectiveInitializeCommunicator + */ + public CollectiveInitializeCommunicator collectiveInitializeCommunicator(Operand groupKey, + Operand rank, Operand groupSize, + CollectiveInitializeCommunicator.Options... options) { + return CollectiveInitializeCommunicator.create(scope, groupKey, rank, groupSize, options); + } + + /** + * An Op to permute tensors across replicated TPU instances. + * Each instance supplies its own input. + *

For example, suppose there are 4 TPU instances: {@code [A, B, C, D]}. Passing + * source_target_pairs={@code [[0,1],[1,2],[2,3],[3,0]]} gets the outputs: + * {@code [D, A, B, C]}. + * + * @param data type for {@code output} output + * @param input The local input to be permuted. Currently only supports float and + * bfloat16. + * @param sourceTargetPairs A tensor with shape [num_pairs, 2]. + * @param data type for {@code CollectivePermute} output and operands + * @return a new instance of CollectivePermute + */ + public CollectivePermute collectivePermute(Operand input, + Operand sourceTargetPairs) { + return CollectivePermute.create(scope, input, sourceTargetPairs); + } + + /** + * Mutually reduces multiple tensors of identical type and shape. + * + * @param data type for {@code data} output + * @param input The input value + * @param communicator The communicator value + * @param groupAssignment The groupAssignment value + * @param reduction The value of the reduction attribute + * @param options carries optional attribute values + * @param data type for {@code CollectiveReduceV3} output and operands + * @return a new instance of CollectiveReduce + */ + public CollectiveReduce collectiveReduce(Operand input, + Operand communicator, Operand groupAssignment, String reduction, + CollectiveReduce.Options... options) { + return CollectiveReduce.create(scope, input, communicator, groupAssignment, reduction, options); + } + + /** + * Mutually reduces multiple tensors of identical type and shape and scatters the result. + * {@code is_stateless} means each op does not need control dependencies to other + * collective ops. In this case, keys that are unique at runtime + * (e.g. {@code instance_key}) should be used to distinguish collective groups. + * + * @param data type for {@code data} output + * @param input The input value + * @param groupSize The groupSize value + * @param groupKey The groupKey value + * @param instanceKey The instanceKey value + * @param orderingToken The orderingToken value + * @param mergeOp The value of the mergeOp attribute + * @param finalOp The value of the finalOp attribute + * @param options carries optional attribute values + * @param data type for {@code CollectiveReduceScatterV2} output and operands + * @return a new instance of CollectiveReduceScatter + */ + public CollectiveReduceScatter collectiveReduceScatter(Operand input, + Operand groupSize, Operand groupKey, Operand instanceKey, + Iterable> orderingToken, String mergeOp, String finalOp, + CollectiveReduceScatter.Options... options) { + return CollectiveReduceScatter.create(scope, input, groupSize, groupKey, instanceKey, orderingToken, mergeOp, finalOp, options); + } + + /** + * Get the parent {@link Ops} object. + */ + public final Ops ops() { + return ops; + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DataExperimentalOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DataExperimentalOps.java new file mode 100644 index 00000000000..4644599e7f1 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DataExperimentalOps.java @@ -0,0 +1,738 @@ +// Copyright 2020-2022 The TensorFlow Authors. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// ============================================================================== +// +// This class has been generated, DO NOT EDIT! +// +package org.tensorflow.op; + +import java.util.List; +import org.tensorflow.ConcreteFunction; +import org.tensorflow.Operand; +import org.tensorflow.ndarray.Shape; +import org.tensorflow.op.data.experimental.AssertNextDataset; +import org.tensorflow.op.data.experimental.AutoShardDataset; +import org.tensorflow.op.data.experimental.BytesProducedStatsDataset; +import org.tensorflow.op.data.experimental.CSVDataset; +import org.tensorflow.op.data.experimental.ChooseFastestDataset; +import org.tensorflow.op.data.experimental.DatasetCardinality; +import org.tensorflow.op.data.experimental.DatasetToTFRecord; +import org.tensorflow.op.data.experimental.DenseToSparseBatchDataset; +import org.tensorflow.op.data.experimental.DirectedInterleaveDataset; +import org.tensorflow.op.data.experimental.GroupByReducerDataset; +import org.tensorflow.op.data.experimental.GroupByWindowDataset; +import org.tensorflow.op.data.experimental.IgnoreErrorsDataset; +import org.tensorflow.op.data.experimental.IteratorGetDevice; +import org.tensorflow.op.data.experimental.LatencyStatsDataset; +import org.tensorflow.op.data.experimental.LmdbDataset; +import org.tensorflow.op.data.experimental.MapAndBatchDataset; +import org.tensorflow.op.data.experimental.MapDataset; +import org.tensorflow.op.data.experimental.MatchingFilesDataset; +import org.tensorflow.op.data.experimental.MaxIntraOpParallelismDataset; +import org.tensorflow.op.data.experimental.NonSerializableDataset; +import org.tensorflow.op.data.experimental.ParallelInterleaveDataset; +import org.tensorflow.op.data.experimental.ParseExampleDataset; +import org.tensorflow.op.data.experimental.PrivateThreadPoolDataset; +import org.tensorflow.op.data.experimental.RandomDataset; +import org.tensorflow.op.data.experimental.RebatchDataset; +import org.tensorflow.op.data.experimental.ScanDataset; +import org.tensorflow.op.data.experimental.SetStatsAggregatorDataset; +import org.tensorflow.op.data.experimental.SleepDataset; +import org.tensorflow.op.data.experimental.SlidingWindowDataset; +import org.tensorflow.op.data.experimental.SqlDataset; +import org.tensorflow.op.data.experimental.StatsAggregatorHandle; +import org.tensorflow.op.data.experimental.StatsAggregatorSummary; +import org.tensorflow.op.data.experimental.TakeWhileDataset; +import org.tensorflow.op.data.experimental.ThreadPoolDataset; +import org.tensorflow.op.data.experimental.ThreadPoolHandle; +import org.tensorflow.op.data.experimental.UnbatchDataset; +import org.tensorflow.op.data.experimental.UniqueDataset; +import org.tensorflow.types.TBool; +import org.tensorflow.types.TInt64; +import org.tensorflow.types.TString; +import org.tensorflow.types.family.TType; + +/** + * An API for building {@code data.experimental} operations as {@link Op Op}s + * + * @see {@link Ops} + */ +public final class DataExperimentalOps { + private final Scope scope; + + private final Ops ops; + + DataExperimentalOps(Ops ops) { + this.scope = ops.scope(); + this.ops = ops; + } + + /** + * The ExperimentalAssertNextDataset operation + * + * @param inputDataset The inputDataset value + * @param transformations The transformations value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of AssertNextDataset + */ + public AssertNextDataset assertNextDataset(Operand inputDataset, + Operand transformations, List> outputTypes, + List outputShapes) { + return AssertNextDataset.create(scope, inputDataset, transformations, outputTypes, outputShapes); + } + + /** + * Creates a dataset that shards the input dataset. + * Creates a dataset that shards the input dataset by num_workers, returning a + * sharded dataset for the index-th worker. This attempts to automatically shard + * a dataset by examining the Dataset graph and inserting a shard op before the + * inputs to a reader Dataset (e.g. CSVDataset, TFRecordDataset). + *

This dataset will throw a NotFound error if we cannot shard the dataset + * automatically. + * + * @param inputDataset A variant tensor representing the input dataset. + * @param numWorkers A scalar representing the number of workers to distribute this dataset across. + * @param index A scalar representing the index of the current worker out of num_workers. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @param options carries optional attribute values + * @return a new instance of AutoShardDataset + */ + public AutoShardDataset autoShardDataset(Operand inputDataset, + Operand numWorkers, Operand index, List> outputTypes, + List outputShapes, AutoShardDataset.Options... options) { + return AutoShardDataset.create(scope, inputDataset, numWorkers, index, outputTypes, outputShapes, options); + } + + /** + * Records the bytes size of each element of {@code input_dataset} in a StatsAggregator. + * + * @param inputDataset The inputDataset value + * @param tag The tag value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of BytesProducedStatsDataset + */ + public BytesProducedStatsDataset bytesProducedStatsDataset(Operand inputDataset, + Operand tag, List> outputTypes, List outputShapes) { + return BytesProducedStatsDataset.create(scope, inputDataset, tag, outputTypes, outputShapes); + } + + /** + * The ExperimentalCSVDataset operation + * + * @param filenames The filenames value + * @param compressionType The compressionType value + * @param bufferSize The bufferSize value + * @param header The header value + * @param fieldDelim The fieldDelim value + * @param useQuoteDelim The useQuoteDelim value + * @param naValue The naValue value + * @param selectCols The selectCols value + * @param recordDefaults The recordDefaults value + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of CSVDataset + */ + public CSVDataset cSVDataset(Operand filenames, Operand compressionType, + Operand bufferSize, Operand header, Operand fieldDelim, + Operand useQuoteDelim, Operand naValue, Operand selectCols, + Iterable> recordDefaults, List outputShapes) { + return CSVDataset.create(scope, filenames, compressionType, bufferSize, header, fieldDelim, useQuoteDelim, naValue, selectCols, recordDefaults, outputShapes); + } + + /** + * The ExperimentalChooseFastestDataset operation + * + * @param inputDatasets The inputDatasets value + * @param numExperiments The value of the numExperiments attribute + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of ChooseFastestDataset + */ + public ChooseFastestDataset chooseFastestDataset(Iterable> inputDatasets, + Long numExperiments, List> outputTypes, List outputShapes) { + return ChooseFastestDataset.create(scope, inputDatasets, numExperiments, outputTypes, outputShapes); + } + + /** + * Returns the cardinality of {@code input_dataset}. + * Returns the cardinality of {@code input_dataset}. + * + * @param inputDataset A variant tensor representing the dataset to return cardinality for. + * @return a new instance of DatasetCardinality + */ + public DatasetCardinality datasetCardinality(Operand inputDataset) { + return DatasetCardinality.create(scope, inputDataset); + } + + /** + * Writes the given dataset to the given file using the TFRecord format. + * + * @param inputDataset A variant tensor representing the dataset to write. + * @param filename A scalar string tensor representing the filename to use. + * @param compressionType A scalar string tensor containing either (i) the empty string (no + * compression), (ii) "ZLIB", or (iii) "GZIP". + * @return a new instance of DatasetToTFRecord + */ + public DatasetToTFRecord datasetToTFRecord(Operand inputDataset, + Operand filename, Operand compressionType) { + return DatasetToTFRecord.create(scope, inputDataset, filename, compressionType); + } + + /** + * Creates a dataset that batches input elements into a SparseTensor. + * + * @param inputDataset A handle to an input dataset. Must have a single component. + * @param batchSize A scalar representing the number of elements to accumulate in a + * batch. + * @param rowShape A vector representing the dense shape of each row in the produced + * SparseTensor. The shape may be partially specified, using {@code -1} to indicate + * that a particular dimension should use the maximum size of all batch elements. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of DenseToSparseBatchDataset + */ + public DenseToSparseBatchDataset denseToSparseBatchDataset(Operand inputDataset, + Operand batchSize, Operand rowShape, List> outputTypes, + List outputShapes) { + return DenseToSparseBatchDataset.create(scope, inputDataset, batchSize, rowShape, outputTypes, outputShapes); + } + + /** + * A substitute for {@code InterleaveDataset} on a fixed list of {@code N} datasets. + * + * @param selectorInputDataset A dataset of scalar {@code DT_INT64} elements that determines which of the + * {@code N} data inputs should produce the next output element. + * @param dataInputDatasets {@code N} datasets with the same type that will be interleaved according to + * the values of {@code selector_input_dataset}. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of DirectedInterleaveDataset + */ + public DirectedInterleaveDataset directedInterleaveDataset( + Operand selectorInputDataset, + Iterable> dataInputDatasets, + List> outputTypes, List outputShapes) { + return DirectedInterleaveDataset.create(scope, selectorInputDataset, dataInputDatasets, outputTypes, outputShapes); + } + + /** + * Creates a dataset that computes a group-by on {@code input_dataset}. + * Creates a dataset that computes a group-by on {@code input_dataset}. + * + * @param inputDataset A variant tensor representing the input dataset. + * @param keyFuncOtherArguments A list of tensors, typically values that were captured when + * building a closure for {@code key_func}. + * @param initFuncOtherArguments A list of tensors, typically values that were captured when + * building a closure for {@code init_func}. + * @param reduceFuncOtherArguments A list of tensors, typically values that were captured when + * building a closure for {@code reduce_func}. + * @param finalizeFuncOtherArguments A list of tensors, typically values that were captured when + * building a closure for {@code finalize_func}. + * @param keyFunc A function mapping an element of {@code input_dataset}, concatenated + * with {@code key_func_other_arguments} to a scalar value of type DT_INT64. + * @param initFunc A function mapping a key of type DT_INT64, concatenated with + * {@code init_func_other_arguments} to the initial reducer state. + * @param reduceFunc A function mapping the current reducer state and an element of {@code input_dataset}, + * concatenated with {@code reduce_func_other_arguments} to a new reducer state. + * @param finalizeFunc A function mapping the final reducer state to an output element. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of GroupByReducerDataset + */ + public GroupByReducerDataset groupByReducerDataset(Operand inputDataset, + Iterable> keyFuncOtherArguments, Iterable> initFuncOtherArguments, + Iterable> reduceFuncOtherArguments, + Iterable> finalizeFuncOtherArguments, ConcreteFunction keyFunc, + ConcreteFunction initFunc, ConcreteFunction reduceFunc, ConcreteFunction finalizeFunc, + List> outputTypes, List outputShapes) { + return GroupByReducerDataset.create(scope, inputDataset, keyFuncOtherArguments, initFuncOtherArguments, reduceFuncOtherArguments, finalizeFuncOtherArguments, keyFunc, initFunc, reduceFunc, finalizeFunc, outputTypes, outputShapes); + } + + /** + * Creates a dataset that computes a windowed group-by on {@code input_dataset}. + * // TODO(mrry): Support non-int64 keys. + * + * @param inputDataset The inputDataset value + * @param keyFuncOtherArguments The keyFuncOtherArguments value + * @param reduceFuncOtherArguments The reduceFuncOtherArguments value + * @param windowSizeFuncOtherArguments The windowSizeFuncOtherArguments value + * @param keyFunc A function mapping an element of {@code input_dataset}, concatenated + * with {@code key_func_other_arguments} to a scalar value of type DT_INT64. + * @param reduceFunc The value of the reduceFunc attribute + * @param windowSizeFunc The value of the windowSizeFunc attribute + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of GroupByWindowDataset + */ + public GroupByWindowDataset groupByWindowDataset(Operand inputDataset, + Iterable> keyFuncOtherArguments, Iterable> reduceFuncOtherArguments, + Iterable> windowSizeFuncOtherArguments, ConcreteFunction keyFunc, + ConcreteFunction reduceFunc, ConcreteFunction windowSizeFunc, + List> outputTypes, List outputShapes) { + return GroupByWindowDataset.create(scope, inputDataset, keyFuncOtherArguments, reduceFuncOtherArguments, windowSizeFuncOtherArguments, keyFunc, reduceFunc, windowSizeFunc, outputTypes, outputShapes); + } + + /** + * Creates a dataset that contains the elements of {@code input_dataset} ignoring errors. + * + * @param inputDataset The inputDataset value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @param options carries optional attribute values + * @return a new instance of IgnoreErrorsDataset + */ + public IgnoreErrorsDataset ignoreErrorsDataset(Operand inputDataset, + List> outputTypes, List outputShapes, + IgnoreErrorsDataset.Options... options) { + return IgnoreErrorsDataset.create(scope, inputDataset, outputTypes, outputShapes, options); + } + + /** + * Returns the name of the device on which {@code resource} has been placed. + * + * @param resource The resource value + * @return a new instance of IteratorGetDevice + */ + public IteratorGetDevice iteratorGetDevice(Operand resource) { + return IteratorGetDevice.create(scope, resource); + } + + /** + * Records the latency of producing {@code input_dataset} elements in a StatsAggregator. + * + * @param inputDataset The inputDataset value + * @param tag The tag value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of LatencyStatsDataset + */ + public LatencyStatsDataset latencyStatsDataset(Operand inputDataset, + Operand tag, List> outputTypes, List outputShapes) { + return LatencyStatsDataset.create(scope, inputDataset, tag, outputTypes, outputShapes); + } + + /** + * The ExperimentalLMDBDataset operation + * + * @param filenames The filenames value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of LmdbDataset + */ + public LmdbDataset lmdbDataset(Operand filenames, + List> outputTypes, List outputShapes) { + return LmdbDataset.create(scope, filenames, outputTypes, outputShapes); + } + + /** + * Creates a dataset that fuses mapping with batching. + * Creates a dataset that applies {@code f} to the outputs of {@code input_dataset} and then + * batches {@code batch_size} of them. + *

Unlike a "MapDataset", which applies {@code f} sequentially, this dataset invokes up + * to {@code batch_size * num_parallel_batches} copies of {@code f} in parallel. + * + * @param inputDataset A variant tensor representing the input dataset. + * @param otherArguments A list of tensors, typically values that were captured when building a closure + * for {@code f}. + * @param batchSize A scalar representing the number of elements to accumulate in a + * batch. It determines the number of concurrent invocations of {@code f} that process + * elements from {@code input_dataset} in parallel. + * @param numParallelCalls A scalar representing the maximum number of parallel invocations of the {@code map_fn} + * function. Applying the {@code map_fn} on consecutive input elements in parallel has + * the potential to improve input pipeline throughput. + * @param dropRemainder A scalar representing whether the last batch should be dropped in case its size + * is smaller than desired. + * @param f A function to apply to the outputs of {@code input_dataset}. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @param options carries optional attribute values + * @return a new instance of MapAndBatchDataset + */ + public MapAndBatchDataset mapAndBatchDataset(Operand inputDataset, + Iterable> otherArguments, Operand batchSize, + Operand numParallelCalls, Operand dropRemainder, ConcreteFunction f, + List> outputTypes, List outputShapes, + MapAndBatchDataset.Options... options) { + return MapAndBatchDataset.create(scope, inputDataset, otherArguments, batchSize, numParallelCalls, dropRemainder, f, outputTypes, outputShapes, options); + } + + /** + * Creates a dataset that applies {@code f} to the outputs of {@code input_dataset}. + * + * @param inputDataset The inputDataset value + * @param otherArguments The otherArguments value + * @param f The value of the f attribute + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @param options carries optional attribute values + * @return a new instance of MapDataset + */ + public MapDataset mapDataset(Operand inputDataset, + Iterable> otherArguments, ConcreteFunction f, + List> outputTypes, List outputShapes, + MapDataset.Options... options) { + return MapDataset.create(scope, inputDataset, otherArguments, f, outputTypes, outputShapes, options); + } + + /** + * The ExperimentalMatchingFilesDataset operation + * + * @param patterns The patterns value + * @return a new instance of MatchingFilesDataset + */ + public MatchingFilesDataset matchingFilesDataset(Operand patterns) { + return MatchingFilesDataset.create(scope, patterns); + } + + /** + * Creates a dataset that overrides the maximum intra-op parallelism. + * + * @param inputDataset The inputDataset value + * @param maxIntraOpParallelism Identifies the maximum intra-op parallelism to use. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of MaxIntraOpParallelismDataset + */ + public MaxIntraOpParallelismDataset maxIntraOpParallelismDataset( + Operand inputDataset, Operand maxIntraOpParallelism, + List> outputTypes, List outputShapes) { + return MaxIntraOpParallelismDataset.create(scope, inputDataset, maxIntraOpParallelism, outputTypes, outputShapes); + } + + /** + * The ExperimentalNonSerializableDataset operation + * + * @param inputDataset The inputDataset value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of NonSerializableDataset + */ + public NonSerializableDataset nonSerializableDataset(Operand inputDataset, + List> outputTypes, List outputShapes) { + return NonSerializableDataset.create(scope, inputDataset, outputTypes, outputShapes); + } + + /** + * Creates a dataset that applies {@code f} to the outputs of {@code input_dataset}. + * The resulting dataset is similar to the {@code InterleaveDataset}, with the exception + * that if retrieving the next value from a dataset would cause the requester to + * block, it will skip that input dataset. This dataset is especially useful + * when loading data from a variable-latency datastores (e.g. HDFS, GCS), as it + * allows the training step to proceed so long as some data is available. + *

!! WARNING !! This dataset is not deterministic! + * + * @param inputDataset The inputDataset value + * @param otherArguments The otherArguments value + * @param cycleLength The cycleLength value + * @param blockLength The blockLength value + * @param sloppy The sloppy value + * @param bufferOutputElements The bufferOutputElements value + * @param prefetchInputElements The prefetchInputElements value + * @param f A function mapping elements of {@code input_dataset}, concatenated with + * {@code other_arguments}, to a Dataset variant that contains elements matching + * {@code output_types} and {@code output_shapes}. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of ParallelInterleaveDataset + */ + public ParallelInterleaveDataset parallelInterleaveDataset(Operand inputDataset, + Iterable> otherArguments, Operand cycleLength, Operand blockLength, + Operand sloppy, Operand bufferOutputElements, + Operand prefetchInputElements, ConcreteFunction f, + List> outputTypes, List outputShapes) { + return ParallelInterleaveDataset.create(scope, inputDataset, otherArguments, cycleLength, blockLength, sloppy, bufferOutputElements, prefetchInputElements, f, outputTypes, outputShapes); + } + + /** + * Transforms {@code input_dataset} containing {@code Example} protos as vectors of DT_STRING into a dataset of {@code Tensor} or {@code SparseTensor} objects representing the parsed features. + * + * @param inputDataset The inputDataset value + * @param numParallelCalls The numParallelCalls value + * @param denseDefaults A dict mapping string keys to {@code Tensor}s. + * The keys of the dict must match the dense_keys of the feature. + * @param sparseKeys A list of string keys in the examples features. + * The results for these keys will be returned as {@code SparseTensor} objects. + * @param denseKeys A list of Ndense string Tensors (scalars). + * The keys expected in the Examples features associated with dense values. + * @param sparseTypes A list of {@code DTypes} of the same length as {@code sparse_keys}. + * Only {@code tf.float32} ({@code FloatList}), {@code tf.int64} ({@code Int64List}), + * and {@code tf.string} ({@code BytesList}) are supported. + * @param denseShapes List of tuples with the same length as {@code dense_keys}. + * The shape of the data for each dense feature referenced by {@code dense_keys}. + * Required for any input tensors identified by {@code dense_keys}. Must be + * either fully defined, or may contain an unknown first dimension. + * An unknown first dimension means the feature is treated as having + * a variable number of blocks, and the output shape along this dimension + * is considered unknown at graph build time. Padding is applied for + * minibatch elements smaller than the maximum number of blocks for the + * given feature along this dimension. + * @param outputTypes The type list for the return values. + * @param outputShapes The list of shapes being produced. + * @param options carries optional attribute values + * @return a new instance of ParseExampleDataset + */ + public ParseExampleDataset parseExampleDataset(Operand inputDataset, + Operand numParallelCalls, Iterable> denseDefaults, List sparseKeys, + List denseKeys, List> sparseTypes, List denseShapes, + List> outputTypes, List outputShapes, + ParseExampleDataset.Options... options) { + return ParseExampleDataset.create(scope, inputDataset, numParallelCalls, denseDefaults, sparseKeys, denseKeys, sparseTypes, denseShapes, outputTypes, outputShapes, options); + } + + /** + * Creates a dataset that uses a custom thread pool to compute {@code input_dataset}. + * + * @param inputDataset The inputDataset value + * @param numThreads Identifies the number of threads to use for the private threadpool. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of PrivateThreadPoolDataset + */ + public PrivateThreadPoolDataset privateThreadPoolDataset(Operand inputDataset, + Operand numThreads, List> outputTypes, + List outputShapes) { + return PrivateThreadPoolDataset.create(scope, inputDataset, numThreads, outputTypes, outputShapes); + } + + /** + * Creates a Dataset that returns pseudorandom numbers. + * + * @param seed A scalar seed for the random number generator. If either seed or + * seed2 is set to be non-zero, the random number generator is seeded + * by the given seed. Otherwise, a random seed is used. + * @param seed2 A second scalar seed to avoid seed collision. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of RandomDataset + */ + public RandomDataset randomDataset(Operand seed, Operand seed2, + List> outputTypes, List outputShapes) { + return RandomDataset.create(scope, seed, seed2, outputTypes, outputShapes); + } + + /** + * Creates a dataset that changes the batch size. + * Creates a dataset that changes the batch size of the dataset to current batch + * size // num_replicas. + * + * @param inputDataset A variant tensor representing the input dataset. + * @param numReplicas A scalar representing the number of replicas to distribute this batch across. As + * a result of this transformation the current batch size would end up being + * divided by this parameter. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @param options carries optional attribute values + * @return a new instance of RebatchDataset + */ + public RebatchDataset rebatchDataset(Operand inputDataset, + Operand numReplicas, List> outputTypes, + List outputShapes, RebatchDataset.Options... options) { + return RebatchDataset.create(scope, inputDataset, numReplicas, outputTypes, outputShapes, options); + } + + /** + * Creates a dataset successively reduces {@code f} over the elements of {@code input_dataset}. + * + * @param inputDataset The inputDataset value + * @param initialState The initialState value + * @param otherArguments The otherArguments value + * @param f The value of the f attribute + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @param options carries optional attribute values + * @return a new instance of ScanDataset + */ + public ScanDataset scanDataset(Operand inputDataset, + Iterable> initialState, Iterable> otherArguments, ConcreteFunction f, + List> outputTypes, List outputShapes, + ScanDataset.Options... options) { + return ScanDataset.create(scope, inputDataset, initialState, otherArguments, f, outputTypes, outputShapes, options); + } + + /** + * The ExperimentalSetStatsAggregatorDataset operation + * + * @param inputDataset The inputDataset value + * @param statsAggregator The statsAggregator value + * @param tag The tag value + * @param counterPrefix The counterPrefix value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of SetStatsAggregatorDataset + */ + public SetStatsAggregatorDataset setStatsAggregatorDataset(Operand inputDataset, + Operand statsAggregator, Operand tag, + Operand counterPrefix, List> outputTypes, + List outputShapes) { + return SetStatsAggregatorDataset.create(scope, inputDataset, statsAggregator, tag, counterPrefix, outputTypes, outputShapes); + } + + /** + * The ExperimentalSleepDataset operation + * + * @param inputDataset The inputDataset value + * @param sleepMicroseconds The sleepMicroseconds value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of SleepDataset + */ + public SleepDataset sleepDataset(Operand inputDataset, + Operand sleepMicroseconds, List> outputTypes, + List outputShapes) { + return SleepDataset.create(scope, inputDataset, sleepMicroseconds, outputTypes, outputShapes); + } + + /** + * Creates a dataset that passes a sliding window over {@code input_dataset}. + * + * @param inputDataset The inputDataset value + * @param windowSize A scalar representing the number of elements in the + * sliding window. + * @param windowShift A scalar representing the steps moving the sliding window + * forward in one iteration. It must be positive. + * @param windowStride A scalar representing the stride of the input elements of the sliding window. + * It must be positive. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of SlidingWindowDataset + */ + public SlidingWindowDataset slidingWindowDataset(Operand inputDataset, + Operand windowSize, Operand windowShift, Operand windowStride, + List> outputTypes, List outputShapes) { + return SlidingWindowDataset.create(scope, inputDataset, windowSize, windowShift, windowStride, outputTypes, outputShapes); + } + + /** + * Creates a dataset that executes a SQL query and emits rows of the result set. + * + * @param driverName The database type. Currently, the only supported type is 'sqlite'. + * @param dataSourceName A connection string to connect to the database. + * @param query A SQL query to execute. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of SqlDataset + */ + public SqlDataset sqlDataset(Operand driverName, Operand dataSourceName, + Operand query, List> outputTypes, List outputShapes) { + return SqlDataset.create(scope, driverName, dataSourceName, query, outputTypes, outputShapes); + } + + /** + * Creates a statistics manager resource. + * + * @param options carries optional attribute values + * @return a new instance of StatsAggregatorHandle + */ + public StatsAggregatorHandle statsAggregatorHandle(StatsAggregatorHandle.Options... options) { + return StatsAggregatorHandle.create(scope, options); + } + + /** + * Produces a summary of any statistics recorded by the given statistics manager. + * + * @param iterator The iterator value + * @return a new instance of StatsAggregatorSummary + */ + public StatsAggregatorSummary statsAggregatorSummary(Operand iterator) { + return StatsAggregatorSummary.create(scope, iterator); + } + + /** + * Creates a dataset that stops iteration when predicate` is false. + * The {@code predicate} function must return a scalar boolean and accept the + * following arguments: + *

    + *
  • One tensor for each component of an element of {@code input_dataset}.
    • + *
  • One tensor for each value in {@code other_arguments}.
    • + *
+ * + * @param inputDataset The inputDataset value + * @param otherArguments A list of tensors, typically values that were captured when + * building a closure for {@code predicate}. + * @param predicate A function returning a scalar boolean. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of TakeWhileDataset + */ + public TakeWhileDataset takeWhileDataset(Operand inputDataset, + Iterable> otherArguments, ConcreteFunction predicate, + List> outputTypes, List outputShapes) { + return TakeWhileDataset.create(scope, inputDataset, otherArguments, predicate, outputTypes, outputShapes); + } + + /** + * Creates a dataset that uses a custom thread pool to compute {@code input_dataset}. + * + * @param inputDataset The inputDataset value + * @param threadPool A resource produced by the ThreadPoolHandle op. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of ThreadPoolDataset + */ + public ThreadPoolDataset threadPoolDataset(Operand inputDataset, + Operand threadPool, List> outputTypes, + List outputShapes) { + return ThreadPoolDataset.create(scope, inputDataset, threadPool, outputTypes, outputShapes); + } + + /** + * Creates a dataset that uses a custom thread pool to compute {@code input_dataset}. + * + * @param numThreads The number of threads in the thread pool. + * @param displayName A human-readable name for the threads that may be visible in some + * visualizations. + * threadpool. + * @param options carries optional attribute values + * @return a new instance of ThreadPoolHandle + */ + public ThreadPoolHandle threadPoolHandle(Long numThreads, String displayName, + ThreadPoolHandle.Options... options) { + return ThreadPoolHandle.create(scope, numThreads, displayName, options); + } + + /** + * A dataset that splits the elements of its input into multiple elements. + * + * @param inputDataset The inputDataset value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of UnbatchDataset + */ + public UnbatchDataset unbatchDataset(Operand inputDataset, + List> outputTypes, List outputShapes) { + return UnbatchDataset.create(scope, inputDataset, outputTypes, outputShapes); + } + + /** + * Creates a dataset that contains the unique elements of {@code input_dataset}. + * + * @param inputDataset The inputDataset value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of UniqueDataset + */ + public UniqueDataset uniqueDataset(Operand inputDataset, + List> outputTypes, List outputShapes) { + return UniqueDataset.create(scope, inputDataset, outputTypes, outputShapes); + } + + /** + * Get the parent {@link Ops} object. + */ + public final Ops ops() { + return ops; + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DataOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DataOps.java index 80f23c71450..77b0a424cae 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DataOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DataOps.java @@ -22,9 +22,11 @@ import org.tensorflow.Operand; import org.tensorflow.ndarray.Shape; import org.tensorflow.op.data.AnonymousIterator; +import org.tensorflow.op.data.AnonymousMemoryCache; import org.tensorflow.op.data.AnonymousMultiDeviceIterator; import org.tensorflow.op.data.AssertCardinalityDataset; import org.tensorflow.op.data.AssertNextDataset; +import org.tensorflow.op.data.AssertPrevDataset; import org.tensorflow.op.data.AutoShardDataset; import org.tensorflow.op.data.BatchDataset; import org.tensorflow.op.data.BytesProducedStatsDataset; @@ -32,6 +34,7 @@ import org.tensorflow.op.data.CacheDataset; import org.tensorflow.op.data.ChooseFastestBranchDataset; import org.tensorflow.op.data.ChooseFastestDataset; +import org.tensorflow.op.data.CompressElement; import org.tensorflow.op.data.ConcatenateDataset; import org.tensorflow.op.data.DataServiceDataset; import org.tensorflow.op.data.DatasetCardinality; @@ -41,9 +44,12 @@ import org.tensorflow.op.data.DatasetToSingleElement; import org.tensorflow.op.data.DatasetToTfRecord; import org.tensorflow.op.data.DeleteIterator; +import org.tensorflow.op.data.DeleteMemoryCache; +import org.tensorflow.op.data.DeleteMultiDeviceIterator; import org.tensorflow.op.data.DenseToSparseBatchDataset; import org.tensorflow.op.data.DeserializeIterator; import org.tensorflow.op.data.DirectedInterleaveDataset; +import org.tensorflow.op.data.DummyIterationCounter; import org.tensorflow.op.data.FilterByLastComponentDataset; import org.tensorflow.op.data.FilterDataset; import org.tensorflow.op.data.FinalizeDataset; @@ -56,13 +62,17 @@ import org.tensorflow.op.data.InitializeTableFromDataset; import org.tensorflow.op.data.InterleaveDataset; import org.tensorflow.op.data.Iterator; +import org.tensorflow.op.data.IteratorFromStringHandle; +import org.tensorflow.op.data.IteratorGetDevice; import org.tensorflow.op.data.IteratorGetNext; import org.tensorflow.op.data.IteratorGetNextAsOptional; import org.tensorflow.op.data.IteratorGetNextSync; import org.tensorflow.op.data.IteratorToStringHandle; import org.tensorflow.op.data.LMDBDataset; import org.tensorflow.op.data.LatencyStatsDataset; +import org.tensorflow.op.data.LeakyReluGrad; import org.tensorflow.op.data.LegacyParallelInterleaveDataset; +import org.tensorflow.op.data.ListDataset; import org.tensorflow.op.data.ListSnapshotChunksDataset; import org.tensorflow.op.data.LoadDataset; import org.tensorflow.op.data.MakeIterator; @@ -71,6 +81,11 @@ import org.tensorflow.op.data.MatchingFilesDataset; import org.tensorflow.op.data.MaxIntraOpParallelismDataset; import org.tensorflow.op.data.ModelDataset; +import org.tensorflow.op.data.MultiDeviceIterator; +import org.tensorflow.op.data.MultiDeviceIteratorFromStringHandle; +import org.tensorflow.op.data.MultiDeviceIteratorGetNextFromShard; +import org.tensorflow.op.data.MultiDeviceIteratorInit; +import org.tensorflow.op.data.MultiDeviceIteratorToStringHandle; import org.tensorflow.op.data.NonSerializableDataset; import org.tensorflow.op.data.OneShotIterator; import org.tensorflow.op.data.OptimizeDataset; @@ -81,6 +96,7 @@ import org.tensorflow.op.data.OptionsDataset; import org.tensorflow.op.data.PaddedBatchDataset; import org.tensorflow.op.data.ParallelBatchDataset; +import org.tensorflow.op.data.ParallelFilterDataset; import org.tensorflow.op.data.ParallelInterleaveDataset; import org.tensorflow.op.data.ParallelMapDataset; import org.tensorflow.op.data.ParseExampleDataset; @@ -92,6 +108,7 @@ import org.tensorflow.op.data.ReduceDataset; import org.tensorflow.op.data.RegisterDataset; import org.tensorflow.op.data.RepeatDataset; +import org.tensorflow.op.data.RewriteDataset; import org.tensorflow.op.data.SamplingDataset; import org.tensorflow.op.data.SaveDataset; import org.tensorflow.op.data.ScanDataset; @@ -103,9 +120,14 @@ import org.tensorflow.op.data.SkipDataset; import org.tensorflow.op.data.SleepDataset; import org.tensorflow.op.data.SlidingWindowDataset; +import org.tensorflow.op.data.SnapshotChunkDataset; import org.tensorflow.op.data.SnapshotDataset; +import org.tensorflow.op.data.SnapshotDatasetReader; +import org.tensorflow.op.data.SnapshotNestedDatasetReader; import org.tensorflow.op.data.SparseTensorSliceDataset; import org.tensorflow.op.data.SqlDataset; +import org.tensorflow.op.data.StatsAggregatorHandle; +import org.tensorflow.op.data.StatsAggregatorSetSummaryWriter; import org.tensorflow.op.data.TakeDataset; import org.tensorflow.op.data.TakeWhileDataset; import org.tensorflow.op.data.TensorDataset; @@ -113,14 +135,18 @@ import org.tensorflow.op.data.TextLineDataset; import org.tensorflow.op.data.TfRecordDataset; import org.tensorflow.op.data.ThreadPoolDataset; +import org.tensorflow.op.data.ThreadPoolHandle; import org.tensorflow.op.data.UnbatchDataset; +import org.tensorflow.op.data.UncompressElement; import org.tensorflow.op.data.UniqueDataset; import org.tensorflow.op.data.UnwrapDatasetVariant; import org.tensorflow.op.data.WindowDataset; +import org.tensorflow.op.data.WindowOp; import org.tensorflow.op.data.WrapDatasetVariant; import org.tensorflow.op.data.ZipDataset; import org.tensorflow.types.TBool; import org.tensorflow.types.TFloat32; +import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; import org.tensorflow.types.family.TNumber; @@ -132,6 +158,8 @@ * @see {@link Ops} */ public final class DataOps { + public final DataExperimentalOps experimental; + private final Scope scope; private final Ops ops; @@ -139,6 +167,7 @@ public final class DataOps { DataOps(Ops ops) { this.scope = ops.scope(); this.ops = ops; + experimental = new DataExperimentalOps(ops); } /** @@ -153,6 +182,15 @@ public AnonymousIterator anonymousIterator(List> outputTy return AnonymousIterator.create(scope, outputTypes, outputShapes); } + /** + * The AnonymousMemoryCache operation + * + * @return a new instance of AnonymousMemoryCache + */ + public AnonymousMemoryCache anonymousMemoryCache() { + return AnonymousMemoryCache.create(scope); + } + /** * A container for a multi device iterator resource. * @@ -205,6 +243,30 @@ public AssertNextDataset assertNextDataset(Operand inputDataset return AssertNextDataset.create(scope, inputDataset, transformations, outputTypes, outputShapes); } + /** + * A transformation that asserts which transformations happened previously. + * This transformation checks the names and, optionally, the attribute name-value + * pairs in the {@code transformations} argument against those of the transformations + * that preceded this transformation. If there is a mismatch, the transformation + * raises an exception. + *

The check occurs when iterating over the contents of the dataset, which + * means that the check happens after any static optimizations are applied + * to the dataset graph. + * + * @param inputDataset A variant tensor representing the input dataset. + * {@code data.AssertPrevDataset} passes through the outputs of its input dataset. + * @param transformations A {@code tf.string} vector {@code tf.Tensor} identifying the transformations, with optional + * attribute name-value pairs, that are expected to have happened previously. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of AssertPrevDataset + */ + public AssertPrevDataset assertPrevDataset(Operand inputDataset, + Operand transformations, List> outputTypes, + List outputShapes) { + return AssertPrevDataset.create(scope, inputDataset, transformations, outputTypes, outputShapes); + } + /** * Creates a dataset that shards the input dataset. * Creates a dataset that shards the input dataset by num_workers, returning a @@ -336,6 +398,16 @@ public ChooseFastestDataset chooseFastestDataset(Iterable> components) { + return CompressElement.create(scope, components); + } + /** * Creates a dataset that concatenates {@code input_dataset} with {@code another_dataset}. * @@ -468,6 +540,32 @@ public DeleteIterator deleteIterator(Operand handle, return DeleteIterator.create(scope, handle, deleter); } + /** + * The DeleteMemoryCache operation + * + * @param handle The handle value + * @param deleter The deleter value + * @return a new instance of DeleteMemoryCache + */ + public DeleteMemoryCache deleteMemoryCache(Operand handle, + Operand deleter) { + return DeleteMemoryCache.create(scope, handle, deleter); + } + + /** + * A container for an iterator resource. + * + * @param multiDeviceIterator A handle to the multi device iterator to delete. + * @param iterators A list of iterator handles (unused). This is added so that automatic control dependencies get added during function tracing that ensure this op runs after all the dependent iterators are deleted. + * @param deleter A variant deleter. + * @return a new instance of DeleteMultiDeviceIterator + */ + public DeleteMultiDeviceIterator deleteMultiDeviceIterator( + Operand multiDeviceIterator, Iterable> iterators, + Operand deleter) { + return DeleteMultiDeviceIterator.create(scope, multiDeviceIterator, iterators, deleter); + } + /** * Creates a dataset that batches input elements into a SparseTensor. * @@ -520,6 +618,15 @@ public DirectedInterleaveDataset directedInterleaveDataset( return DirectedInterleaveDataset.create(scope, selectorInputDataset, dataInputDatasets, outputTypes, outputShapes, options); } + /** + * The DummyIterationCounter operation + * + * @return a new instance of DummyIterationCounter + */ + public DummyIterationCounter dummyIterationCounter() { + return DummyIterationCounter.create(scope); + } + /** * Creates a dataset containing elements of first component of {@code input_dataset} having true in the last component. * @@ -765,6 +872,29 @@ public Iterator iterator(String sharedName, String container, return Iterator.create(scope, sharedName, container, outputTypes, outputShapes); } + /** + * The IteratorFromStringHandleV2 operation + * + * @param stringHandle The stringHandle value + * @param outputTypes The value of the outputTypes attribute + * @param options carries optional attribute values + * @return a new instance of IteratorFromStringHandle + */ + public IteratorFromStringHandle iteratorFromStringHandle(Operand stringHandle, + List> outputTypes, IteratorFromStringHandle.Options... options) { + return IteratorFromStringHandle.create(scope, stringHandle, outputTypes, options); + } + + /** + * Returns the name of the device on which {@code resource} has been placed. + * + * @param resource The resource value + * @return a new instance of IteratorGetDevice + */ + public IteratorGetDevice iteratorGetDevice(Operand resource) { + return IteratorGetDevice.create(scope, resource); + } + /** * Gets the next output from the given iterator . * @@ -854,6 +984,22 @@ public LatencyStatsDataset latencyStatsDataset(Operand inputDat return LatencyStatsDataset.create(scope, inputDataset, tag, outputTypes, outputShapes); } + /** + * Computes rectified linear gradients for a LeakyRelu operation. + * + * @param data type for {@code backprops} output + * @param gradients The backpropagated gradients to the corresponding LeakyRelu operation. + * @param features The features passed as input to the corresponding LeakyRelu operation, + * OR the outputs of that operation (both work equivalently). + * @param options carries optional attribute values + * @param data type for {@code LeakyReluGrad} output and operands + * @return a new instance of LeakyReluGrad + */ + public LeakyReluGrad leakyReluGrad(Operand gradients, + Operand features, LeakyReluGrad.Options... options) { + return LeakyReluGrad.create(scope, gradients, features, options); + } + /** * Creates a dataset that applies {@code f} to the outputs of {@code input_dataset}. * The resulting dataset is similar to the {@code InterleaveDataset}, with the exception @@ -886,6 +1032,21 @@ public LegacyParallelInterleaveDataset legacyParallelInterleaveDataset( return LegacyParallelInterleaveDataset.create(scope, inputDataset, otherArguments, cycleLength, blockLength, bufferOutputElements, prefetchInputElements, f, outputTypes, outputShapes, options); } + /** + * Creates a dataset that emits each of {@code tensors} once. + * + * @param tensors The tensors value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @param options carries optional attribute values + * @return a new instance of ListDataset + */ + public ListDataset listDataset(Iterable> tensors, + List> outputTypes, List outputShapes, + ListDataset.Options... options) { + return ListDataset.create(scope, tensors, outputTypes, outputShapes, options); + } + /** * The ListSnapshotChunksDataset operation * @@ -1021,6 +1182,78 @@ public ModelDataset modelDataset(Operand inputDataset, return ModelDataset.create(scope, inputDataset, outputTypes, outputShapes, options); } + /** + * Creates a MultiDeviceIterator resource. + * + * @param devices A list of devices the iterator works across. + * @param sharedName If non-empty, this resource will be shared under the given name + * across multiple sessions. + * @param container If non-empty, this resource is placed in the given container. + * Otherwise, a default container is used. + * @param outputTypes The type list for the return values. + * @param outputShapes The list of shapes being produced. + * @return a new instance of MultiDeviceIterator + */ + public MultiDeviceIterator multiDeviceIterator(List devices, String sharedName, + String container, List> outputTypes, List outputShapes) { + return MultiDeviceIterator.create(scope, devices, sharedName, container, outputTypes, outputShapes); + } + + /** + * Generates a MultiDeviceIterator resource from its provided string handle. + * + * @param stringHandle String representing the resource. + * @param outputTypes The type list for the return values. + * @param options carries optional attribute values + * @return a new instance of MultiDeviceIteratorFromStringHandle + */ + public MultiDeviceIteratorFromStringHandle multiDeviceIteratorFromStringHandle( + Operand stringHandle, List> outputTypes, + MultiDeviceIteratorFromStringHandle.Options... options) { + return MultiDeviceIteratorFromStringHandle.create(scope, stringHandle, outputTypes, options); + } + + /** + * Gets next element for the provided shard number. + * + * @param multiDeviceIterator A MultiDeviceIterator resource. + * @param shardNum Integer representing which shard to fetch data for. + * @param incarnationId Which incarnation of the MultiDeviceIterator is running. + * @param outputTypes The type list for the return values. + * @param outputShapes The list of shapes being produced. + * @return a new instance of MultiDeviceIteratorGetNextFromShard + */ + public MultiDeviceIteratorGetNextFromShard multiDeviceIteratorGetNextFromShard( + Operand multiDeviceIterator, Operand shardNum, + Operand incarnationId, List> outputTypes, + List outputShapes) { + return MultiDeviceIteratorGetNextFromShard.create(scope, multiDeviceIterator, shardNum, incarnationId, outputTypes, outputShapes); + } + + /** + * Initializes the multi device iterator with the given dataset. + * + * @param dataset Dataset to be iterated upon. + * @param multiDeviceIterator A MultiDeviceIteratorResource. + * @param maxBufferSize The maximum size of the host side per device buffer to keep. + * @return a new instance of MultiDeviceIteratorInit + */ + public MultiDeviceIteratorInit multiDeviceIteratorInit(Operand dataset, + Operand multiDeviceIterator, Operand maxBufferSize) { + return MultiDeviceIteratorInit.create(scope, dataset, multiDeviceIterator, maxBufferSize); + } + + /** + * Produces a string handle for the given MultiDeviceIterator. + * + * @param multiDeviceIterator A MultiDeviceIterator resource. + * @return a new instance of MultiDeviceIteratorToStringHandle + */ + public MultiDeviceIteratorToStringHandle multiDeviceIteratorToStringHandle( + Operand multiDeviceIterator) { + return MultiDeviceIteratorToStringHandle.create(scope, multiDeviceIterator); + } + /** * The NonSerializableDataset operation * @@ -1188,6 +1421,35 @@ public ParallelBatchDataset parallelBatchDataset(Operand inputD return ParallelBatchDataset.create(scope, inputDataset, batchSize, numParallelCalls, dropRemainder, outputTypes, outputShapes, options); } + /** + * Creates a dataset containing elements of {@code input_dataset} matching {@code predicate}. + * The {@code predicate} function must return a scalar boolean and accept the + * following arguments: + *

    + *
  • One tensor for each component of an element of {@code input_dataset}.
    • + *
  • One tensor for each value in {@code other_arguments}.
    • + *
+ *

Unlike a "FilterDataset", which applies {@code predicate} sequentially, this dataset + * invokes up to {@code num_parallel_calls} copies of {@code predicate} in parallel. + * + * @param inputDataset The inputDataset value + * @param otherArguments A list of tensors, typically values that were captured when + * building a closure for {@code predicate}. + * @param numParallelCalls The number of concurrent invocations of {@code predicate} that process + * elements from {@code input_dataset} in parallel. + * @param predicate A function returning a scalar boolean. + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @param options carries optional attribute values + * @return a new instance of ParallelFilterDataset + */ + public ParallelFilterDataset parallelFilterDataset(Operand inputDataset, + Iterable> otherArguments, Operand numParallelCalls, + ConcreteFunction predicate, List> outputTypes, + List outputShapes, ParallelFilterDataset.Options... options) { + return ParallelFilterDataset.create(scope, inputDataset, otherArguments, numParallelCalls, predicate, outputTypes, outputShapes, options); + } + /** * Creates a dataset that applies {@code f} to the outputs of {@code input_dataset}. * The resulting dataset is similar to the {@code InterleaveDataset}, except that the @@ -1445,6 +1707,21 @@ public RepeatDataset repeatDataset(Operand inputDataset, Operan return RepeatDataset.create(scope, inputDataset, count, outputTypes, outputShapes, options); } + /** + * The RewriteDataset operation + * + * @param inputDataset The inputDataset value + * @param rewriteName The rewriteName value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of RewriteDataset + */ + public RewriteDataset rewriteDataset(Operand inputDataset, + Operand rewriteName, List> outputTypes, + List outputShapes) { + return RewriteDataset.create(scope, inputDataset, rewriteName, outputTypes, outputShapes); + } + /** * Creates a dataset that takes a Bernoulli sample of the contents of another dataset. * There is no transformation in the {@code tf.data} Python API for creating this dataset. @@ -1649,6 +1926,21 @@ public SlidingWindowDataset slidingWindowDataset(Operand inputD return SlidingWindowDataset.create(scope, inputDataset, windowSize, windowShift, windowStride, outputTypes, outputShapes, options); } + /** + * The SnapshotChunkDataset operation + * + * @param chunkFile The chunkFile value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @param options carries optional attribute values + * @return a new instance of SnapshotChunkDataset + */ + public SnapshotChunkDataset snapshotChunkDataset(Operand chunkFile, + List> outputTypes, List outputShapes, + SnapshotChunkDataset.Options... options) { + return SnapshotChunkDataset.create(scope, chunkFile, outputTypes, outputShapes, options); + } + /** * Creates a dataset that will write to / read from a snapshot. * This dataset attempts to determine whether a valid snapshot exists at the @@ -1675,6 +1967,37 @@ public SnapshotDataset snapshotDataset(Operand inputDataset, return SnapshotDataset.create(scope, inputDataset, path, readerFuncOtherArgs, shardFuncOtherArgs, outputTypes, outputShapes, readerFunc, shardFunc, options); } + /** + * The SnapshotDatasetReader operation + * + * @param shardDir The shardDir value + * @param startIndex The startIndex value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @param version The value of the version attribute + * @param options carries optional attribute values + * @return a new instance of SnapshotDatasetReader + */ + public SnapshotDatasetReader snapshotDatasetReader(Operand shardDir, + Operand startIndex, List> outputTypes, + List outputShapes, Long version, SnapshotDatasetReader.Options... options) { + return SnapshotDatasetReader.create(scope, shardDir, startIndex, outputTypes, outputShapes, version, options); + } + + /** + * The SnapshotNestedDatasetReader operation + * + * @param inputs The inputs value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of SnapshotNestedDatasetReader + */ + public SnapshotNestedDatasetReader snapshotNestedDatasetReader( + Iterable> inputs, List> outputTypes, + List outputShapes) { + return SnapshotNestedDatasetReader.create(scope, inputs, outputTypes, outputShapes); + } + /** * Creates a dataset that splits a SparseTensor into elements row-wise. * @@ -1703,6 +2026,28 @@ public SqlDataset sqlDataset(Operand driverName, Operand dataS return SqlDataset.create(scope, driverName, dataSourceName, query, outputTypes, outputShapes); } + /** + * The StatsAggregatorHandleV2 operation + * + * @param options carries optional attribute values + * @return a new instance of StatsAggregatorHandle + */ + public StatsAggregatorHandle statsAggregatorHandle(StatsAggregatorHandle.Options... options) { + return StatsAggregatorHandle.create(scope, options); + } + + /** + * Set a summary_writer_interface to record statistics using given stats_aggregator. + * + * @param statsAggregator The statsAggregator value + * @param summary The summary value + * @return a new instance of StatsAggregatorSetSummaryWriter + */ + public StatsAggregatorSetSummaryWriter statsAggregatorSetSummaryWriter( + Operand statsAggregator, Operand summary) { + return StatsAggregatorSetSummaryWriter.create(scope, statsAggregator, summary); + } + /** * Creates a dataset that contains {@code count} elements from the {@code input_dataset}. * @@ -1824,6 +2169,21 @@ public ThreadPoolDataset threadPoolDataset(Operand inputDataset return ThreadPoolDataset.create(scope, inputDataset, threadPool, outputTypes, outputShapes); } + /** + * Creates a dataset that uses a custom thread pool to compute {@code input_dataset}. + * + * @param numThreads The number of threads in the thread pool. + * @param displayName A human-readable name for the threads that may be visible in some + * visualizations. + * threadpool. + * @param options carries optional attribute values + * @return a new instance of ThreadPoolHandle + */ + public ThreadPoolHandle threadPoolHandle(Long numThreads, String displayName, + ThreadPoolHandle.Options... options) { + return ThreadPoolHandle.create(scope, numThreads, displayName, options); + } + /** * A dataset that splits the elements of its input into multiple elements. * @@ -1839,6 +2199,19 @@ public UnbatchDataset unbatchDataset(Operand inputDataset, return UnbatchDataset.create(scope, inputDataset, outputTypes, outputShapes, options); } + /** + * Uncompresses a compressed dataset element. + * + * @param compressed The compressed value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of UncompressElement + */ + public UncompressElement uncompressElement(Operand compressed, + List> outputTypes, List outputShapes) { + return UncompressElement.create(scope, compressed, outputTypes, outputShapes); + } + /** * Creates a dataset that contains the unique elements of {@code input_dataset}. * @@ -1925,6 +2298,19 @@ public WindowDataset windowDataset(Operand inputDataset, return WindowDataset.create(scope, inputDataset, sizeOutput, shift, stride, dropRemainder, outputTypes, outputShapes, options); } + /** + * The WindowOp operation + * + * @param inputs The inputs value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of WindowOp + */ + public WindowOp windowOp(Iterable> inputs, List> outputTypes, + List outputShapes) { + return WindowOp.create(scope, inputs, outputTypes, outputShapes); + } + /** * The WrapDatasetVariant operation * diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DebuggingOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DebuggingOps.java new file mode 100644 index 00000000000..289a355dea3 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DebuggingOps.java @@ -0,0 +1,62 @@ +// Copyright 2020-2022 The TensorFlow Authors. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// ============================================================================== +// +// This class has been generated, DO NOT EDIT! +// +package org.tensorflow.op; + +import org.tensorflow.Operand; +import org.tensorflow.op.debugging.CheckNumerics; +import org.tensorflow.types.family.TNumber; + +/** + * An API for building {@code debugging} operations as {@link Op Op}s + * + * @see {@link Ops} + */ +public final class DebuggingOps { + private final Scope scope; + + private final Ops ops; + + DebuggingOps(Ops ops) { + this.scope = ops.scope(); + this.ops = ops; + } + + /** + * Checks a tensor for NaN, -Inf and +Inf values. + * When run, reports an {@code InvalidArgument} error if {@code tensor} has any values + * that are not a number (NaN) or infinity (Inf). Otherwise, returns the input + * tensor. Unlike CheckNumerics (V1), CheckNumericsV2 distinguishes -Inf and +Inf + * in the errors it throws. + * + * @param data type for {@code output} output + * @param tensor The tensor value + * @param message Prefix of the error message. + * @param data type for {@code CheckNumericsV2} output and operands + * @return a new instance of CheckNumerics + */ + public CheckNumerics checkNumerics(Operand tensor, String message) { + return CheckNumerics.create(scope, tensor, message); + } + + /** + * Get the parent {@link Ops} object. + */ + public final Ops ops() { + return ops; + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DistributeOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DistributeOps.java new file mode 100644 index 00000000000..50f5a2aa63f --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DistributeOps.java @@ -0,0 +1,113 @@ +// Copyright 2020-2022 The TensorFlow Authors. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// ============================================================================== +// +// This class has been generated, DO NOT EDIT! +// +package org.tensorflow.op; + +import org.tensorflow.Operand; +import org.tensorflow.ndarray.Shape; +import org.tensorflow.op.distribute.NcclAllReduce; +import org.tensorflow.op.distribute.NcclBroadcast; +import org.tensorflow.op.distribute.NcclReduce; +import org.tensorflow.types.family.TNumber; + +/** + * An API for building {@code distribute} operations as {@link Op Op}s + * + * @see {@link Ops} + */ +public final class DistributeOps { + private final Scope scope; + + private final Ops ops; + + DistributeOps(Ops ops) { + this.scope = ops.scope(); + this.ops = ops; + } + + /** + * Outputs a tensor containing the reduction across all input tensors. + * Outputs a tensor containing the reduction across all input tensors passed to ops + * within the same `shared_name. + *

The graph should be constructed so if one op runs with shared_name value {@code c}, + * then {@code num_devices} ops will run with shared_name value {@code c}. Failure to do so + * will cause the graph execution to fail to complete. + *

input: the input to the reduction + * data: the value of the reduction across all {@code num_devices} devices. + * reduction: the reduction operation to perform. + * num_devices: The number of devices participating in this reduction. + * shared_name: Identifier that shared between ops of the same reduction. + * + * @param data type for {@code data} output + * @param input The input value + * @param reduction The value of the reduction attribute + * @param numDevices The value of the numDevices attribute + * @param sharedName The value of the sharedName attribute + * @param data type for {@code NcclAllReduce} output and operands + * @return a new instance of NcclAllReduce + */ + public NcclAllReduce ncclAllReduce(Operand input, String reduction, + Long numDevices, String sharedName) { + return NcclAllReduce.create(scope, input, reduction, numDevices, sharedName); + } + + /** + * Sends {@code input} to all devices that are connected to the output. + * Sends {@code input} to all devices that are connected to the output. + *

The graph should be constructed so that all ops connected to the output have a + * valid device assignment, and the op itself is assigned one of these devices. + *

input: The input to the broadcast. + * output: The same as input. + * shape: The shape of the input tensor. + * + * @param data type for {@code output} output + * @param input The input value + * @param shape The value of the shape attribute + * @param data type for {@code NcclBroadcast} output and operands + * @return a new instance of NcclBroadcast + */ + public NcclBroadcast ncclBroadcast(Operand input, Shape shape) { + return NcclBroadcast.create(scope, input, shape); + } + + /** + * Reduces {@code input} from {@code num_devices} using {@code reduction} to a single device. + * Reduces {@code input} from {@code num_devices} using {@code reduction} to a single device. + *

The graph should be constructed so that all inputs have a valid device + * assignment, and the op itself is assigned one of these devices. + *

input: The input to the reduction. + * data: the value of the reduction across all {@code num_devices} devices. + * reduction: the reduction operation to perform. + * + * @param data type for {@code data} output + * @param input The input value + * @param reduction The value of the reduction attribute + * @param data type for {@code NcclReduce} output and operands + * @return a new instance of NcclReduce + */ + public NcclReduce ncclReduce(Iterable> input, + String reduction) { + return NcclReduce.create(scope, input, reduction); + } + + /** + * Get the parent {@link Ops} object. + */ + public final Ops ops() { + return ops; + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DtypesOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DtypesOps.java index 319ee68d295..8304e333023 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DtypesOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/DtypesOps.java @@ -21,6 +21,7 @@ import org.tensorflow.op.dtypes.AsString; import org.tensorflow.op.dtypes.Cast; import org.tensorflow.op.dtypes.Complex; +import org.tensorflow.op.dtypes.ToBool; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -107,6 +108,31 @@ public Complex complex(Operand real, return Complex.create(scope, real, imag, Tout); } + /** + * Converts a tensor to a scalar predicate. + * Converts a tensor to a scalar predicate with the following rules: + *

    + *
  • + *

    For 0D tensors, truthiness is determined by comparing against a "zero" + * value. For numerical types it is the obvious zero. For strings it is the + * empty string. + *

    • + *
  • + *

    For >0D tensors, truthiness is determined by looking at the number of + * elements. If has zero elements, then the result is false. Otherwise the + * result is true. + *

    • + *
+ *

This matches the behavior of If and While for determining if a tensor counts + * as true/false for a branch condition. + * + * @param input The input value + * @return a new instance of ToBool + */ + public ToBool toBool(Operand input) { + return ToBool.create(scope, input); + } + /** * Get the parent {@link Ops} object. */ diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/ImageOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/ImageOps.java index 3f93386ebb3..801e020d814 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/ImageOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/ImageOps.java @@ -36,20 +36,29 @@ import org.tensorflow.op.image.EncodeJpeg; import org.tensorflow.op.image.EncodeJpegVariableQuality; import org.tensorflow.op.image.EncodePng; +import org.tensorflow.op.image.ExtractGlimpse; import org.tensorflow.op.image.ExtractImagePatches; import org.tensorflow.op.image.ExtractJpegShape; +import org.tensorflow.op.image.GenerateBoundingBoxProposals; import org.tensorflow.op.image.HsvToRgb; +import org.tensorflow.op.image.ImageProjectiveTransformV2; +import org.tensorflow.op.image.ImageProjectiveTransformV3; +import org.tensorflow.op.image.NearestNeighbors; import org.tensorflow.op.image.NonMaxSuppression; import org.tensorflow.op.image.NonMaxSuppressionWithOverlaps; import org.tensorflow.op.image.QuantizedResizeBilinear; import org.tensorflow.op.image.RandomCrop; import org.tensorflow.op.image.ResizeArea; import org.tensorflow.op.image.ResizeBicubic; +import org.tensorflow.op.image.ResizeBicubicGrad; import org.tensorflow.op.image.ResizeBilinear; +import org.tensorflow.op.image.ResizeBilinearGrad; import org.tensorflow.op.image.ResizeNearestNeighbor; +import org.tensorflow.op.image.ResizeNearestNeighborGrad; import org.tensorflow.op.image.RgbToHsv; import org.tensorflow.op.image.SampleDistortedBoundingBox; import org.tensorflow.op.image.ScaleAndTranslate; +import org.tensorflow.op.image.ScaleAndTranslateGrad; import org.tensorflow.op.image.StatelessSampleDistortedBoundingBox; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -554,6 +563,42 @@ public EncodePng encodePng(Operand image, EncodePng.Options.. return EncodePng.create(scope, image, options); } + /** + * Extracts a glimpse from the input tensor. + * Returns a set of windows called glimpses extracted at location + * {@code offsets} from the input tensor. If the windows only partially + * overlaps the inputs, the non overlapping areas will be filled with + * random noise. + *

The result is a 4-D tensor of shape {@code [batch_size, glimpse_height, glimpse_width, channels]}. The channels and batch dimensions are the + * same as that of the input tensor. The height and width of the output + * windows are specified in the {@code size} parameter. + *

The argument {@code normalized} and {@code centered} controls how the windows are built: + *

    + *
  • If the coordinates are normalized but not centered, 0.0 and 1.0 + * correspond to the minimum and maximum of each height and width + * dimension.
    • + *
  • If the coordinates are both normalized and centered, they range from + * -1.0 to 1.0. The coordinates (-1.0, -1.0) correspond to the upper + * left corner, the lower right corner is located at (1.0, 1.0) and the + * center is at (0, 0).
    • + *
  • If the coordinates are not normalized they are interpreted as + * numbers of pixels.
    • + *
+ * + * @param input A 4-D float tensor of shape {@code [batch_size, height, width, channels]}. + * @param sizeOutput A 1-D tensor of 2 elements containing the size of the glimpses + * to extract. The glimpse height must be specified first, following + * by the glimpse width. + * @param offsets A 2-D integer tensor of shape {@code [batch_size, 2]} containing + * the y, x locations of the center of each window. + * @param options carries optional attribute values + * @return a new instance of ExtractGlimpse + */ + public ExtractGlimpse extractGlimpse(Operand input, Operand sizeOutput, + Operand offsets, ExtractGlimpse.Options... options) { + return ExtractGlimpse.create(scope, input, sizeOutput, offsets, options); + } + /** * Extract {@code patches} from {@code images} and put them in the "depth" output dimension. * @@ -605,6 +650,40 @@ public ExtractJpegShape extractJpegShape(Operand return ExtractJpegShape.create(scope, contents, outputType); } + /** + * This op produces Region of Interests from given bounding boxes(bbox_deltas) encoded wrt anchors according to eq.2 in arXiv:1506.01497 + * 
+   *    The op selects top `pre_nms_topn` scoring boxes, decodes them with respect to anchors,
+   *    applies non-maximal suppression on overlapping boxes with higher than
+   *    `nms_threshold` intersection-over-union (iou) value, discarding boxes where shorter
+   *    side is less than `min_size`.
+   *    Inputs:
+   *    `scores`: A 4D tensor of shape [Batch, Height, Width, Num Anchors] containing the scores per anchor at given position
+   *    `bbox_deltas`: is a tensor of shape [Batch, Height, Width, 4 x Num Anchors] boxes encoded to each anchor
+   *    `anchors`: A 1D tensor of shape [4 x Num Anchors], representing the anchors.
+   *    Outputs:
+   *    `rois`: output RoIs, a 3D tensor of shape [Batch, post_nms_topn, 4], padded by 0 if less than post_nms_topn candidates found.
+   *    `roi_probabilities`: probability scores of each roi in 'rois', a 2D tensor of shape [Batch,post_nms_topn], padded with 0 if needed, sorted by scores.
+   *
+ * + * @param scores A 4-D float tensor of shape {@code [num_images, height, width, num_achors]} containing scores of the boxes for given anchors, can be unsorted. + * @param bboxDeltas A 4-D float tensor of shape {@code [num_images, height, width, 4 x num_anchors]}. encoding boxes with respec to each anchor. + * Coordinates are given in the form [dy, dx, dh, dw]. + * @param imageInfo A 2-D float tensor of shape {@code [num_images, 5]} containing image information Height, Width, Scale. + * @param anchors A 2-D float tensor of shape {@code [num_anchors, 4]} describing the anchor boxes. Boxes are formatted in the form [y1, x1, y2, x2]. + * @param nmsThreshold A scalar float tensor for non-maximal-suppression threshold. + * @param preNmsTopn A scalar int tensor for the number of top scoring boxes to be used as input. + * @param minSize A scalar float tensor. Any box that has a smaller size than min_size will be discarded. + * @param options carries optional attribute values + * @return a new instance of GenerateBoundingBoxProposals + */ + public GenerateBoundingBoxProposals generateBoundingBoxProposals(Operand scores, + Operand bboxDeltas, Operand imageInfo, Operand anchors, + Operand nmsThreshold, Operand preNmsTopn, Operand minSize, + GenerateBoundingBoxProposals.Options... options) { + return GenerateBoundingBoxProposals.create(scope, scores, bboxDeltas, imageInfo, anchors, nmsThreshold, preNmsTopn, minSize, options); + } + /** * Convert one or more images from HSV to RGB. * Outputs a tensor of the same shape as the {@code images} tensor, containing the RGB @@ -621,6 +700,75 @@ public HsvToRgb hsvToRgb(Operand images) { return HsvToRgb.create(scope, images); } + /** + * Applies the given transform to each of the images. + * If one row of {@code transforms} is {@code [a0, a1, a2, b0, b1, b2, c0, c1]}, then it maps + * the output point {@code (x, y)} to a transformed input point + * {@code (x', y') = ((a0 x + a1 y + a2) / k, (b0 x + b1 y + b2) / k)}, where + * {@code k = c0 x + c1 y + 1}. If the transformed point lays outside of the input + * image, the output pixel is set to 0. + * + * @param data type for {@code transformed_images} output + * @param images 4-D with shape {@code [batch, height, width, channels]}. + * @param transforms 2-D Tensor, {@code [batch, 8]} or {@code [1, 8]} matrix, where each row corresponds to a 3 x 3 + * projective transformation matrix, with the last entry assumed to be 1. If there + * is one row, the same transformation will be applied to all images. + * @param outputShape 1-D Tensor [new_height, new_width]. + * @param interpolation Interpolation method, "NEAREST" or "BILINEAR". + * @param options carries optional attribute values + * @param data type for {@code ImageProjectiveTransformV2} output and operands + * @return a new instance of ImageProjectiveTransformV2 + */ + public ImageProjectiveTransformV2 imageProjectiveTransformV2( + Operand images, Operand transforms, Operand outputShape, + String interpolation, ImageProjectiveTransformV2.Options... options) { + return ImageProjectiveTransformV2.create(scope, images, transforms, outputShape, interpolation, options); + } + + /** + * Applies the given transform to each of the images. + * If one row of {@code transforms} is {@code [a0, a1, a2, b0, b1, b2, c0, c1]}, then it maps + * the output point {@code (x, y)} to a transformed input point + * {@code (x', y') = ((a0 x + a1 y + a2) / k, (b0 x + b1 y + b2) / k)}, where + * {@code k = c0 x + c1 y + 1}. If the transformed point lays outside of the input + * image, the output pixel is set to fill_value. + * + * @param data type for {@code transformed_images} output + * @param images 4-D with shape {@code [batch, height, width, channels]}. + * @param transforms 2-D Tensor, {@code [batch, 8]} or {@code [1, 8]} matrix, where each row corresponds to a 3 x 3 + * projective transformation matrix, with the last entry assumed to be 1. If there + * is one row, the same transformation will be applied to all images. + * @param outputShape 1-D Tensor [new_height, new_width]. + * @param fillValue float, the value to be filled when fill_mode is constant". + * @param interpolation Interpolation method, "NEAREST" or "BILINEAR". + * @param options carries optional attribute values + * @param data type for {@code ImageProjectiveTransformV3} output and operands + * @return a new instance of ImageProjectiveTransformV3 + */ + public ImageProjectiveTransformV3 imageProjectiveTransformV3( + Operand images, Operand transforms, Operand outputShape, + Operand fillValue, String interpolation, + ImageProjectiveTransformV3.Options... options) { + return ImageProjectiveTransformV3.create(scope, images, transforms, outputShape, fillValue, interpolation, options); + } + + /** + * Selects the k nearest centers for each point. + * Rows of points are assumed to be input points. Rows of centers are assumed to be + * the list of candidate centers. For each point, the k centers that have least L2 + * distance to it are computed. + * + * @param points Matrix of shape (n, d). Rows are assumed to be input points. + * @param centers Matrix of shape (m, d). Rows are assumed to be centers. + * @param k Number of nearest centers to return for each point. If k is larger than m, then + * only m centers are returned. + * @return a new instance of NearestNeighbors + */ + public NearestNeighbors nearestNeighbors(Operand points, Operand centers, + Operand k) { + return NearestNeighbors.create(scope, points, centers, k); + } + /** * Greedily selects a subset of bounding boxes in descending order of score, * pruning away boxes that have high intersection-over-union (IOU) overlap @@ -780,6 +928,22 @@ public ResizeBicubic resizeBicubic(Operand images, Operand data type for {@code output} output + * @param grads 4-D with shape {@code [batch, height, width, channels]}. + * @param originalImage 4-D with shape {@code [batch, orig_height, orig_width, channels]}, + * The image tensor that was resized. + * @param options carries optional attribute values + * @param data type for {@code ResizeBicubicGrad} output and operands + * @return a new instance of ResizeBicubicGrad + */ + public ResizeBicubicGrad resizeBicubicGrad(Operand grads, + Operand originalImage, ResizeBicubicGrad.Options... options) { + return ResizeBicubicGrad.create(scope, grads, originalImage, options); + } + /** * Resize {@code images} to {@code size} using bilinear interpolation. * Input images can be of different types but output images are always float. @@ -795,6 +959,22 @@ public ResizeBilinear resizeBilinear(Operand images, return ResizeBilinear.create(scope, images, sizeOutput, options); } + /** + * Computes the gradient of bilinear interpolation. + * + * @param data type for {@code output} output + * @param grads 4-D with shape {@code [batch, height, width, channels]}. + * @param originalImage 4-D with shape {@code [batch, orig_height, orig_width, channels]}, + * The image tensor that was resized. + * @param options carries optional attribute values + * @param data type for {@code ResizeBilinearGrad} output and operands + * @return a new instance of ResizeBilinearGrad + */ + public ResizeBilinearGrad resizeBilinearGrad(Operand grads, + Operand originalImage, ResizeBilinearGrad.Options... options) { + return ResizeBilinearGrad.create(scope, grads, originalImage, options); + } + /** * Resize {@code images} to {@code size} using nearest neighbor interpolation. * @@ -811,6 +991,22 @@ public ResizeNearestNeighbor resizeNearestNeighbor(Operan return ResizeNearestNeighbor.create(scope, images, sizeOutput, options); } + /** + * Computes the gradient of nearest neighbor interpolation. + * + * @param data type for {@code output} output + * @param grads 4-D with shape {@code [batch, height, width, channels]}. + * @param sizeOutput = A 1-D int32 Tensor of 2 elements: {@code orig_height, orig_width}. The + * original input size. + * @param options carries optional attribute values + * @param data type for {@code ResizeNearestNeighborGrad} output and operands + * @return a new instance of ResizeNearestNeighborGrad + */ + public ResizeNearestNeighborGrad resizeNearestNeighborGrad( + Operand grads, Operand sizeOutput, ResizeNearestNeighborGrad.Options... options) { + return ResizeNearestNeighborGrad.create(scope, grads, sizeOutput, options); + } + /** * Converts one or more images from RGB to HSV. * Outputs a tensor of the same shape as the {@code images} tensor, containing the HSV @@ -914,6 +1110,24 @@ public ScaleAndTranslate scaleAndTranslate(Operand images, return ScaleAndTranslate.create(scope, images, sizeOutput, scale, translation, options); } + /** + * The ScaleAndTranslateGrad operation + * + * @param data type for {@code output} output + * @param grads The grads value + * @param originalImage The originalImage value + * @param scale The scale value + * @param translation The translation value + * @param options carries optional attribute values + * @param data type for {@code ScaleAndTranslateGrad} output and operands + * @return a new instance of ScaleAndTranslateGrad + */ + public ScaleAndTranslateGrad scaleAndTranslateGrad(Operand grads, + Operand originalImage, Operand scale, Operand translation, + ScaleAndTranslateGrad.Options... options) { + return ScaleAndTranslateGrad.create(scope, grads, originalImage, scale, translation, options); + } + /** * Generate a randomly distorted bounding box for an image deterministically. * Bounding box annotations are often supplied in addition to ground-truth labels diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/IoOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/IoOps.java index 303b4d5d618..4f33bea6a52 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/IoOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/IoOps.java @@ -27,7 +27,9 @@ import org.tensorflow.op.io.DecodePaddedRaw; import org.tensorflow.op.io.DecodeRaw; import org.tensorflow.op.io.DeserializeManySparse; +import org.tensorflow.op.io.DisableCopyOnRead; import org.tensorflow.op.io.EncodeBase64; +import org.tensorflow.op.io.FakeQueue; import org.tensorflow.op.io.FifoQueue; import org.tensorflow.op.io.FixedLengthRecordReader; import org.tensorflow.op.io.IdentityReader; @@ -241,6 +243,17 @@ public DeserializeManySparse deserializeManySparse( return DeserializeManySparse.create(scope, serializedSparse, dtype); } + /** + * Turns off the copy-on-read mode. + * Turns off the copy-on-read mode of a resource variable. If the variable is not in copy-on-read mode, this op has no effect. + * + * @param resource The resource handle of the resource variable. + * @return a new instance of DisableCopyOnRead + */ + public DisableCopyOnRead disableCopyOnRead(Operand resource) { + return DisableCopyOnRead.create(scope, resource); + } + /** * Encode strings into web-safe base64 format. * Refer to this article for more information on @@ -257,6 +270,16 @@ public EncodeBase64 encodeBase64(Operand input, EncodeBase64.Options... return EncodeBase64.create(scope, input, options); } + /** + * Deprecated. Do not use. + * + * @param resource The resource value + * @return a new instance of FakeQueue + */ + public FakeQueue fakeQueue(Operand resource) { + return FakeQueue.create(scope, resource); + } + /** * A queue that produces elements in first-in first-out order. * diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/LinalgOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/LinalgOps.java index b1882373d4b..9a6e88ddcef 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/LinalgOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/LinalgOps.java @@ -19,6 +19,7 @@ import org.tensorflow.Operand; import org.tensorflow.op.linalg.BandPart; +import org.tensorflow.op.linalg.BandedTriangularSolve; import org.tensorflow.op.linalg.BatchCholesky; import org.tensorflow.op.linalg.BatchCholeskyGrad; import org.tensorflow.op.linalg.BatchMatrixBandPart; @@ -49,10 +50,15 @@ import org.tensorflow.op.linalg.MatrixDiagPart; import org.tensorflow.op.linalg.MatrixDiagPartV3; import org.tensorflow.op.linalg.MatrixDiagV3; +import org.tensorflow.op.linalg.MatrixExponential; +import org.tensorflow.op.linalg.MatrixLogarithm; import org.tensorflow.op.linalg.MatrixSetDiag; import org.tensorflow.op.linalg.MatrixSolveLs; import org.tensorflow.op.linalg.Qr; import org.tensorflow.op.linalg.QuantizedMatMul; +import org.tensorflow.op.linalg.QuantizedMatMulWithBias; +import org.tensorflow.op.linalg.QuantizedMatMulWithBiasAndRelu; +import org.tensorflow.op.linalg.QuantizedMatMulWithBiasAndReluAndRequantize; import org.tensorflow.op.linalg.SelfAdjointEig; import org.tensorflow.op.linalg.Solve; import org.tensorflow.op.linalg.Sqrtm; @@ -61,6 +67,8 @@ import org.tensorflow.op.linalg.TensorDiagPart; import org.tensorflow.op.linalg.Transpose; import org.tensorflow.op.linalg.TriangularSolve; +import org.tensorflow.op.linalg.TridiagonalMatMul; +import org.tensorflow.op.linalg.TridiagonalSolve; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TFloat64; import org.tensorflow.types.TInt32; @@ -75,6 +83,8 @@ * @see {@link Ops} */ public final class LinalgOps { + public final LinalgSparseOps sparse; + private final Scope scope; private final Ops ops; @@ -82,6 +92,7 @@ public final class LinalgOps { LinalgOps(Ops ops) { this.scope = ops.scope(); this.ops = ops; + sparse = new LinalgSparseOps(ops); } /** @@ -131,6 +142,21 @@ public BandPart bandPart(Operand inpu return BandPart.create(scope, input, numLower, numUpper); } + /** + * The BandedTriangularSolve operation + * + * @param data type for {@code output} output + * @param matrix The matrix value + * @param rhs The rhs value + * @param options carries optional attribute values + * @param data type for {@code BandedTriangularSolve} output and operands + * @return a new instance of BandedTriangularSolve + */ + public BandedTriangularSolve bandedTriangularSolve(Operand matrix, + Operand rhs, BandedTriangularSolve.Options... options) { + return BandedTriangularSolve.create(scope, matrix, rhs, options); + } + /** * The BatchCholesky operation * @@ -1121,6 +1147,41 @@ public MatrixDiagV3 matrixDiagV3(Operand diagonal, Opera return MatrixDiagV3.create(scope, diagonal, k, numRows, numCols, paddingValue, options); } + /** + * Deprecated, use python implementation tf.linalg.matrix_exponential. + * + * @param data type for {@code output} output + * @param input The input value + * @param data type for {@code MatrixExponential} output and operands + * @return a new instance of MatrixExponential + */ + public MatrixExponential matrixExponential(Operand input) { + return MatrixExponential.create(scope, input); + } + + /** + * Computes the matrix logarithm of one or more square matrices: + * \(log(exp(A)) = A\) + *

This op is only defined for complex matrices. If A is positive-definite and + * real, then casting to a complex matrix, taking the logarithm and casting back + * to a real matrix will give the correct result. + *

This function computes the matrix logarithm using the Schur-Parlett algorithm. + * Details of the algorithm can be found in Section 11.6.2 of: + * Nicholas J. Higham, Functions of Matrices: Theory and Computation, SIAM 2008. + * ISBN 978-0-898716-46-7. + *

The input is a tensor of shape {@code [..., M, M]} whose inner-most 2 dimensions + * form square matrices. The output is a tensor of the same shape as the input + * containing the exponential for all input submatrices {@code [..., :, :]}. + * + * @param data type for {@code output} output + * @param input Shape is {@code [..., M, M]}. + * @param data type for {@code MatrixLogarithm} output and operands + * @return a new instance of MatrixLogarithm + */ + public MatrixLogarithm matrixLogarithm(Operand input) { + return MatrixLogarithm.create(scope, input); + } + /** * Returns a batched matrix tensor with new batched diagonal values. * Given {@code input} and {@code diagonal}, this operation returns a tensor with the @@ -1341,6 +1402,103 @@ public QuantizedMatMul quantizedMatMul return QuantizedMatMul.create(scope, a, b, minA, maxA, minB, maxB, Toutput, Tactivation, options); } + /** + * Performs a quantized matrix multiplication of {@code a} by the matrix {@code b} with bias + * add. + * The inputs must be two-dimensional matrices and 1D bias vector. And the inner + * dimension of {@code a} (after being transposed if {@code transpose_a} is non-zero) must + * match the outer dimension of {@code b} (after being transposed if {@code transposed_b} is + * non-zero). Then do broadcast add operation with bias values on the matrix + * multiplication result. The bias size must match inner dimension of {@code b}. + * + * @param data type for {@code out} output + * @param a A matrix to be multiplied. Must be a two-dimensional tensor of type {@code quint8}. + * @param b A matrix to be multiplied and must be a two-dimensional tensor of type {@code qint8}. + * @param bias A 1D bias tensor with size matching inner dimension of {@code b} (after being + * transposed if {@code transposed_b} is non-zero). + * @param minA The float value that the lowest quantized {@code a} value represents. + * @param maxA The float value that the highest quantized {@code a} value represents. + * @param minB The float value that the lowest quantized {@code b} value represents. + * @param maxB The float value that the highest quantized {@code b} value represents. + * @param Toutput The value of the Toutput attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedMatMulWithBias} output and operands + * @return a new instance of QuantizedMatMulWithBias + */ + public QuantizedMatMulWithBias quantizedMatMulWithBias( + Operand a, Operand b, Operand bias, + Operand minA, Operand maxA, Operand minB, + Operand maxB, Class Toutput, QuantizedMatMulWithBias.Options... options) { + return QuantizedMatMulWithBias.create(scope, a, b, bias, minA, maxA, minB, maxB, Toutput, options); + } + + /** + * Perform a quantized matrix multiplication of {@code a} by the matrix {@code b} with bias + * add and relu fusion. + * The inputs must be two-dimensional matrices and 1D bias vector. And the inner + * dimension of {@code a} (after being transposed if {@code transpose_a} is non-zero) must + * match the outer dimension of {@code b} (after being transposed if {@code transposed_b} is + * non-zero). Then do broadcast add operation with bias values on the matrix + * multiplication result. The bias size must match inner dimension of {@code b}. Then do + * relu activation to get non-negative result. + * + * @param data type for {@code out} output + * @param a A matrix to be multiplied. Must be a two-dimensional tensor of type {@code quint8}. + * @param b A matrix to be multiplied and must be a two-dimensional tensor of type {@code qint8}. + * @param bias A 1D bias tensor with size matching with inner dimension of {@code b} (after being + * transposed if {@code transposed_b} is non-zero). + * @param minA The float value that the lowest quantized {@code a} value represents. + * @param maxA The float value that the highest quantized {@code a} value represents. + * @param minB The float value that the lowest quantized {@code b} value represents. + * @param maxB The float value that the highest quantized {@code b} value represents. + * @param Toutput The value of the Toutput attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedMatMulWithBiasAndRelu} output and operands + * @return a new instance of QuantizedMatMulWithBiasAndRelu + */ + public QuantizedMatMulWithBiasAndRelu quantizedMatMulWithBiasAndRelu( + Operand a, Operand b, Operand bias, + Operand minA, Operand maxA, Operand minB, + Operand maxB, Class Toutput, QuantizedMatMulWithBiasAndRelu.Options... options) { + return QuantizedMatMulWithBiasAndRelu.create(scope, a, b, bias, minA, maxA, minB, maxB, Toutput, options); + } + + /** + * Perform a quantized matrix multiplication of {@code a} by the matrix {@code b} with bias + * add and relu and requantize fusion. + * The inputs must be two-dimensional matrices and 1D bias vector. And the inner + * dimension of {@code a} (after being transposed if {@code transpose_a} is non-zero) must + * match the outer dimension of {@code b} (after being transposed if {@code transposed_b} is + * non-zero). Then do broadcast add operation with bias values on the matrix + * multiplication result. The bias size must match inner dimension of {@code b}. Then do + * relu activation to get non-negative result. Then do requantize operation to get + * final uint8 result. + * + * @param data type for {@code out} output + * @param a A matrix to be multiplied. Must be a two-dimensional tensor of type {@code quint8}. + * @param b A matrix to be multiplied and must be a two-dimensional tensor of type {@code qint8}. + * @param bias A 1D bias tensor with size matching with inner dimension of {@code b} (after being + * transposed if {@code transposed_b} is non-zero). + * @param minA The float value that the lowest quantized {@code a} value represents. + * @param maxA The float value that the highest quantized {@code a} value represents. + * @param minB The float value that the lowest quantized {@code b} value represents. + * @param maxB The float value that the highest quantized {@code b} value represents. + * @param minFreezedOutput The float value that the highest quantized output value after requantize. + * @param maxFreezedOutput The maxFreezedOutput value + * @param Toutput The value of the Toutput attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedMatMulWithBiasAndReluAndRequantize} output and operands + * @return a new instance of QuantizedMatMulWithBiasAndReluAndRequantize + */ + public QuantizedMatMulWithBiasAndReluAndRequantize quantizedMatMulWithBiasAndReluAndRequantize( + Operand a, Operand b, Operand bias, + Operand minA, Operand maxA, Operand minB, + Operand maxB, Operand minFreezedOutput, + Operand maxFreezedOutput, Class Toutput, + QuantizedMatMulWithBiasAndReluAndRequantize.Options... options) { + return QuantizedMatMulWithBiasAndReluAndRequantize.create(scope, a, b, bias, minA, maxA, minB, maxB, minFreezedOutput, maxFreezedOutput, Toutput, options); + } + /** * Computes the eigen decomposition of one or more square self-adjoint matrices. * Computes the eigenvalues and (optionally) eigenvectors of each inner matrix in @@ -1557,6 +1715,53 @@ public TriangularSolve triangularSolve(Operand matrix, O return TriangularSolve.create(scope, matrix, rhs, options); } + /** + * Calculate product with tridiagonal matrix. + * Calculates product of two matrices, where left matrix is a tridiagonal matrix. + * + * @param data type for {@code output} output + * @param superdiag Tensor of shape {@code [..., 1, M]}, representing superdiagonals of + * tri-diagonal matrices to the left of multiplication. Last element is ignored. + * @param maindiag Tensor of shape {@code [..., 1, M]}, representing main diagonals of tri-diagonal + * matrices to the left of multiplication. + * @param subdiag Tensor of shape {@code [..., 1, M]}, representing subdiagonals of tri-diagonal + * matrices to the left of multiplication. First element is ignored. + * @param rhs Tensor of shape {@code [..., M, N]}, representing MxN matrices to the right of + * multiplication. + * @param data type for {@code TridiagonalMatMul} output and operands + * @return a new instance of TridiagonalMatMul + */ + public TridiagonalMatMul tridiagonalMatMul(Operand superdiag, + Operand maindiag, Operand subdiag, Operand rhs) { + return TridiagonalMatMul.create(scope, superdiag, maindiag, subdiag, rhs); + } + + /** + * Solves tridiagonal systems of equations. + * Solves tridiagonal systems of equations. + * Supports batch dimensions and multiple right-hand sides per each left-hand + * side. + * On CPU, solution is computed via Gaussian elimination with or without partial + * pivoting, depending on {@code partial_pivoting} attribute. On GPU, Nvidia's cuSPARSE + * library is used: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv + * Partial pivoting is not yet supported by XLA backends. + * + * @param data type for {@code output} output + * @param diagonals Tensor of shape {@code [..., 3, M]} whose innermost 2 dimensions represent the + * tridiagonal matrices with three rows being the superdiagonal, diagonals, and + * subdiagonals, in order. The last element of the superdiagonal and the first + * element of the subdiagonal is ignored. + * @param rhs Tensor of shape {@code [..., M, K]}, representing K right-hand sides per each + * left-hand side. + * @param options carries optional attribute values + * @param data type for {@code TridiagonalSolve} output and operands + * @return a new instance of TridiagonalSolve + */ + public TridiagonalSolve tridiagonalSolve(Operand diagonals, + Operand rhs, TridiagonalSolve.Options... options) { + return TridiagonalSolve.create(scope, diagonals, rhs, options); + } + /** * Get the parent {@link Ops} object. */ diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/LinalgSparseOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/LinalgSparseOps.java new file mode 100644 index 00000000000..0e0e2f25664 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/LinalgSparseOps.java @@ -0,0 +1,484 @@ +// Copyright 2020-2022 The TensorFlow Authors. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// ============================================================================== +// +// This class has been generated, DO NOT EDIT! +// +package org.tensorflow.op; + +import org.tensorflow.Operand; +import org.tensorflow.op.linalg.sparse.CSRSparseMatrixComponents; +import org.tensorflow.op.linalg.sparse.CSRSparseMatrixToDense; +import org.tensorflow.op.linalg.sparse.CSRSparseMatrixToSparseTensor; +import org.tensorflow.op.linalg.sparse.DenseToCSRSparseMatrix; +import org.tensorflow.op.linalg.sparse.SparseMatrixAdd; +import org.tensorflow.op.linalg.sparse.SparseMatrixMatMul; +import org.tensorflow.op.linalg.sparse.SparseMatrixMul; +import org.tensorflow.op.linalg.sparse.SparseMatrixNNZ; +import org.tensorflow.op.linalg.sparse.SparseMatrixOrderingAMD; +import org.tensorflow.op.linalg.sparse.SparseMatrixSoftmax; +import org.tensorflow.op.linalg.sparse.SparseMatrixSoftmaxGrad; +import org.tensorflow.op.linalg.sparse.SparseMatrixSparseCholesky; +import org.tensorflow.op.linalg.sparse.SparseMatrixSparseMatMul; +import org.tensorflow.op.linalg.sparse.SparseMatrixTranspose; +import org.tensorflow.op.linalg.sparse.SparseMatrixZeros; +import org.tensorflow.op.linalg.sparse.SparseTensorToCSRSparseMatrix; +import org.tensorflow.types.TInt32; +import org.tensorflow.types.TInt64; +import org.tensorflow.types.family.TNumber; +import org.tensorflow.types.family.TType; + +/** + * An API for building {@code linalg.sparse} operations as {@link Op Op}s + * + * @see {@link Ops} + */ +public final class LinalgSparseOps { + private final Scope scope; + + private final Ops ops; + + LinalgSparseOps(Ops ops) { + this.scope = ops.scope(); + this.ops = ops; + } + + /** + * Reads out the CSR components at batch {@code index}. + * This op is meant only for debugging / testing, and its interface is not expected + * to be stable. + * + * @param data type for {@code values} output + * @param csrSparseMatrix A batched CSRSparseMatrix. + * @param index The index in {@code csr_sparse_matrix}'s batch. + * @param type The value of the type attribute + * @param data type for {@code CSRSparseMatrixComponents} output and operands + * @return a new instance of CSRSparseMatrixComponents + */ + public CSRSparseMatrixComponents cSRSparseMatrixComponents( + Operand csrSparseMatrix, Operand index, Class type) { + return CSRSparseMatrixComponents.create(scope, csrSparseMatrix, index, type); + } + + /** + * Convert a (possibly batched) CSRSparseMatrix to dense. + * + * @param data type for {@code dense_output} output + * @param sparseInput A batched CSRSparseMatrix. + * @param type The value of the type attribute + * @param data type for {@code CSRSparseMatrixToDense} output and operands + * @return a new instance of CSRSparseMatrixToDense + */ + public CSRSparseMatrixToDense cSRSparseMatrixToDense( + Operand sparseInput, Class type) { + return CSRSparseMatrixToDense.create(scope, sparseInput, type); + } + + /** + * Converts a (possibly batched) CSRSparesMatrix to a SparseTensor. + * + * @param data type for {@code values} output + * @param sparseMatrix A (possibly batched) CSRSparseMatrix. + * @param type The value of the type attribute + * @param data type for {@code CSRSparseMatrixToSparseTensor} output and operands + * @return a new instance of CSRSparseMatrixToSparseTensor + */ + public CSRSparseMatrixToSparseTensor cSRSparseMatrixToSparseTensor( + Operand sparseMatrix, Class type) { + return CSRSparseMatrixToSparseTensor.create(scope, sparseMatrix, type); + } + + /** + * Converts a dense tensor to a (possibly batched) CSRSparseMatrix. + * + * @param denseInput A Dense tensor. + * @param indices Indices of nonzero elements. + * @return a new instance of DenseToCSRSparseMatrix + */ + public DenseToCSRSparseMatrix denseToCSRSparseMatrix(Operand denseInput, + Operand indices) { + return DenseToCSRSparseMatrix.create(scope, denseInput, indices); + } + + /** + * Sparse addition of two CSR matrices, C = alpha * A + beta * B. + * The gradients of SparseMatrixAdd outputs with respect to alpha and beta are not + * currently defined (TensorFlow will return zeros for these entries). + * + * @param a A CSRSparseMatrix. + * @param b A CSRSparseMatrix. + * @param alpha A constant scalar. + * @param beta A constant scalar. + * @param data type for {@code SparseMatrixAdd} output and operands + * @return a new instance of SparseMatrixAdd + */ + public SparseMatrixAdd sparseMatrixAdd(Operand a, + Operand b, Operand alpha, Operand beta) { + return SparseMatrixAdd.create(scope, a, b, alpha, beta); + } + + /** + * Matrix-multiplies a sparse matrix with a dense matrix. + * Returns a dense matrix. + * For inputs A and B, where A is CSR and B is dense; this op returns a dense C; + *

If transpose_output is false, returns: + * 

+   *    C = A . B
+   *
+ *

If transpose_output is {@code true}, returns: + * 

+   *    C = transpose(A . B) = transpose(B) . transpose(A)
+   *
+ *

where the transposition is performed along the two innermost (matrix) + * dimensions. + *

If conjugate_output is {@code true}, returns: + * 

+   *    C = conjugate(A . B) = conjugate(A) . conjugate(B)
+   *
+ *

If both conjugate_output and transpose_output are {@code true}, returns: + * 

+   *    C = conjugate(transpose(A . B)) = conjugate(transpose(B)) .
+   *                                      conjugate(transpose(A))
+   *
+ * + * @param data type for {@code output} output + * @param a A CSRSparseMatrix. + * @param b A dense tensor. + * @param options carries optional attribute values + * @param data type for {@code SparseMatrixMatMul} output and operands + * @return a new instance of SparseMatrixMatMul + */ + public SparseMatrixMatMul sparseMatrixMatMul(Operand a, + Operand b, SparseMatrixMatMul.Options... options) { + return SparseMatrixMatMul.create(scope, a, b, options); + } + + /** + * Element-wise multiplication of a sparse matrix with a dense tensor. + * Returns a sparse matrix. + *

The dense tensor {@code b} may be either a scalar; otherwise {@code a} must be a rank-3 + * {@code SparseMatrix}; in this case {@code b} must be shaped {@code [batch_size, 1, 1]} and the + * multiply operation broadcasts. + *

NOTE even if {@code b} is zero, the sparsity structure of the output does not + * change. + * + * @param a A CSRSparseMatrix. + * @param b A dense tensor. + * @return a new instance of SparseMatrixMul + */ + public SparseMatrixMul sparseMatrixMul(Operand a, Operand b) { + return SparseMatrixMul.create(scope, a, b); + } + + /** + * Returns the number of nonzeroes of {@code sparse_matrix}. + * + * @param sparseMatrix A CSRSparseMatrix. + * @return a new instance of SparseMatrixNNZ + */ + public SparseMatrixNNZ sparseMatrixNNZ(Operand sparseMatrix) { + return SparseMatrixNNZ.create(scope, sparseMatrix); + } + + /** + * Computes the Approximate Minimum Degree (AMD) ordering of {@code input}. + * Computes the Approximate Minimum Degree (AMD) ordering for a sparse matrix. + *

The returned permutation may be used to permute the rows and columns of the + * given sparse matrix. This typically results in permuted sparse matrix's sparse + * Cholesky (or other decompositions) in having fewer zero fill-in compared to + * decomposition of the original matrix. + *

The input sparse matrix may have rank 2 or rank 3. The output Tensor, + * representing would then have rank 1 or 2 respectively, with the same batch + * shape as the input. + *

Each component of the input sparse matrix must represent a square symmetric + * matrix; only the lower triangular part of the matrix is read. The values of the + * sparse matrix does not affect the returned permutation, only the sparsity + * pattern of the sparse matrix is used. Hence, a single AMD ordering may be + * reused for the Cholesky decompositions of sparse matrices with the same sparsity + * pattern but with possibly different values. + *

Each batch component of the output permutation represents a permutation of {@code N} + * elements, where the input sparse matrix components each have {@code N} rows. That is, + * the component contains each of the integers {@code {0, .. N-1}} exactly once. The + * {@code i}th element represents the row index that the {@code i}th row maps to. + *

Usage example: + * 

+   *      from tensorflow.python.ops.linalg.sparse import sparse_csr_matrix_ops
+   *
+   *      a_indices = np.array([[0, 0], [1, 1], [2, 1], [2, 2], [3, 3]])
+   *      a_values = np.array([1.0, 2.0, 1.0, 3.0, 4.0], np.float32)
+   *      a_dense_shape = [4, 4]
+   *
+   *      with tf.Session() as sess:
+   *        # Define (COO format) SparseTensor over Numpy array.
+   *        a_st = tf.sparse.SparseTensor(a_indices, a_values, a_dense_shape)
+   *
+   *        # Convert SparseTensors to CSR SparseMatrix.
+   *        a_sm = sparse_csr_matrix_ops.sparse_tensor_to_csr_sparse_matrix(
+   *            a_st.indices, a_st.values, a_st.dense_shape)
+   *
+   *        # Obtain the AMD Ordering for the CSR SparseMatrix.
+   *        ordering_amd = sparse_csr_matrix_ops.sparse_matrix_ordering_amd(sparse_matrix)
+   *
+   *        ordering_amd_value = sess.run(ordering_amd)
+   *
+ *

{@code ordering_amd_value} stores the AMD ordering: {@code [1 2 3 0]}. + *

input: A {@code CSRSparseMatrix}. + * + * @param input A {@code CSRSparseMatrix}. + * @return a new instance of SparseMatrixOrderingAMD + */ + public SparseMatrixOrderingAMD sparseMatrixOrderingAMD(Operand input) { + return SparseMatrixOrderingAMD.create(scope, input); + } + + /** + * Calculates the softmax of a CSRSparseMatrix. + * Calculate the softmax of the innermost dimensions of a SparseMatrix. + *

Missing values are treated as {@code -inf} (i.e., logits of zero probability); and + * the output has the same sparsity structure as the input (though missing values + * in the output may now be treated as having probability zero). + * + * @param logits A CSRSparseMatrix. + * @param type The value of the type attribute + * @param data type for {@code SparseMatrixSoftmax} output and operands + * @return a new instance of SparseMatrixSoftmax + */ + public SparseMatrixSoftmax sparseMatrixSoftmax( + Operand logits, Class type) { + return SparseMatrixSoftmax.create(scope, logits, type); + } + + /** + * Calculates the gradient of the SparseMatrixSoftmax op. + * + * @param softmax A CSRSparseMatrix. + * @param gradSoftmax The gradient of {@code softmax}. + * @param type The value of the type attribute + * @param data type for {@code SparseMatrixSoftmaxGrad} output and operands + * @return a new instance of SparseMatrixSoftmaxGrad + */ + public SparseMatrixSoftmaxGrad sparseMatrixSoftmaxGrad( + Operand softmax, Operand gradSoftmax, Class type) { + return SparseMatrixSoftmaxGrad.create(scope, softmax, gradSoftmax, type); + } + + /** + * Computes the sparse Cholesky decomposition of {@code input}. + * Computes the Sparse Cholesky decomposition of a sparse matrix, with the given + * fill-in reducing permutation. + *

The input sparse matrix and the fill-in reducing permutation {@code permutation} must + * have compatible shapes. If the sparse matrix has rank 3; with the batch + * dimension {@code B}, then the {@code permutation} must be of rank 2; with the same batch + * dimension {@code B}. There is no support for broadcasting. + *

Furthermore, each component vector of {@code permutation} must be of length {@code N}, + * containing each of the integers {0, 1, ..., N - 1} exactly once, where {@code N} is + * the number of rows of each component of the sparse matrix. + *

Each component of the input sparse matrix must represent a symmetric positive + * definite (SPD) matrix; although only the lower triangular part of the matrix is + * read. If any individual component is not SPD, then an InvalidArgument error is + * thrown. + *

The returned sparse matrix has the same dense shape as the input sparse matrix. + * For each component {@code A} of the input sparse matrix, the corresponding output + * sparse matrix represents {@code L}, the lower triangular Cholesky factor satisfying + * the following identity: + * 

+   *    A = L * Lt
+   *
+ *

where Lt denotes the transpose of L (or its conjugate transpose, if {@code type} is + * {@code complex64} or {@code complex128}). + *

The {@code type} parameter denotes the type of the matrix elements. The supported + * types are: {@code float32}, {@code float64}, {@code complex64} and {@code complex128}. + *

Usage example: + * 

+   *      from tensorflow.python.ops.linalg.sparse import sparse_csr_matrix_ops
+   *
+   *      a_indices = np.array([[0, 0], [1, 1], [2, 1], [2, 2], [3, 3]])
+   *      a_values = np.array([1.0, 2.0, 1.0, 3.0, 4.0], np.float32)
+   *      a_dense_shape = [4, 4]
+   *
+   *      with tf.Session() as sess:
+   *        # Define (COO format) SparseTensor over Numpy array.
+   *        a_st = tf.sparse.SparseTensor(a_indices, a_values, a_dense_shape)
+   *
+   *        # Convert SparseTensors to CSR SparseMatrix.
+   *        a_sm = sparse_csr_matrix_ops.sparse_tensor_to_csr_sparse_matrix(
+   *            a_st.indices, a_st.values, a_st.dense_shape)
+   *
+   *        # Obtain the Sparse Cholesky factor using AMD Ordering for reducing zero
+   *        # fill-in (number of structural non-zeros in the sparse Cholesky factor).
+   *        ordering_amd = sparse_csr_matrix_ops.sparse_matrix_ordering_amd(sparse_matrix)
+   *        cholesky_sparse_matrices = (
+   *            sparse_csr_matrix_ops.sparse_matrix_sparse_cholesky(
+   *                sparse_matrix, ordering_amd, type=tf.float32))
+   *
+   *        # Convert the CSRSparseMatrix Cholesky factor to a dense Tensor
+   *        dense_cholesky = sparse_csr_matrix_ops.csr_sparse_matrix_to_dense(
+   *            cholesky_sparse_matrices, tf.float32)
+   *
+   *        # Evaluate the dense Tensor value.
+   *        dense_cholesky_value = sess.run(dense_cholesky)
+   *
+ *

{@code dense_cholesky_value} stores the dense Cholesky factor: + * 

+   *      [[  1.  0.    0.    0.]
+   *       [  0.  1.41  0.    0.]
+   *       [  0.  0.70  1.58  0.]
+   *       [  0.  0.    0.    2.]]
+   *
+ *

input: A {@code CSRSparseMatrix}. + * permutation: A {@code Tensor}. + * type: The type of {@code input}. + * + * @param input A {@code CSRSparseMatrix}. + * @param permutation A fill-in reducing permutation matrix. + * @param type The value of the type attribute + * @param data type for {@code SparseMatrixSparseCholesky} output and operands + * @return a new instance of SparseMatrixSparseCholesky + */ + public SparseMatrixSparseCholesky sparseMatrixSparseCholesky( + Operand input, Operand permutation, Class type) { + return SparseMatrixSparseCholesky.create(scope, input, permutation, type); + } + + /** + * Sparse-matrix-multiplies two CSR matrices {@code a} and {@code b}. + * Performs a matrix multiplication of a sparse matrix {@code a} with a sparse matrix + * {@code b}; returns a sparse matrix {@code a * b}, unless either {@code a} or {@code b} is transposed or + * adjointed. + *

Each matrix may be transposed or adjointed (conjugated and transposed) + * according to the Boolean parameters {@code transpose_a}, {@code adjoint_a}, {@code transpose_b} + * and {@code adjoint_b}. At most one of {@code transpose_a} or {@code adjoint_a} may be True. + * Similarly, at most one of {@code transpose_b} or {@code adjoint_b} may be True. + *

The inputs must have compatible shapes. That is, the inner dimension of {@code a} + * must be equal to the outer dimension of {@code b}. This requirement is adjusted + * according to whether either {@code a} or {@code b} is transposed or adjointed. + *

The {@code type} parameter denotes the type of the matrix elements. Both {@code a} and {@code b} + * must have the same type. The supported types are: {@code float32}, {@code float64}, + * {@code complex64} and {@code complex128}. + *

Both {@code a} and {@code b} must have the same rank. Broadcasting is not supported. If they + * have rank 3, each batch of 2D CSRSparseMatrices within {@code a} and {@code b} must have the + * same dense shape. + *

The sparse matrix product may have numeric (non-structural) zeros. + * TODO(anudhyan): Consider adding a boolean attribute to control whether to prune + * zeros. + *

Usage example: + * 

+   *      from tensorflow.python.ops.linalg.sparse import sparse_csr_matrix_ops
+   *
+   *      a_indices = np.array([[0, 0], [2, 3], [2, 4], [3, 0]])
+   *      a_values = np.array([1.0, 5.0, -1.0, -2.0], np.float32)
+   *      a_dense_shape = [4, 5]
+   *
+   *      b_indices = np.array([[0, 0], [3, 0], [3, 1]])
+   *      b_values = np.array([2.0, 7.0, 8.0], np.float32)
+   *      b_dense_shape = [5, 3]
+   *
+   *      with tf.Session() as sess:
+   *        # Define (COO format) Sparse Tensors over Numpy arrays
+   *        a_st = tf.sparse.SparseTensor(a_indices, a_values, a_dense_shape)
+   *        b_st = tf.sparse.SparseTensor(b_indices, b_values, b_dense_shape)
+   *
+   *        # Convert SparseTensors to CSR SparseMatrix
+   *        a_sm = sparse_csr_matrix_ops.sparse_tensor_to_csr_sparse_matrix(
+   *            a_st.indices, a_st.values, a_st.dense_shape)
+   *        b_sm = sparse_csr_matrix_ops.sparse_tensor_to_csr_sparse_matrix(
+   *            b_st.indices, b_st.values, b_st.dense_shape)
+   *
+   *        # Compute the CSR SparseMatrix matrix multiplication
+   *        c_sm = sparse_csr_matrix_ops.sparse_matrix_sparse_mat_mul(
+   *            a=a_sm, b=b_sm, type=tf.float32)
+   *
+   *        # Convert the CSR SparseMatrix product to a dense Tensor
+   *        c_sm_dense = sparse_csr_matrix_ops.csr_sparse_matrix_to_dense(
+   *            c_sm, tf.float32)
+   *        # Evaluate the dense Tensor value
+   *        c_sm_dense_value = sess.run(c_sm_dense)
+   *
+ *

{@code c_sm_dense_value} stores the dense matrix product: + * 

+   *      [[  2.   0.   0.]
+   *       [  0.   0.   0.]
+   *       [ 35.  40.   0.]
+   *       [ -4.   0.   0.]]
+   *
+ *

a: A {@code CSRSparseMatrix}. + * b: A {@code CSRSparseMatrix} with the same type and rank as {@code a}. + * type: The type of both {@code a} and {@code b}. + * transpose_a: If True, {@code a} transposed before multiplication. + * transpose_b: If True, {@code b} transposed before multiplication. + * adjoint_a: If True, {@code a} adjointed before multiplication. + * adjoint_b: If True, {@code b} adjointed before multiplication. + * + * @param a A CSRSparseMatrix. + * @param b A CSRSparseMatrix. + * @param type The value of the type attribute + * @param options carries optional attribute values + * @param data type for {@code SparseMatrixSparseMatMul} output and operands + * @return a new instance of SparseMatrixSparseMatMul + */ + public SparseMatrixSparseMatMul sparseMatrixSparseMatMul( + Operand a, Operand b, Class type, + SparseMatrixSparseMatMul.Options... options) { + return SparseMatrixSparseMatMul.create(scope, a, b, type, options); + } + + /** + * Transposes the inner (matrix) dimensions of a CSRSparseMatrix. + * Transposes the inner (matrix) dimensions of a SparseMatrix and optionally + * conjugates its values. + * + * @param input A CSRSparseMatrix. + * @param type The value of the type attribute + * @param options carries optional attribute values + * @param data type for {@code SparseMatrixTranspose} output and operands + * @return a new instance of SparseMatrixTranspose + */ + public SparseMatrixTranspose sparseMatrixTranspose( + Operand input, Class type, SparseMatrixTranspose.Options... options) { + return SparseMatrixTranspose.create(scope, input, type, options); + } + + /** + * Creates an all-zeros CSRSparseMatrix with shape {@code dense_shape}. + * + * @param denseShape The desired matrix shape. + * @param type The value of the type attribute + * @param data type for {@code SparseMatrixZeros} output and operands + * @return a new instance of SparseMatrixZeros + */ + public SparseMatrixZeros sparseMatrixZeros(Operand denseShape, + Class type) { + return SparseMatrixZeros.create(scope, denseShape, type); + } + + /** + * Converts a SparseTensor to a (possibly batched) CSRSparseMatrix. + * + * @param indices SparseTensor indices. + * @param values SparseTensor values. + * @param denseShape SparseTensor dense shape. + * @return a new instance of SparseTensorToCSRSparseMatrix + */ + public SparseTensorToCSRSparseMatrix sparseTensorToCSRSparseMatrix(Operand indices, + Operand values, Operand denseShape) { + return SparseTensorToCSRSparseMatrix.create(scope, indices, values, denseShape); + } + + /** + * Get the parent {@link Ops} object. + */ + public final Ops ops() { + return ops; + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/MathOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/MathOps.java index ae95a5c9cd6..b9002b321d9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/MathOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/MathOps.java @@ -34,6 +34,10 @@ import org.tensorflow.op.math.Atan; import org.tensorflow.op.math.Atan2; import org.tensorflow.op.math.Atanh; +import org.tensorflow.op.math.BesselI0; +import org.tensorflow.op.math.BesselI0e; +import org.tensorflow.op.math.BesselI1; +import org.tensorflow.op.math.BesselI1e; import org.tensorflow.op.math.Betainc; import org.tensorflow.op.math.Bincount; import org.tensorflow.op.math.Ceil; @@ -43,6 +47,7 @@ import org.tensorflow.op.math.Cosh; import org.tensorflow.op.math.Cumprod; import org.tensorflow.op.math.Cumsum; +import org.tensorflow.op.math.CumulativeLogsumexp; import org.tensorflow.op.math.DenseBincount; import org.tensorflow.op.math.Digamma; import org.tensorflow.op.math.Div; @@ -59,6 +64,7 @@ import org.tensorflow.op.math.Greater; import org.tensorflow.op.math.GreaterEqual; import org.tensorflow.op.math.Igamma; +import org.tensorflow.op.math.IgammaGradA; import org.tensorflow.op.math.Igammac; import org.tensorflow.op.math.Imag; import org.tensorflow.op.math.InvertPermutation; @@ -91,27 +97,37 @@ import org.tensorflow.op.math.Real; import org.tensorflow.op.math.RealDiv; import org.tensorflow.op.math.Reciprocal; +import org.tensorflow.op.math.ReciprocalGrad; +import org.tensorflow.op.math.RequantizationRangePerChannel; +import org.tensorflow.op.math.RequantizePerChannel; import org.tensorflow.op.math.Rint; import org.tensorflow.op.math.Round; import org.tensorflow.op.math.Rsqrt; +import org.tensorflow.op.math.RsqrtGrad; import org.tensorflow.op.math.SegmentMax; import org.tensorflow.op.math.SegmentMean; import org.tensorflow.op.math.SegmentMin; import org.tensorflow.op.math.SegmentProd; import org.tensorflow.op.math.SegmentSum; import org.tensorflow.op.math.Sigmoid; +import org.tensorflow.op.math.SigmoidGrad; import org.tensorflow.op.math.Sign; import org.tensorflow.op.math.Sin; import org.tensorflow.op.math.Sinh; +import org.tensorflow.op.math.SobolSample; import org.tensorflow.op.math.Softplus; +import org.tensorflow.op.math.SoftplusGrad; import org.tensorflow.op.math.Sqrt; +import org.tensorflow.op.math.SqrtGrad; import org.tensorflow.op.math.Square; import org.tensorflow.op.math.SquaredDifference; import org.tensorflow.op.math.Sub; import org.tensorflow.op.math.Tan; import org.tensorflow.op.math.Tanh; +import org.tensorflow.op.math.TanhGrad; import org.tensorflow.op.math.TruncateDiv; import org.tensorflow.op.math.TruncateMod; +import org.tensorflow.op.math.UniformQuantizedAdd; import org.tensorflow.op.math.UnsortedSegmentMax; import org.tensorflow.op.math.UnsortedSegmentMin; import org.tensorflow.op.math.UnsortedSegmentProd; @@ -134,6 +150,8 @@ * @see {@link Ops} */ public final class MathOps { + public final MathSpecialOps special; + private final Scope scope; private final Ops ops; @@ -141,6 +159,7 @@ public final class MathOps { MathOps(Ops ops) { this.scope = ops.scope(); this.ops = ops; + special = new MathSpecialOps(ops); } /** @@ -528,6 +547,54 @@ public Atanh atanh(Operand x) { return Atanh.create(scope, x); } + /** + * The BesselI0 operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselI0} output and operands + * @return a new instance of BesselI0 + */ + public BesselI0 besselI0(Operand x) { + return BesselI0.create(scope, x); + } + + /** + * The BesselI0e operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselI0e} output and operands + * @return a new instance of BesselI0e + */ + public BesselI0e besselI0e(Operand x) { + return BesselI0e.create(scope, x); + } + + /** + * The BesselI1 operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselI1} output and operands + * @return a new instance of BesselI1 + */ + public BesselI1 besselI1(Operand x) { + return BesselI1.create(scope, x); + } + + /** + * The BesselI1e operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselI1e} output and operands + * @return a new instance of BesselI1e + */ + public BesselI1e besselI1e(Operand x) { + return BesselI1e.create(scope, x); + } + /** * Compute the regularized incomplete beta integral \(I_x(a, b)\). * The regularized incomplete beta integral is defined as: @@ -772,6 +839,38 @@ public Cumsum cumsum(Operand x, Operand + * tf.math.cumulative_logsumexp([a, b, c]) # => [a, log(exp(a) + exp(b)), log(exp(a) + exp(b) + exp(c))] + * + *

By setting the {@code exclusive} kwarg to {@code True}, an exclusive cumulative log-sum-exp is + * performed instead: + * 

+   *  tf.cumulative_logsumexp([a, b, c], exclusive=True)  # => [-inf, a, log(exp(a) * exp(b))]
+   *
+ *

Note that the neutral element of the log-sum-exp operation is {@code -inf}, + * however, for performance reasons, the minimal value representable by the + * floating point type is used instead. + *

By setting the {@code reverse} kwarg to {@code True}, the cumulative log-sum-exp is performed in the + * opposite direction. + * + * @param data type for {@code out} output + * @param x A {@code Tensor}. Must be one of the following types: {@code float16}, {@code float32}, {@code float64}. + * @param axis A {@code Tensor} of type {@code int32} (default: 0). Must be in the range + * {@code [-rank(x), rank(x))}. + * @param options carries optional attribute values + * @param data type for {@code CumulativeLogsumexp} output and operands + * @return a new instance of CumulativeLogsumexp + */ + public CumulativeLogsumexp cumulativeLogsumexp(Operand x, + Operand axis, CumulativeLogsumexp.Options... options) { + return CumulativeLogsumexp.create(scope, x, axis, options); + } + /** * Counts the number of occurrences of each value in an integer array. * Outputs a vector with length {@code size} and the same dtype as {@code weights}. If @@ -1079,6 +1178,19 @@ public Igamma igamma(Operand a, Operand x) { return Igamma.create(scope, a, x); } + /** + * Computes the gradient of {@code igamma(a, x)} wrt {@code a}. + * + * @param data type for {@code z} output + * @param a The a value + * @param x The x value + * @param data type for {@code IgammaGradA} output and operands + * @return a new instance of IgammaGradA + */ + public IgammaGradA igammaGradA(Operand a, Operand x) { + return IgammaGradA.create(scope, a, x); + } + /** * Compute the upper regularized incomplete Gamma function {@code Q(a, x)}. * The upper regularized incomplete Gamma function is defined as: @@ -1676,6 +1788,57 @@ public Reciprocal reciprocal(Operand x) { return Reciprocal.create(scope, x); } + /** + * Computes the gradient for the inverse of {@code x} wrt its input. + * Specifically, {@code grad = -dy * y*y}, where {@code y = 1/x}, and {@code dy} + * is the corresponding input gradient. + * + * @param data type for {@code z} output + * @param y The y value + * @param dy The dy value + * @param data type for {@code ReciprocalGrad} output and operands + * @return a new instance of ReciprocalGrad + */ + public ReciprocalGrad reciprocalGrad(Operand y, Operand dy) { + return ReciprocalGrad.create(scope, y, dy); + } + + /** + * Computes requantization range per channel. + * + * @param input The original input tensor. + * @param inputMin The minimum value of the input tensor + * @param inputMax The maximum value of the input tensor. + * @param clipValueMax The maximum value of the output that needs to be clipped. + * Example: set this to 6 for Relu6. + * @return a new instance of RequantizationRangePerChannel + */ + public RequantizationRangePerChannel requantizationRangePerChannel( + Operand input, Operand inputMin, Operand inputMax, + Float clipValueMax) { + return RequantizationRangePerChannel.create(scope, input, inputMin, inputMax, clipValueMax); + } + + /** + * Requantizes input with min and max values known per channel. + * + * @param data type for {@code output} output + * @param input The original input tensor. + * @param inputMin The minimum value of the input tensor + * @param inputMax The maximum value of the input tensor. + * @param requestedOutputMin The minimum value of the output tensor requested. + * @param requestedOutputMax The maximum value of the output tensor requested. + * @param outType The quantized type of output tensor that needs to be converted. + * @param data type for {@code RequantizePerChannel} output and operands + * @return a new instance of RequantizePerChannel + */ + public RequantizePerChannel requantizePerChannel( + Operand input, Operand inputMin, Operand inputMax, + Operand requestedOutputMin, Operand requestedOutputMax, + Class outType) { + return RequantizePerChannel.create(scope, input, inputMin, inputMax, requestedOutputMin, requestedOutputMax, outType); + } + /** * Returns element-wise integer closest to x. * If the result is midway between two representable values, @@ -1723,6 +1886,21 @@ public Rsqrt rsqrt(Operand x) { return Rsqrt.create(scope, x); } + /** + * Computes the gradient for the rsqrt of {@code x} wrt its input. + * Specifically, {@code grad = dy * -0.5 * y^3}, where {@code y = rsqrt(x)}, and {@code dy} + * is the corresponding input gradient. + * + * @param data type for {@code z} output + * @param y The y value + * @param dy The dy value + * @param data type for {@code RsqrtGrad} output and operands + * @return a new instance of RsqrtGrad + */ + public RsqrtGrad rsqrtGrad(Operand y, Operand dy) { + return RsqrtGrad.create(scope, y, dy); + } + /** * Computes the maximum along segments of a tensor. * Read @@ -1972,6 +2150,21 @@ public Sigmoid sigmoid(Operand x) { return Sigmoid.create(scope, x); } + /** + * Computes the gradient of the sigmoid of {@code x} wrt its input. + * Specifically, {@code grad = dy * y * (1 - y)}, where {@code y = sigmoid(x)}, and + * {@code dy} is the corresponding input gradient. + * + * @param data type for {@code z} output + * @param y The y value + * @param dy The dy value + * @param data type for {@code SigmoidGrad} output and operands + * @return a new instance of SigmoidGrad + */ + public SigmoidGrad sigmoidGrad(Operand y, Operand dy) { + return SigmoidGrad.create(scope, y, dy); + } + /** * Returns an element-wise indication of the sign of a number. * {@code y = sign(x) = -1} if {@code x < 0}; 0 if {@code x == 0}; 1 if {@code x > 0}. @@ -2033,6 +2226,44 @@ public Sinh sinh(Operand x) { return Sinh.create(scope, x); } + /** + * Generates points from the Sobol sequence. + * Creates a Sobol sequence with {@code num_results} samples. Each sample has dimension + * {@code dim}. Skips the first {@code skip} samples. + * + * @param data type for {@code samples} output + * @param dim Positive scalar {@code Tensor} representing each sample's dimension. + * @param numResults Positive scalar {@code Tensor} of dtype int32. The number of Sobol points to return + * in the output. + * @param skip Positive scalar {@code Tensor} of dtype int32. The number of initial points of the + * Sobol sequence to skip. + * @return a new instance of SobolSample, with default output types + */ + public SobolSample sobolSample(Operand dim, Operand numResults, + Operand skip) { + return SobolSample.create(scope, dim, numResults, skip); + } + + /** + * Generates points from the Sobol sequence. + * Creates a Sobol sequence with {@code num_results} samples. Each sample has dimension + * {@code dim}. Skips the first {@code skip} samples. + * + * @param data type for {@code samples} output + * @param dim Positive scalar {@code Tensor} representing each sample's dimension. + * @param numResults Positive scalar {@code Tensor} of dtype int32. The number of Sobol points to return + * in the output. + * @param skip Positive scalar {@code Tensor} of dtype int32. The number of initial points of the + * Sobol sequence to skip. + * @param dtype The type of the sample. One of: {@code float32} or {@code float64}. + * @param data type for {@code SobolSample} output and operands + * @return a new instance of SobolSample + */ + public SobolSample sobolSample(Operand dim, + Operand numResults, Operand skip, Class dtype) { + return SobolSample.create(scope, dim, numResults, skip, dtype); + } + /** * The Softplus operation * @@ -2045,6 +2276,20 @@ public Softplus softplus(Operand features) { return Softplus.create(scope, features); } + /** + * Computes softplus gradients for a softplus operation. + * + * @param data type for {@code backprops} output + * @param gradients The backpropagated gradients to the corresponding softplus operation. + * @param features The features passed as input to the corresponding softplus operation. + * @param data type for {@code SoftplusGrad} output and operands + * @return a new instance of SoftplusGrad + */ + public SoftplusGrad softplusGrad(Operand gradients, + Operand features) { + return SoftplusGrad.create(scope, gradients, features); + } + /** * Computes square root of x element-wise. * I.e., \(y = \sqrt{x} = x^{1/2}\). @@ -2058,6 +2303,21 @@ public Sqrt sqrt(Operand x) { return Sqrt.create(scope, x); } + /** + * Computes the gradient for the sqrt of {@code x} wrt its input. + * Specifically, {@code grad = dy * 0.5 / y}, where {@code y = sqrt(x)}, and {@code dy} + * is the corresponding input gradient. + * + * @param data type for {@code z} output + * @param y The y value + * @param dy The dy value + * @param data type for {@code SqrtGrad} output and operands + * @return a new instance of SqrtGrad + */ + public SqrtGrad sqrtGrad(Operand y, Operand dy) { + return SqrtGrad.create(scope, y, dy); + } + /** * Computes square of x element-wise. * I.e., \(y = x * x = x^2\). @@ -2147,6 +2407,21 @@ public Tanh tanh(Operand x) { return Tanh.create(scope, x); } + /** + * Computes the gradient for the tanh of {@code x} wrt its input. + * Specifically, {@code grad = dy * (1 - y*y)}, where {@code y = tanh(x)}, and {@code dy} + * is the corresponding input gradient. + * + * @param data type for {@code z} output + * @param y The y value + * @param dy The dy value + * @param data type for {@code TanhGrad} output and operands + * @return a new instance of TanhGrad + */ + public TanhGrad tanhGrad(Operand y, Operand dy) { + return TanhGrad.create(scope, y, dy); + } + /** * Returns x / y element-wise, rounded towards zero. * Truncation designates that negative numbers will round fractional quantities @@ -2182,6 +2457,61 @@ public TruncateMod truncateMod(Operand x, Operand y return TruncateMod.create(scope, x, y); } + /** + * Perform quantized add of quantized Tensor {@code lhs} and quantized Tensor {@code rhs} to make quantized {@code output}. + * Given quantized {@code lhs} and quantized {@code rhs}, performs quantized add on {@code lhs} and {@code rhs} to make quantized {@code output}. + *

{@code math.UniformQuantizedAdd} follows Numpy broadcasting rules. + * The two input array shapes are compared element-wise. + * Starting with the trailing dimensions, the two dimensions either have to be equal or one of them needs to be 1. + *

{@code lhs} and {@code rhs} must be quantized Tensor, where data value is quantized using the formula: + * 

+   *  quantized_data = clip(original_data / scale + zero_point, quantization_min_val, quantization_max_val)
+   *
+ *

{@code output} is also quantized, using the same formula. + *

If {@code lhs} and {@code output} is both per-axis quantized, the quantization axis must match. + * Also, if {@code rhs} and {@code output} is both per-axis quantized, the quantization axis must match. + * Match means the axis must match when adding, regarding the broadcasting. + * i.e. For both operands {@code lhs} and {@code rhs}, + * if {@code operand.quantization_axis} >= 0 and {@code output.quantization_axis} >= 0, + * {@code operand.dims} - {@code operand.quantization_axis} must be equal to {@code output.dims} - {@code output.quantization_axis}. + * + * @param data type for {@code output} output + * @param lhs Must be a quantized tensor. + * @param rhs Must be a quantized tensor. + * @param lhsScales The float value(s) used as scale factors when quantizing the original data that {@code lhs} represents. + * @param lhsZeroPoints The int32 value(s) used as zero points when quantizing original data that {@code lhs} represents. + * Must have same shape with {@code lhs_scales}. + * @param rhsScales The float value(s) used as scale factors when quantizing the original data that {@code rhs} represents. + * @param rhsZeroPoints The int32 value(s) used as zero points when quantizing original data that {@code rhs} represents. + * Must have same shape with {@code rhs_scales}. + * @param outputScales The float value(s) to use as scale factors when quantizing original data that {@code output} represents. + * @param outputZeroPoints The int32 value(s) used as zero points when quantizing original data that output represents. + * Must have same shape with {@code output_scales}. + * @param lhsQuantizationMinVal The min value of the quantized data stored in {@code lhs}. + * For example, if {@code Tin} is {@code qint8}, this must be set to -127 if narrow range quantized or -128 if not. + * @param lhsQuantizationMaxVal The max value of the quantized data stored in {@code lhs}. + * For example, if {@code Tin} is {@code qint8}, this must be set to 127. + * @param rhsQuantizationMinVal The min value of the quantized data stored in {@code rhs}. + * For example, if {@code Tin} is {@code qint8}, this must be set to -127 if narrow range quantized or -128 if not. + * @param rhsQuantizationMaxVal The max value of the quantized data stored in {@code rhs}. + * For example, if {@code Tin} is {@code qint8}, this must be set to 127. + * @param outputQuantizationMinVal The min value of the quantized data stored in {@code output}. + * For example, if {@code Tout} is {@code qint8}, this must be set to -127 if narrow range quantized or -128 if not. + * @param outputQuantizationMaxVal The max value of the quantized data stored in {@code output}. + * For example, if {@code Tout} is {@code qint8}, this must be set to 127. + * @param options carries optional attribute values + * @param data type for {@code UniformQuantizedAdd} output and operands + * @return a new instance of UniformQuantizedAdd + */ + public UniformQuantizedAdd uniformQuantizedAdd(Operand lhs, + Operand rhs, Operand lhsScales, Operand lhsZeroPoints, + Operand rhsScales, Operand rhsZeroPoints, Operand outputScales, + Operand outputZeroPoints, Long lhsQuantizationMinVal, Long lhsQuantizationMaxVal, + Long rhsQuantizationMinVal, Long rhsQuantizationMaxVal, Long outputQuantizationMinVal, + Long outputQuantizationMaxVal, UniformQuantizedAdd.Options... options) { + return UniformQuantizedAdd.create(scope, lhs, rhs, lhsScales, lhsZeroPoints, rhsScales, rhsZeroPoints, outputScales, outputZeroPoints, lhsQuantizationMinVal, lhsQuantizationMaxVal, rhsQuantizationMinVal, rhsQuantizationMaxVal, outputQuantizationMinVal, outputQuantizationMaxVal, options); + } + /** * Computes the maximum along segments of a tensor. * Read diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/MathSpecialOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/MathSpecialOps.java new file mode 100644 index 00000000000..569c20e2c02 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/MathSpecialOps.java @@ -0,0 +1,213 @@ +// Copyright 2020-2022 The TensorFlow Authors. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// ============================================================================== +// +// This class has been generated, DO NOT EDIT! +// +package org.tensorflow.op; + +import org.tensorflow.Operand; +import org.tensorflow.op.math.special.BesselJ0; +import org.tensorflow.op.math.special.BesselJ1; +import org.tensorflow.op.math.special.BesselK0; +import org.tensorflow.op.math.special.BesselK0e; +import org.tensorflow.op.math.special.BesselK1; +import org.tensorflow.op.math.special.BesselK1e; +import org.tensorflow.op.math.special.BesselY0; +import org.tensorflow.op.math.special.BesselY1; +import org.tensorflow.op.math.special.Dawsn; +import org.tensorflow.op.math.special.Expint; +import org.tensorflow.op.math.special.FresnelCos; +import org.tensorflow.op.math.special.FresnelSin; +import org.tensorflow.op.math.special.Spence; +import org.tensorflow.types.family.TNumber; + +/** + * An API for building {@code math.special} operations as {@link Op Op}s + * + * @see {@link Ops} + */ +public final class MathSpecialOps { + private final Scope scope; + + private final Ops ops; + + MathSpecialOps(Ops ops) { + this.scope = ops.scope(); + this.ops = ops; + } + + /** + * The BesselJ0 operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselJ0} output and operands + * @return a new instance of BesselJ0 + */ + public BesselJ0 besselJ0(Operand x) { + return BesselJ0.create(scope, x); + } + + /** + * The BesselJ1 operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselJ1} output and operands + * @return a new instance of BesselJ1 + */ + public BesselJ1 besselJ1(Operand x) { + return BesselJ1.create(scope, x); + } + + /** + * The BesselK0 operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselK0} output and operands + * @return a new instance of BesselK0 + */ + public BesselK0 besselK0(Operand x) { + return BesselK0.create(scope, x); + } + + /** + * The BesselK0e operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselK0e} output and operands + * @return a new instance of BesselK0e + */ + public BesselK0e besselK0e(Operand x) { + return BesselK0e.create(scope, x); + } + + /** + * The BesselK1 operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselK1} output and operands + * @return a new instance of BesselK1 + */ + public BesselK1 besselK1(Operand x) { + return BesselK1.create(scope, x); + } + + /** + * The BesselK1e operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselK1e} output and operands + * @return a new instance of BesselK1e + */ + public BesselK1e besselK1e(Operand x) { + return BesselK1e.create(scope, x); + } + + /** + * The BesselY0 operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselY0} output and operands + * @return a new instance of BesselY0 + */ + public BesselY0 besselY0(Operand x) { + return BesselY0.create(scope, x); + } + + /** + * The BesselY1 operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code BesselY1} output and operands + * @return a new instance of BesselY1 + */ + public BesselY1 besselY1(Operand x) { + return BesselY1.create(scope, x); + } + + /** + * The Dawsn operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code Dawsn} output and operands + * @return a new instance of Dawsn + */ + public Dawsn dawsn(Operand x) { + return Dawsn.create(scope, x); + } + + /** + * The Expint operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code Expint} output and operands + * @return a new instance of Expint + */ + public Expint expint(Operand x) { + return Expint.create(scope, x); + } + + /** + * The FresnelCos operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code FresnelCos} output and operands + * @return a new instance of FresnelCos + */ + public FresnelCos fresnelCos(Operand x) { + return FresnelCos.create(scope, x); + } + + /** + * The FresnelSin operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code FresnelSin} output and operands + * @return a new instance of FresnelSin + */ + public FresnelSin fresnelSin(Operand x) { + return FresnelSin.create(scope, x); + } + + /** + * The Spence operation + * + * @param data type for {@code y} output + * @param x The x value + * @param data type for {@code Spence} output and operands + * @return a new instance of Spence + */ + public Spence spence(Operand x) { + return Spence.create(scope, x); + } + + /** + * Get the parent {@link Ops} object. + */ + public final Ops ops() { + return ops; + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/NnOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/NnOps.java index 7ac1a318348..c4838b02a39 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/NnOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/NnOps.java @@ -22,10 +22,14 @@ import org.tensorflow.op.nn.AvgPool; import org.tensorflow.op.nn.AvgPool3d; import org.tensorflow.op.nn.AvgPool3dGrad; +import org.tensorflow.op.nn.AvgPoolGrad; import org.tensorflow.op.nn.BatchNormWithGlobalNormalization; import org.tensorflow.op.nn.BatchNormWithGlobalNormalizationGrad; import org.tensorflow.op.nn.BiasAdd; import org.tensorflow.op.nn.BiasAddGrad; +import org.tensorflow.op.nn.BlockLSTM; +import org.tensorflow.op.nn.BlockLSTMGrad; +import org.tensorflow.op.nn.CTCLossV2; import org.tensorflow.op.nn.ComputeAccidentalHits; import org.tensorflow.op.nn.Conv; import org.tensorflow.op.nn.Conv2d; @@ -37,6 +41,8 @@ import org.tensorflow.op.nn.CtcBeamSearchDecoder; import org.tensorflow.op.nn.CtcGreedyDecoder; import org.tensorflow.op.nn.CtcLoss; +import org.tensorflow.op.nn.CudnnRNN; +import org.tensorflow.op.nn.CudnnRNNBackprop; import org.tensorflow.op.nn.CudnnRNNCanonicalToParams; import org.tensorflow.op.nn.CudnnRNNParamsToCanonical; import org.tensorflow.op.nn.CudnnRnnParamsSize; @@ -50,18 +56,28 @@ import org.tensorflow.op.nn.Dilation2dBackpropFilter; import org.tensorflow.op.nn.Dilation2dBackpropInput; import org.tensorflow.op.nn.Elu; +import org.tensorflow.op.nn.EluGrad; import org.tensorflow.op.nn.FixedUnigramCandidateSampler; import org.tensorflow.op.nn.FractionalAvgPool; +import org.tensorflow.op.nn.FractionalAvgPoolGrad; import org.tensorflow.op.nn.FractionalMaxPool; +import org.tensorflow.op.nn.FractionalMaxPoolGrad; import org.tensorflow.op.nn.FusedBatchNorm; import org.tensorflow.op.nn.FusedBatchNormGrad; import org.tensorflow.op.nn.FusedPadConv2d; import org.tensorflow.op.nn.FusedResizeAndPadConv2d; +import org.tensorflow.op.nn.GRUBlockCell; +import org.tensorflow.op.nn.GRUBlockCellGrad; import org.tensorflow.op.nn.InTopK; +import org.tensorflow.op.nn.InvGrad; +import org.tensorflow.op.nn.IsotonicRegression; import org.tensorflow.op.nn.L2Loss; +import org.tensorflow.op.nn.LSTMBlockCell; +import org.tensorflow.op.nn.LSTMBlockCellGrad; import org.tensorflow.op.nn.LeakyRelu; import org.tensorflow.op.nn.LearnedUnigramCandidateSampler; import org.tensorflow.op.nn.LocalResponseNormalization; +import org.tensorflow.op.nn.LocalResponseNormalizationGrad; import org.tensorflow.op.nn.LogSoftmax; import org.tensorflow.op.nn.MaxPool; import org.tensorflow.op.nn.MaxPool3d; @@ -70,12 +86,28 @@ import org.tensorflow.op.nn.MaxPoolGrad; import org.tensorflow.op.nn.MaxPoolGradGrad; import org.tensorflow.op.nn.MaxPoolGradGradWithArgmax; +import org.tensorflow.op.nn.MaxPoolGradWithArgmax; import org.tensorflow.op.nn.MaxPoolWithArgmax; import org.tensorflow.op.nn.NthElement; import org.tensorflow.op.nn.QuantizedAvgPool; import org.tensorflow.op.nn.QuantizedBatchNormWithGlobalNormalization; import org.tensorflow.op.nn.QuantizedBiasAdd; +import org.tensorflow.op.nn.QuantizedConv2DAndRelu; +import org.tensorflow.op.nn.QuantizedConv2DAndReluAndRequantize; +import org.tensorflow.op.nn.QuantizedConv2DAndRequantize; +import org.tensorflow.op.nn.QuantizedConv2DPerChannel; +import org.tensorflow.op.nn.QuantizedConv2DWithBias; +import org.tensorflow.op.nn.QuantizedConv2DWithBiasAndRelu; +import org.tensorflow.op.nn.QuantizedConv2DWithBiasAndReluAndRequantize; +import org.tensorflow.op.nn.QuantizedConv2DWithBiasAndRequantize; +import org.tensorflow.op.nn.QuantizedConv2DWithBiasSignedSumAndReluAndRequantize; +import org.tensorflow.op.nn.QuantizedConv2DWithBiasSumAndRelu; +import org.tensorflow.op.nn.QuantizedConv2DWithBiasSumAndReluAndRequantize; import org.tensorflow.op.nn.QuantizedConv2d; +import org.tensorflow.op.nn.QuantizedDepthwiseConv2D; +import org.tensorflow.op.nn.QuantizedDepthwiseConv2DWithBias; +import org.tensorflow.op.nn.QuantizedDepthwiseConv2DWithBiasAndRelu; +import org.tensorflow.op.nn.QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize; import org.tensorflow.op.nn.QuantizedInstanceNorm; import org.tensorflow.op.nn.QuantizedMaxPool; import org.tensorflow.op.nn.QuantizedRelu; @@ -83,14 +115,20 @@ import org.tensorflow.op.nn.QuantizedReluX; import org.tensorflow.op.nn.Relu; import org.tensorflow.op.nn.Relu6; +import org.tensorflow.op.nn.Relu6Grad; +import org.tensorflow.op.nn.ReluGrad; import org.tensorflow.op.nn.Selu; +import org.tensorflow.op.nn.SeluGrad; import org.tensorflow.op.nn.Softmax; import org.tensorflow.op.nn.SoftmaxCrossEntropyWithLogits; import org.tensorflow.op.nn.Softsign; +import org.tensorflow.op.nn.SoftsignGrad; import org.tensorflow.op.nn.SpaceToBatch; import org.tensorflow.op.nn.SpaceToDepth; import org.tensorflow.op.nn.SparseSoftmaxCrossEntropyWithLogits; import org.tensorflow.op.nn.TopK; +import org.tensorflow.op.nn.UniformQuantizedConvolution; +import org.tensorflow.op.nn.UniformQuantizedConvolutionHybrid; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -173,6 +211,26 @@ public AvgPool3dGrad avgPool3dGrad(Operand origIn return AvgPool3dGrad.create(scope, origInputShape, grad, ksize, strides, padding, options); } + /** + * Computes gradients of the average pooling function. + * + * @param data type for {@code output} output + * @param origInputShape 1-D. Shape of the original input to {@code avg_pool}. + * @param grad 4-D with shape {@code [batch, height, width, channels]}. Gradients w.r.t. + * the output of {@code avg_pool}. + * @param ksize The size of the sliding window for each dimension of the input. + * @param strides The stride of the sliding window for each dimension of the input. + * @param padding The type of padding algorithm to use. + * @param options carries optional attribute values + * @param data type for {@code AvgPoolGrad} output and operands + * @return a new instance of AvgPoolGrad + */ + public AvgPoolGrad avgPoolGrad(Operand origInputShape, + Operand grad, List ksize, List strides, String padding, + AvgPoolGrad.Options... options) { + return AvgPoolGrad.create(scope, origInputShape, grad, ksize, strides, padding, options); + } + /** * Batch normalization. * This op is deprecated. Prefer {@code tf.nn.batch_normalization}. @@ -264,6 +322,106 @@ public BiasAddGrad biasAddGrad(Operand outBackprop, return BiasAddGrad.create(scope, outBackprop, options); } + /** + * Computes the LSTM cell forward propagation for all the time steps. + * This is equivalent to applying LSTMBlockCell in a loop, like so: + * 

+   *  for x1 in unpack(x):
+   *    i1, cs1, f1, o1, ci1, co1, h1 = LSTMBlock(
+   *      x1, cs_prev, h_prev, w, wci, wcf, wco, b)
+   *    cs_prev = cs1
+   *    h_prev = h1
+   *    i.append(i1)
+   *    cs.append(cs1)
+   *    f.append(f1)
+   *    o.append(o1)
+   *    ci.append(ci1)
+   *    co.append(co1)
+   *    h.append(h1)
+   *  return pack(i), pack(cs), pack(f), pack(o), pack(ci), pack(ch), pack(h)
+   *
+   *  Note that unlike LSTMBlockCell (and BlockLSTM) which uses ICFO gate layout,
+   *  this op uses IFCO. So in order for the following snippet to be equivalent
+   *  all gate-related outputs should be reordered.
+   *
+ * + * @param data type for {@code i} output + * @param seqLenMax Maximum time length actually used by this input. Outputs are padded + * with zeros beyond this length. + * @param x The sequence input to the LSTM, shape (timelen, batch_size, num_inputs). + * @param csPrev Value of the initial cell state. + * @param hPrev Initial output of cell (to be used for peephole). + * @param w The weight matrix. + * @param wci The weight matrix for input gate peephole connection. + * @param wcf The weight matrix for forget gate peephole connection. + * @param wco The weight matrix for output gate peephole connection. + * @param b The bias vector. + * @param options carries optional attribute values + * @param data type for {@code BlockLSTMV2} output and operands + * @return a new instance of BlockLSTM + */ + public BlockLSTM blockLSTM(Operand seqLenMax, Operand x, + Operand csPrev, Operand hPrev, Operand w, Operand wci, Operand wcf, + Operand wco, Operand b, BlockLSTM.Options... options) { + return BlockLSTM.create(scope, seqLenMax, x, csPrev, hPrev, w, wci, wcf, wco, b, options); + } + + /** + * Computes the LSTM cell backward propagation for the entire time sequence. + * This implementation is to be used in conjunction of BlockLSTMV2. + * + * @param data type for {@code x_grad} output + * @param seqLenMax Maximum time length actually used by this input. Outputs are padded + * with zeros beyond this length. + * @param x The sequence input to the LSTM, shape (timelen, batch_size, num_inputs). + * @param csPrev Value of the initial cell state. + * @param hPrev Initial output of cell (to be used for peephole). + * @param w The weight matrix. + * @param wci The weight matrix for input gate peephole connection. + * @param wcf The weight matrix for forget gate peephole connection. + * @param wco The weight matrix for output gate peephole connection. + * @param b The bias vector. + * @param i The input gate over the whole time sequence. + * @param cs The cell state before the tanh over the whole time sequence. + * @param f The forget gate over the whole time sequence. + * @param o The output gate over the whole time sequence. + * @param ci The cell input over the whole time sequence. + * @param co The cell after the tanh over the whole time sequence. + * @param h The output h vector over the whole time sequence. + * @param csGrad The current gradient of cs. + * @param hGrad The gradient of h vector. + * @param usePeephole Whether to use peephole weights. + * @param data type for {@code BlockLSTMGradV2} output and operands + * @return a new instance of BlockLSTMGrad + */ + public BlockLSTMGrad blockLSTMGrad(Operand seqLenMax, Operand x, + Operand csPrev, Operand hPrev, Operand w, Operand wci, Operand wcf, + Operand wco, Operand b, Operand i, Operand cs, Operand f, Operand o, + Operand ci, Operand co, Operand h, Operand csGrad, Operand hGrad, + Boolean usePeephole) { + return BlockLSTMGrad.create(scope, seqLenMax, x, csPrev, hPrev, w, wci, wcf, wco, b, i, cs, f, o, ci, co, h, csGrad, hGrad, usePeephole); + } + + /** + * Calculates the CTC Loss (log probability) for each batch entry. Also calculates + * the gradient. This class performs the softmax operation for you, so inputs + * should be e.g. linear projections of outputs by an LSTM. + * + * @param inputs 3-D, shape: {@code (max_time x batch_size x num_classes)}, the logits. Default blank + * label is 0 rather num_classes - 1. + * @param labelsIndices The indices of a {@code SparseTensor}. + * {@code labels_indices(i, :) == [b, t]} means {@code labels_values(i)} stores the id for + * {@code (batch b, time t)}. + * @param labelsValues The values (labels) associated with the given batch and time. + * @param sequenceLength A vector containing sequence lengths (batch). + * @param options carries optional attribute values + * @return a new instance of CTCLossV2 + */ + public CTCLossV2 cTCLossV2(Operand inputs, Operand labelsIndices, + Operand labelsValues, Operand sequenceLength, CTCLossV2.Options... options) { + return CTCLossV2.create(scope, inputs, labelsIndices, labelsValues, sequenceLength, options); + } + /** * Computes the ids of the positions in sampled_candidates that match true_labels. * When doing log-odds NCE, the result of this op should be passed through a @@ -533,6 +691,136 @@ public CtcLoss ctcLoss(Operand inputs, Operand return CtcLoss.create(scope, inputs, labelsIndices, labelsValues, sequenceLength, options); } + /** + * A RNN backed by cuDNN. + * Computes the RNN from the input and initial states, with respect to the params + * buffer. Accepts one extra input "sequence_lengths" than CudnnRNN. + *

rnn_mode: Indicates the type of the RNN model. + * input_mode: Indicates whether there is a linear projection between the input and + * the actual computation before the first layer. 'skip_input' is only allowed + * when input_size == num_units; 'auto_select' implies 'skip_input' when + * input_size == num_units; otherwise, it implies 'linear_input'. + * direction: Indicates whether a bidirectional model will be used. Should be + * "unidirectional" or "bidirectional". + * dropout: Dropout probability. When set to 0., dropout is disabled. + * seed: The 1st part of a seed to initialize dropout. + * seed2: The 2nd part of a seed to initialize dropout. + * input: If time_major is true, this is a 3-D tensor with the shape of + * [seq_length, batch_size, input_size]. If time_major is false, the shape is + * [batch_size, seq_length, input_size]. + * input_h: If time_major is true, this is a 3-D tensor with the shape of + * [num_layer * dir, batch_size, num_units]. If time_major is false, the shape + * is [batch_size, num_layer * dir, num_units]. + * input_c: For LSTM, a 3-D tensor with the shape of + * [num_layer * dir, batch, num_units]. For other models, it is ignored. + * params: A 1-D tensor that contains the weights and biases in an opaque layout. + * The size must be created through CudnnRNNParamsSize, and initialized + * separately. Note that they might not be compatible across different + * generations. So it is a good idea to save and restore + * sequence_lengths: a vector of lengths of each input sequence. + * output: If time_major is true, this is a 3-D tensor with the shape of + * [seq_length, batch_size, dir * num_units]. If time_major is false, the + * shape is [batch_size, seq_length, dir * num_units]. + * output_h: The same shape has input_h. + * output_c: The same shape as input_c for LSTM. An empty tensor for other models. + * is_training: Indicates whether this operation is used for inference or + * training. + * time_major: Indicates whether the input/output format is time major or batch + * major. + * reserve_space: An opaque tensor that can be used in backprop calculation. It + * is only produced if is_training is true. + * + * @param data type for {@code output} output + * @param input The input value + * @param inputH The inputH value + * @param inputC The inputC value + * @param params The params value + * @param sequenceLengths The sequenceLengths value + * @param options carries optional attribute values + * @param data type for {@code CudnnRNNV3} output and operands + * @return a new instance of CudnnRNN + */ + public CudnnRNN cudnnRNN(Operand input, Operand inputH, + Operand inputC, Operand params, Operand sequenceLengths, + CudnnRNN.Options... options) { + return CudnnRNN.create(scope, input, inputH, inputC, params, sequenceLengths, options); + } + + /** + * Backprop step of CudnnRNNV3. + * Compute the backprop of both data and weights in a RNN. Takes an extra + * "sequence_lengths" input than CudnnRNNBackprop. + *

rnn_mode: Indicates the type of the RNN model. + * input_mode: Indicates whether there is a linear projection between the input and + * the actual computation before the first layer. 'skip_input' is only allowed + * when input_size == num_units; 'auto_select' implies 'skip_input' when + * input_size == num_units; otherwise, it implies 'linear_input'. + * direction: Indicates whether a bidirectional model will be used. Should be + * "unidirectional" or "bidirectional". + * dropout: Dropout probability. When set to 0., dropout is disabled. + * seed: The 1st part of a seed to initialize dropout. + * seed2: The 2nd part of a seed to initialize dropout. + * input: If time_major is true, this is a 3-D tensor with the shape of + * [seq_length, batch_size, input_size]. If time_major is false, the shape is + * [batch_size, seq_length, input_size]. + * input_h: If time_major is true, this is a 3-D tensor with the shape of + * [num_layer * dir, batch_size, num_units]. If time_major is false, the shape + * is [batch_size, num_layer * dir, num_units]. + * input_c: For LSTM, a 3-D tensor with the shape of + * [num_layer * dir, batch, num_units]. For other models, it is ignored. + * params: A 1-D tensor that contains the weights and biases in an opaque layout. + * The size must be created through CudnnRNNParamsSize, and initialized + * separately. Note that they might not be compatible across different + * generations. So it is a good idea to save and restore + * sequence_lengths: a vector of lengths of each input sequence. + * output: If time_major is true, this is a 3-D tensor with the shape of + * [seq_length, batch_size, dir * num_units]. If time_major is false, the + * shape is [batch_size, seq_length, dir * num_units]. + * output_h: The same shape has input_h. + * output_c: The same shape as input_c for LSTM. An empty tensor for other models. + * output_backprop: A 3-D tensor with the same shape as output in the forward pass. + * output_h_backprop: A 3-D tensor with the same shape as output_h in the forward + * pass. + * output_c_backprop: A 3-D tensor with the same shape as output_c in the forward + * pass. + * time_major: Indicates whether the input/output format is time major or batch + * major. + * reserve_space: The same reserve_space produced in the forward operation. + * input_backprop: The backprop to input in the forward pass. Has the same shape + * as input. + * input_h_backprop: The backprop to input_h in the forward pass. Has the same + * shape as input_h. + * input_c_backprop: The backprop to input_c in the forward pass. Has the same + * shape as input_c. + * params_backprop: The backprop to the params buffer in the forward pass. Has the + * same shape as params. + * + * @param data type for {@code input_backprop} output + * @param input The input value + * @param inputH The inputH value + * @param inputC The inputC value + * @param params The params value + * @param sequenceLengths The sequenceLengths value + * @param output The output value + * @param outputH The outputH value + * @param outputC The outputC value + * @param outputBackprop The outputBackprop value + * @param outputHBackprop The outputHBackprop value + * @param outputCBackprop The outputCBackprop value + * @param reserveSpace The reserveSpace value + * @param hostReserved The hostReserved value + * @param options carries optional attribute values + * @param data type for {@code CudnnRNNBackpropV3} output and operands + * @return a new instance of CudnnRNNBackprop + */ + public CudnnRNNBackprop cudnnRNNBackprop(Operand input, + Operand inputH, Operand inputC, Operand params, Operand sequenceLengths, + Operand output, Operand outputH, Operand outputC, Operand outputBackprop, + Operand outputHBackprop, Operand outputCBackprop, Operand reserveSpace, + Operand hostReserved, CudnnRNNBackprop.Options... options) { + return CudnnRNNBackprop.create(scope, input, inputH, inputC, params, sequenceLengths, output, outputH, outputC, outputBackprop, outputHBackprop, outputCBackprop, reserveSpace, hostReserved, options); + } + /** * Converts CudnnRNN params from canonical form to usable form. It supports the projection in LSTM. * Writes a set of weights into the opaque params buffer so they can be used in @@ -1019,6 +1307,19 @@ public Elu elu(Operand features) { return Elu.create(scope, features); } + /** + * Computes gradients for the exponential linear (Elu) operation. + * + * @param data type for {@code backprops} output + * @param gradients The backpropagated gradients to the corresponding Elu operation. + * @param outputs The outputs of the corresponding Elu operation. + * @param data type for {@code EluGrad} output and operands + * @return a new instance of EluGrad + */ + public EluGrad eluGrad(Operand gradients, Operand outputs) { + return EluGrad.create(scope, gradients, outputs); + } + /** * Generates labels for candidate sampling with a learned unigram distribution. * A unigram sampler could use a fixed unigram distribution read from a @@ -1074,6 +1375,33 @@ public FractionalAvgPool fractionalAvgPool(Operand val return FractionalAvgPool.create(scope, value, poolingRatio, options); } + /** + * Computes gradient of the FractionalAvgPool function. + * Unlike FractionalMaxPoolGrad, we don't need to find arg_max for + * FractionalAvgPoolGrad, we just need to evenly back-propagate each element of + * out_backprop to those indices that form the same pooling cell. Therefore, we + * just need to know the shape of original input tensor, instead of the whole + * tensor. + * + * @param data type for {@code output} output + * @param origInputTensorShape Original input tensor shape for {@code fractional_avg_pool} + * @param outBackprop 4-D with shape {@code [batch, height, width, channels]}. Gradients + * w.r.t. the output of {@code fractional_avg_pool}. + * @param rowPoolingSequence row pooling sequence, form pooling region with + * col_pooling_sequence. + * @param colPoolingSequence column pooling sequence, form pooling region with + * row_pooling sequence. + * @param options carries optional attribute values + * @param data type for {@code FractionalAvgPoolGrad} output and operands + * @return a new instance of FractionalAvgPoolGrad + */ + public FractionalAvgPoolGrad fractionalAvgPoolGrad( + Operand origInputTensorShape, Operand outBackprop, + Operand rowPoolingSequence, Operand colPoolingSequence, + FractionalAvgPoolGrad.Options... options) { + return FractionalAvgPoolGrad.create(scope, origInputTensorShape, outBackprop, rowPoolingSequence, colPoolingSequence, options); + } + /** * Performs fractional max pooling on the input. * Fractional max pooling is slightly different than regular max pooling. In @@ -1120,6 +1448,28 @@ public FractionalMaxPool fractionalMaxPool(Operand val return FractionalMaxPool.create(scope, value, poolingRatio, options); } + /** + * Computes gradient of the FractionalMaxPool function. + * + * @param data type for {@code output} output + * @param origInput Original input for {@code fractional_max_pool} + * @param origOutput Original output for {@code fractional_max_pool} + * @param outBackprop 4-D with shape {@code [batch, height, width, channels]}. Gradients + * w.r.t. the output of {@code fractional_max_pool}. + * @param rowPoolingSequence row pooling sequence, form pooling region with + * col_pooling_sequence. + * @param colPoolingSequence column pooling sequence, form pooling region with + * row_pooling sequence. + * @param options carries optional attribute values + * @param data type for {@code FractionalMaxPoolGrad} output and operands + * @return a new instance of FractionalMaxPoolGrad + */ + public FractionalMaxPoolGrad fractionalMaxPoolGrad(Operand origInput, + Operand origOutput, Operand outBackprop, Operand rowPoolingSequence, + Operand colPoolingSequence, FractionalMaxPoolGrad.Options... options) { + return FractionalMaxPoolGrad.create(scope, origInput, origOutput, outBackprop, rowPoolingSequence, colPoolingSequence, options); + } + /** * Batch normalization. * Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW". @@ -1246,6 +1596,158 @@ public FusedResizeAndPadConv2d fusedResizeAndPadConv2d(Op return FusedResizeAndPadConv2d.create(scope, input, sizeOutput, paddings, filter, mode, strides, padding, options); } + /** + * Computes the GRU cell forward propagation for 1 time step. + * Args + * x: Input to the GRU cell. + * h_prev: State input from the previous GRU cell. + * w_ru: Weight matrix for the reset and update gate. + * w_c: Weight matrix for the cell connection gate. + * b_ru: Bias vector for the reset and update gate. + * b_c: Bias vector for the cell connection gate. + *

Returns + * r: Output of the reset gate. + * u: Output of the update gate. + * c: Output of the cell connection gate. + * h: Current state of the GRU cell. + *

Note on notation of the variables: + *

Concatenation of a and b is represented by a_b + * Element-wise dot product of a and b is represented by ab + * Element-wise dot product is represented by \circ + * Matrix multiplication is represented by * + *

Biases are initialized with : + * {@code b_ru} - constant_initializer(1.0) + * {@code b_c} - constant_initializer(0.0) + *

This kernel op implements the following mathematical equations: + * 

+   *  x_h_prev = [x, h_prev]
+   *
+   *  [r_bar u_bar] = x_h_prev * w_ru + b_ru
+   *
+   *  r = sigmoid(r_bar)
+   *  u = sigmoid(u_bar)
+   *
+   *  h_prevr = h_prev \circ r
+   *
+   *  x_h_prevr = [x h_prevr]
+   *
+   *  c_bar = x_h_prevr * w_c + b_c
+   *  c = tanh(c_bar)
+   *
+   *  h = (1-u) \circ c + u \circ h_prev
+   *
+ * + * @param data type for {@code r} output + * @param x The x value + * @param hPrev The hPrev value + * @param wRu The wRu value + * @param wC The wC value + * @param bRu The bRu value + * @param bC The bC value + * @param data type for {@code GRUBlockCell} output and operands + * @return a new instance of GRUBlockCell + */ + public GRUBlockCell gRUBlockCell(Operand x, Operand hPrev, + Operand wRu, Operand wC, Operand bRu, Operand bC) { + return GRUBlockCell.create(scope, x, hPrev, wRu, wC, bRu, bC); + } + + /** + * Computes the GRU cell back-propagation for 1 time step. + * Args + * x: Input to the GRU cell. + * h_prev: State input from the previous GRU cell. + * w_ru: Weight matrix for the reset and update gate. + * w_c: Weight matrix for the cell connection gate. + * b_ru: Bias vector for the reset and update gate. + * b_c: Bias vector for the cell connection gate. + * r: Output of the reset gate. + * u: Output of the update gate. + * c: Output of the cell connection gate. + * d_h: Gradients of the h_new wrt to objective function. + *

Returns + * d_x: Gradients of the x wrt to objective function. + * d_h_prev: Gradients of the h wrt to objective function. + * d_c_bar Gradients of the c_bar wrt to objective function. + * d_r_bar_u_bar Gradients of the r_bar & u_bar wrt to objective function. + *

This kernel op implements the following mathematical equations: + *

Note on notation of the variables: + *

Concatenation of a and b is represented by a_b + * Element-wise dot product of a and b is represented by ab + * Element-wise dot product is represented by \circ + * Matrix multiplication is represented by * + *

Additional notes for clarity: + *

{@code w_ru} can be segmented into 4 different matrices. + * 

+   *  w_ru = [w_r_x w_u_x
+   *          w_r_h_prev w_u_h_prev]
+   *
+ *

Similarly, {@code w_c} can be segmented into 2 different matrices. + * 

+   *  w_c = [w_c_x w_c_h_prevr]
+   *
+ *

Same goes for biases. + * 

+   *  b_ru = [b_ru_x b_ru_h]
+   *  b_c = [b_c_x b_c_h]
+   *
+ *

Another note on notation: + * 

+   *  d_x = d_x_component_1 + d_x_component_2
+   *
+   *  where d_x_component_1 = d_r_bar * w_r_x^T + d_u_bar * w_r_x^T
+   *  and d_x_component_2 = d_c_bar * w_c_x^T
+   *
+   *  d_h_prev = d_h_prev_component_1 + d_h_prevr \circ r + d_h \circ u
+   *  where d_h_prev_componenet_1 = d_r_bar * w_r_h_prev^T + d_u_bar * w_r_h_prev^T
+   *
+ *

Mathematics behind the Gradients below: + * 

+   *  d_c_bar = d_h \circ (1-u) \circ (1-c \circ c)
+   *  d_u_bar = d_h \circ (h-c) \circ u \circ (1-u)
+   *
+   *  d_r_bar_u_bar = [d_r_bar d_u_bar]
+   *
+   *  [d_x_component_1 d_h_prev_component_1] = d_r_bar_u_bar * w_ru^T
+   *
+   *  [d_x_component_2 d_h_prevr] = d_c_bar * w_c^T
+   *
+   *  d_x = d_x_component_1 + d_x_component_2
+   *
+   *  d_h_prev = d_h_prev_component_1 + d_h_prevr \circ r + u
+   *
+ *

Below calculation is performed in the python wrapper for the Gradients + * (not in the gradient kernel.) + * 

+   *  d_w_ru = x_h_prevr^T * d_c_bar
+   *
+   *  d_w_c = x_h_prev^T * d_r_bar_u_bar
+   *
+   *  d_b_ru = sum of d_r_bar_u_bar along axis = 0
+   *
+   *  d_b_c = sum of d_c_bar along axis = 0
+   *
+ * + * @param data type for {@code d_x} output + * @param x The x value + * @param hPrev The hPrev value + * @param wRu The wRu value + * @param wC The wC value + * @param bRu The bRu value + * @param bC The bC value + * @param r The r value + * @param u The u value + * @param c The c value + * @param dH The dH value + * @param data type for {@code GRUBlockCellGrad} output and operands + * @return a new instance of GRUBlockCellGrad + */ + public GRUBlockCellGrad gRUBlockCellGrad(Operand x, Operand hPrev, + Operand wRu, Operand wC, Operand bRu, Operand bC, Operand r, Operand u, + Operand c, Operand dH) { + return GRUBlockCellGrad.create(scope, x, hPrev, wRu, wC, bRu, bC, r, u, c, dH); + } + /** * Says whether the targets are in the top {@code K} predictions. * This outputs a {@code batch_size} bool array, an entry {@code out[i]} is {@code true} if the @@ -1271,6 +1773,46 @@ public InTopK inTopK(Operand predictions, Operand< return InTopK.create(scope, predictions, targets, k); } + /** + * Computes the gradient for the inverse of {@code x} wrt its input. + * Specifically, {@code grad = -dy * y*y}, where {@code y = 1/x}, and {@code dy} + * is the corresponding input gradient. + * + * @param data type for {@code z} output + * @param y The y value + * @param dy The dy value + * @param data type for {@code InvGrad} output and operands + * @return a new instance of InvGrad + */ + public InvGrad invGrad(Operand y, Operand dy) { + return InvGrad.create(scope, y, dy); + } + + /** + * Solves a batch of isotonic regression problems. + * + * @param data type for {@code output} output + * @param input A (batch_size, dim)-tensor holding a batch of inputs. + * @return a new instance of IsotonicRegression, with default output types + */ + public IsotonicRegression isotonicRegression(Operand input) { + return IsotonicRegression.create(scope, input); + } + + /** + * Solves a batch of isotonic regression problems. + * + * @param data type for {@code output} output + * @param input A (batch_size, dim)-tensor holding a batch of inputs. + * @param outputDtype Dtype of output. + * @param data type for {@code IsotonicRegression} output and operands + * @return a new instance of IsotonicRegression + */ + public IsotonicRegression isotonicRegression( + Operand input, Class outputDtype) { + return IsotonicRegression.create(scope, input, outputDtype); + } + /** * L2 Loss. * Computes half the L2 norm of a tensor without the {@code sqrt}: @@ -1287,6 +1829,82 @@ public L2Loss l2Loss(Operand t) { return L2Loss.create(scope, t); } + /** + * Computes the LSTM cell forward propagation for 1 time step. + * This implementation uses 1 weight matrix and 1 bias vector, and there's an + * optional peephole connection. + *

This kernel op implements the following mathematical equations: + * 

+   *  xh = [x, h_prev]
+   *  [i, f, ci, o] = xh * w + b
+   *  f = f + forget_bias
+   *
+   *  if not use_peephole:
+   *    wci = wcf = wco = 0
+   *
+   *  i = sigmoid(cs_prev * wci + i)
+   *  f = sigmoid(cs_prev * wcf + f)
+   *  ci = tanh(ci)
+   *
+   *  cs = ci .* i + cs_prev .* f
+   *  cs = clip(cs, cell_clip)
+   *
+   *  o = sigmoid(cs * wco + o)
+   *  co = tanh(cs)
+   *  h = co .* o
+   *
+ * + * @param data type for {@code i} output + * @param x The input to the LSTM cell, shape (batch_size, num_inputs). + * @param csPrev Value of the cell state at previous time step. + * @param hPrev Output of the previous cell at previous time step. + * @param w The weight matrix. + * @param wci The weight matrix for input gate peephole connection. + * @param wcf The weight matrix for forget gate peephole connection. + * @param wco The weight matrix for output gate peephole connection. + * @param b The bias vector. + * @param options carries optional attribute values + * @param data type for {@code LSTMBlockCell} output and operands + * @return a new instance of LSTMBlockCell + */ + public LSTMBlockCell lSTMBlockCell(Operand x, Operand csPrev, + Operand hPrev, Operand w, Operand wci, Operand wcf, Operand wco, Operand b, + LSTMBlockCell.Options... options) { + return LSTMBlockCell.create(scope, x, csPrev, hPrev, w, wci, wcf, wco, b, options); + } + + /** + * Computes the LSTM cell backward propagation for 1 timestep. + * This implementation is to be used in conjunction of LSTMBlockCell. + * + * @param data type for {@code cs_prev_grad} output + * @param x The input to the LSTM cell, shape (batch_size, num_inputs). + * @param csPrev The previous cell state. + * @param hPrev The previous h state. + * @param w The weight matrix. + * @param wci The weight matrix for input gate peephole connection. + * @param wcf The weight matrix for forget gate peephole connection. + * @param wco The weight matrix for output gate peephole connection. + * @param b The bias vector. + * @param i The input gate. + * @param cs The cell state before the tanh. + * @param f The forget gate. + * @param o The output gate. + * @param ci The cell input. + * @param co The cell after the tanh. + * @param csGrad The current gradient of cs. + * @param hGrad The gradient of h vector. + * @param usePeephole Whether the cell uses peephole connections. + * @param data type for {@code LSTMBlockCellGrad} output and operands + * @return a new instance of LSTMBlockCellGrad + */ + public LSTMBlockCellGrad lSTMBlockCellGrad(Operand x, Operand csPrev, + Operand hPrev, Operand w, Operand wci, Operand wcf, Operand wco, Operand b, + Operand i, Operand cs, Operand f, Operand o, Operand ci, Operand co, + Operand csGrad, Operand hGrad, Boolean usePeephole) { + return LSTMBlockCellGrad.create(scope, x, csPrev, hPrev, w, wci, wcf, wco, b, i, cs, f, o, ci, co, csGrad, hGrad, usePeephole); + } + /** * Computes rectified linear: {@code max(features, features * alpha)}. * @@ -1353,6 +1971,23 @@ public LocalResponseNormalization localResponseNormalizat return LocalResponseNormalization.create(scope, input, options); } + /** + * Gradients for Local Response Normalization. + * + * @param data type for {@code output} output + * @param inputGrads 4-D with shape {@code [batch, height, width, channels]}. + * @param inputImage 4-D with shape {@code [batch, height, width, channels]}. + * @param outputImage 4-D with shape {@code [batch, height, width, channels]}. + * @param options carries optional attribute values + * @param data type for {@code LRNGrad} output and operands + * @return a new instance of LocalResponseNormalizationGrad + */ + public LocalResponseNormalizationGrad localResponseNormalizationGrad( + Operand inputGrads, Operand inputImage, Operand outputImage, + LocalResponseNormalizationGrad.Options... options) { + return LocalResponseNormalizationGrad.create(scope, inputGrads, inputImage, outputImage, options); + } + /** * Computes log softmax activations. * For each batch {@code i} and class {@code j} we have @@ -1515,6 +2150,28 @@ public MaxPoolGradGradWithArgmax maxPoolGradGradWithArgma return MaxPoolGradGradWithArgmax.create(scope, input, grad, argmax, ksize, strides, padding, options); } + /** + * Computes gradients of the maxpooling function. + * + * @param data type for {@code output} output + * @param input The original input. + * @param grad 4-D with shape {@code [batch, height, width, channels]}. Gradients w.r.t. the + * output of {@code max_pool}. + * @param argmax The indices of the maximum values chosen for each output of {@code max_pool}. + * @param ksize The size of the window for each dimension of the input tensor. + * @param strides The stride of the sliding window for each dimension of the + * input tensor. + * @param padding The type of padding algorithm to use. + * @param options carries optional attribute values + * @param data type for {@code MaxPoolGradWithArgmax} output and operands + * @return a new instance of MaxPoolGradWithArgmax + */ + public MaxPoolGradWithArgmax maxPoolGradWithArgmax(Operand input, + Operand grad, Operand argmax, List ksize, List strides, + String padding, MaxPoolGradWithArgmax.Options... options) { + return MaxPoolGradWithArgmax.create(scope, input, grad, argmax, ksize, strides, padding, options); + } + /** * Performs max pooling on the input and outputs both max values and indices. * The indices in {@code argmax} are flattened, so that a maximum value at position @@ -1682,6 +2339,317 @@ public QuantizedBiasAdd quantizedBiasAdd(Operand data type for {@code output} output + * @param input The input value + * @param filter The filter value + * @param minInput The minInput value + * @param maxInput The maxInput value + * @param minFilter The minFilter value + * @param maxFilter The maxFilter value + * @param outType The value of the outType attribute + * @param strides The value of the strides attribute + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedConv2DAndRelu} output and operands + * @return a new instance of QuantizedConv2DAndRelu + */ + public QuantizedConv2DAndRelu quantizedConv2DAndRelu( + Operand input, Operand filter, + Operand minInput, Operand maxInput, Operand minFilter, + Operand maxFilter, Class outType, List strides, String padding, + QuantizedConv2DAndRelu.Options... options) { + return QuantizedConv2DAndRelu.create(scope, input, filter, minInput, maxInput, minFilter, maxFilter, outType, strides, padding, options); + } + + /** + * The QuantizedConv2DAndReluAndRequantize operation + * + * @param data type for {@code output} output + * @param input The input value + * @param filter The filter value + * @param minInput The minInput value + * @param maxInput The maxInput value + * @param minFilter The minFilter value + * @param maxFilter The maxFilter value + * @param minFreezedOutput The minFreezedOutput value + * @param maxFreezedOutput The maxFreezedOutput value + * @param outType The value of the outType attribute + * @param strides The value of the strides attribute + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedConv2DAndReluAndRequantize} output and operands + * @return a new instance of QuantizedConv2DAndReluAndRequantize + */ + public QuantizedConv2DAndReluAndRequantize quantizedConv2DAndReluAndRequantize( + Operand input, Operand filter, + Operand minInput, Operand maxInput, Operand minFilter, + Operand maxFilter, Operand minFreezedOutput, + Operand maxFreezedOutput, Class outType, List strides, String padding, + QuantizedConv2DAndReluAndRequantize.Options... options) { + return QuantizedConv2DAndReluAndRequantize.create(scope, input, filter, minInput, maxInput, minFilter, maxFilter, minFreezedOutput, maxFreezedOutput, outType, strides, padding, options); + } + + /** + * The QuantizedConv2DAndRequantize operation + * + * @param data type for {@code output} output + * @param input The input value + * @param filter The filter value + * @param minInput The minInput value + * @param maxInput The maxInput value + * @param minFilter The minFilter value + * @param maxFilter The maxFilter value + * @param minFreezedOutput The minFreezedOutput value + * @param maxFreezedOutput The maxFreezedOutput value + * @param outType The value of the outType attribute + * @param strides The value of the strides attribute + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedConv2DAndRequantize} output and operands + * @return a new instance of QuantizedConv2DAndRequantize + */ + public QuantizedConv2DAndRequantize quantizedConv2DAndRequantize( + Operand input, Operand filter, + Operand minInput, Operand maxInput, Operand minFilter, + Operand maxFilter, Operand minFreezedOutput, + Operand maxFreezedOutput, Class outType, List strides, String padding, + QuantizedConv2DAndRequantize.Options... options) { + return QuantizedConv2DAndRequantize.create(scope, input, filter, minInput, maxInput, minFilter, maxFilter, minFreezedOutput, maxFreezedOutput, outType, strides, padding, options); + } + + /** + * Computes QuantizedConv2D per channel. + * + * @param data type for {@code output} output + * @param input The original input tensor. + * @param filter The original filter tensor. + * @param minInput The minimum value of the input tensor + * @param maxInput The maximum value of the input tensor. + * @param minFilter The minimum value of the filter tensor. + * @param maxFilter The maximum value of the filter tensor. + * @param outType The quantized type of output tensor that needs to be converted. + * @param strides list of stride values. + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedConv2DPerChannel} output and operands + * @return a new instance of QuantizedConv2DPerChannel + */ + public QuantizedConv2DPerChannel quantizedConv2DPerChannel( + Operand input, Operand filter, + Operand minInput, Operand maxInput, Operand minFilter, + Operand maxFilter, Class outType, List strides, String padding, + QuantizedConv2DPerChannel.Options... options) { + return QuantizedConv2DPerChannel.create(scope, input, filter, minInput, maxInput, minFilter, maxFilter, outType, strides, padding, options); + } + + /** + * The QuantizedConv2DWithBias operation + * + * @param data type for {@code output} output + * @param input The input value + * @param filter The filter value + * @param bias The bias value + * @param minInput The minInput value + * @param maxInput The maxInput value + * @param minFilter The minFilter value + * @param maxFilter The maxFilter value + * @param outType The value of the outType attribute + * @param strides The value of the strides attribute + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedConv2DWithBias} output and operands + * @return a new instance of QuantizedConv2DWithBias + */ + public QuantizedConv2DWithBias quantizedConv2DWithBias( + Operand input, Operand filter, Operand bias, + Operand minInput, Operand maxInput, Operand minFilter, + Operand maxFilter, Class outType, List strides, String padding, + QuantizedConv2DWithBias.Options... options) { + return QuantizedConv2DWithBias.create(scope, input, filter, bias, minInput, maxInput, minFilter, maxFilter, outType, strides, padding, options); + } + + /** + * The QuantizedConv2DWithBiasAndRelu operation + * + * @param data type for {@code output} output + * @param input The input value + * @param filter The filter value + * @param bias The bias value + * @param minInput The minInput value + * @param maxInput The maxInput value + * @param minFilter The minFilter value + * @param maxFilter The maxFilter value + * @param outType The value of the outType attribute + * @param strides The value of the strides attribute + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedConv2DWithBiasAndRelu} output and operands + * @return a new instance of QuantizedConv2DWithBiasAndRelu + */ + public QuantizedConv2DWithBiasAndRelu quantizedConv2DWithBiasAndRelu( + Operand input, Operand filter, Operand bias, + Operand minInput, Operand maxInput, Operand minFilter, + Operand maxFilter, Class outType, List strides, String padding, + QuantizedConv2DWithBiasAndRelu.Options... options) { + return QuantizedConv2DWithBiasAndRelu.create(scope, input, filter, bias, minInput, maxInput, minFilter, maxFilter, outType, strides, padding, options); + } + + /** + * The QuantizedConv2DWithBiasAndReluAndRequantize operation + * + * @param data type for {@code output} output + * @param input The input value + * @param filter The filter value + * @param bias The bias value + * @param minInput The minInput value + * @param maxInput The maxInput value + * @param minFilter The minFilter value + * @param maxFilter The maxFilter value + * @param minFreezedOutput The minFreezedOutput value + * @param maxFreezedOutput The maxFreezedOutput value + * @param outType The value of the outType attribute + * @param strides The value of the strides attribute + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedConv2DWithBiasAndReluAndRequantize} output and operands + * @return a new instance of QuantizedConv2DWithBiasAndReluAndRequantize + */ + public QuantizedConv2DWithBiasAndReluAndRequantize quantizedConv2DWithBiasAndReluAndRequantize( + Operand input, Operand filter, + Operand bias, Operand minInput, Operand maxInput, + Operand minFilter, Operand maxFilter, Operand minFreezedOutput, + Operand maxFreezedOutput, Class outType, List strides, String padding, + QuantizedConv2DWithBiasAndReluAndRequantize.Options... options) { + return QuantizedConv2DWithBiasAndReluAndRequantize.create(scope, input, filter, bias, minInput, maxInput, minFilter, maxFilter, minFreezedOutput, maxFreezedOutput, outType, strides, padding, options); + } + + /** + * The QuantizedConv2DWithBiasAndRequantize operation + * + * @param data type for {@code output} output + * @param input The input value + * @param filter The filter value + * @param bias The bias value + * @param minInput The minInput value + * @param maxInput The maxInput value + * @param minFilter The minFilter value + * @param maxFilter The maxFilter value + * @param minFreezedOutput The minFreezedOutput value + * @param maxFreezedOutput The maxFreezedOutput value + * @param outType The value of the outType attribute + * @param strides The value of the strides attribute + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedConv2DWithBiasAndRequantize} output and operands + * @return a new instance of QuantizedConv2DWithBiasAndRequantize + */ + public QuantizedConv2DWithBiasAndRequantize quantizedConv2DWithBiasAndRequantize( + Operand input, Operand filter, + Operand bias, Operand minInput, Operand maxInput, + Operand minFilter, Operand maxFilter, Operand minFreezedOutput, + Operand maxFreezedOutput, Class outType, List strides, String padding, + QuantizedConv2DWithBiasAndRequantize.Options... options) { + return QuantizedConv2DWithBiasAndRequantize.create(scope, input, filter, bias, minInput, maxInput, minFilter, maxFilter, minFreezedOutput, maxFreezedOutput, outType, strides, padding, options); + } + + /** + * The QuantizedConv2DWithBiasSignedSumAndReluAndRequantize operation + * + * @param data type for {@code output} output + * @param input The input value + * @param filter The filter value + * @param bias The bias value + * @param minInput The minInput value + * @param maxInput The maxInput value + * @param minFilter The minFilter value + * @param maxFilter The maxFilter value + * @param minFreezedOutput The minFreezedOutput value + * @param maxFreezedOutput The maxFreezedOutput value + * @param summand The summand value + * @param minSummand The minSummand value + * @param maxSummand The maxSummand value + * @param outType The value of the outType attribute + * @param strides The value of the strides attribute + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedConv2DWithBiasSignedSumAndReluAndRequantize} output and operands + * @return a new instance of QuantizedConv2DWithBiasSignedSumAndReluAndRequantize + */ + public QuantizedConv2DWithBiasSignedSumAndReluAndRequantize quantizedConv2DWithBiasSignedSumAndReluAndRequantize( + Operand input, Operand filter, + Operand bias, Operand minInput, Operand maxInput, + Operand minFilter, Operand maxFilter, Operand minFreezedOutput, + Operand maxFreezedOutput, Operand summand, + Operand minSummand, Operand maxSummand, Class outType, + List strides, String padding, + QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.Options... options) { + return QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.create(scope, input, filter, bias, minInput, maxInput, minFilter, maxFilter, minFreezedOutput, maxFreezedOutput, summand, minSummand, maxSummand, outType, strides, padding, options); + } + + /** + * The QuantizedConv2DWithBiasSumAndRelu operation + * + * @param data type for {@code output} output + * @param input The input value + * @param filter The filter value + * @param bias The bias value + * @param minInput The minInput value + * @param maxInput The maxInput value + * @param minFilter The minFilter value + * @param maxFilter The maxFilter value + * @param summand The summand value + * @param outType The value of the outType attribute + * @param strides The value of the strides attribute + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedConv2DWithBiasSumAndRelu} output and operands + * @return a new instance of QuantizedConv2DWithBiasSumAndRelu + */ + public QuantizedConv2DWithBiasSumAndRelu quantizedConv2DWithBiasSumAndRelu( + Operand input, Operand filter, Operand bias, + Operand minInput, Operand maxInput, Operand minFilter, + Operand maxFilter, Operand summand, Class outType, List strides, + String padding, QuantizedConv2DWithBiasSumAndRelu.Options... options) { + return QuantizedConv2DWithBiasSumAndRelu.create(scope, input, filter, bias, minInput, maxInput, minFilter, maxFilter, summand, outType, strides, padding, options); + } + + /** + * The QuantizedConv2DWithBiasSumAndReluAndRequantize operation + * + * @param data type for {@code output} output + * @param input The input value + * @param filter The filter value + * @param bias The bias value + * @param minInput The minInput value + * @param maxInput The maxInput value + * @param minFilter The minFilter value + * @param maxFilter The maxFilter value + * @param minFreezedOutput The minFreezedOutput value + * @param maxFreezedOutput The maxFreezedOutput value + * @param summand The summand value + * @param minSummand The minSummand value + * @param maxSummand The maxSummand value + * @param outType The value of the outType attribute + * @param strides The value of the strides attribute + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedConv2DWithBiasSumAndReluAndRequantize} output and operands + * @return a new instance of QuantizedConv2DWithBiasSumAndReluAndRequantize + */ + public QuantizedConv2DWithBiasSumAndReluAndRequantize quantizedConv2DWithBiasSumAndReluAndRequantize( + Operand input, Operand filter, + Operand bias, Operand minInput, Operand maxInput, + Operand minFilter, Operand maxFilter, Operand minFreezedOutput, + Operand maxFreezedOutput, Operand summand, + Operand minSummand, Operand maxSummand, Class outType, + List strides, String padding, + QuantizedConv2DWithBiasSumAndReluAndRequantize.Options... options) { + return QuantizedConv2DWithBiasSumAndReluAndRequantize.create(scope, input, filter, bias, minInput, maxInput, minFilter, maxFilter, minFreezedOutput, maxFreezedOutput, summand, minSummand, maxSummand, outType, strides, padding, options); + } + /** * Computes a 2D convolution given quantized 4D input and filter tensors. * The inputs are quantized tensors where the lowest value represents the real @@ -1711,6 +2679,112 @@ public QuantizedConv2d quantizedConv2d(Operand data type for {@code output} output + * @param input The original input tensor. + * @param filter The original filter tensor. + * @param minInput The float value that the minimum quantized input value represents. + * @param maxInput The float value that the maximum quantized input value represents. + * @param minFilter The float value that the minimum quantized filter value represents. + * @param maxFilter The float value that the maximum quantized filter value represents. + * @param outType The type of the output. + * @param strides List of stride values. + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedDepthwiseConv2D} output and operands + * @return a new instance of QuantizedDepthwiseConv2D + */ + public QuantizedDepthwiseConv2D quantizedDepthwiseConv2D( + Operand input, Operand filter, + Operand minInput, Operand maxInput, Operand minFilter, + Operand maxFilter, Class outType, List strides, String padding, + QuantizedDepthwiseConv2D.Options... options) { + return QuantizedDepthwiseConv2D.create(scope, input, filter, minInput, maxInput, minFilter, maxFilter, outType, strides, padding, options); + } + + /** + * Computes quantized depthwise Conv2D with Bias. + * + * @param data type for {@code output} output + * @param input The original input tensor. + * @param filter The original filter tensor. + * @param bias The original bias tensor. + * @param minInput The float value that the minimum quantized input value represents. + * @param maxInput The float value that the maximum quantized input value represents. + * @param minFilter The float value that the minimum quantized filter value represents. + * @param maxFilter The float value that the maximum quantized filter value represents. + * @param outType The type of the output. + * @param strides List of stride values. + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedDepthwiseConv2DWithBias} output and operands + * @return a new instance of QuantizedDepthwiseConv2DWithBias + */ + public QuantizedDepthwiseConv2DWithBias quantizedDepthwiseConv2DWithBias( + Operand input, Operand filter, Operand bias, + Operand minInput, Operand maxInput, Operand minFilter, + Operand maxFilter, Class outType, List strides, String padding, + QuantizedDepthwiseConv2DWithBias.Options... options) { + return QuantizedDepthwiseConv2DWithBias.create(scope, input, filter, bias, minInput, maxInput, minFilter, maxFilter, outType, strides, padding, options); + } + + /** + * Computes quantized depthwise Conv2D with Bias and Relu. + * + * @param data type for {@code output} output + * @param input The original input tensor. + * @param filter The original filter tensor. + * @param bias The original bias tensor. + * @param minInput The float value that the minimum quantized input value represents. + * @param maxInput The float value that the maximum quantized input value represents. + * @param minFilter The float value that the minimum quantized filter value represents. + * @param maxFilter The float value that the maximum quantized filter value represents. + * @param outType The type of the output. + * @param strides List of stride values. + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedDepthwiseConv2DWithBiasAndRelu} output and operands + * @return a new instance of QuantizedDepthwiseConv2DWithBiasAndRelu + */ + public QuantizedDepthwiseConv2DWithBiasAndRelu quantizedDepthwiseConv2DWithBiasAndRelu( + Operand input, Operand filter, Operand bias, + Operand minInput, Operand maxInput, Operand minFilter, + Operand maxFilter, Class outType, List strides, String padding, + QuantizedDepthwiseConv2DWithBiasAndRelu.Options... options) { + return QuantizedDepthwiseConv2DWithBiasAndRelu.create(scope, input, filter, bias, minInput, maxInput, minFilter, maxFilter, outType, strides, padding, options); + } + + /** + * Computes quantized depthwise Conv2D with Bias, Relu and Requantize. + * + * @param data type for {@code output} output + * @param input The original input tensor. + * @param filter The original filter tensor. + * @param bias The original bias tensor. + * @param minInput The float value that the minimum quantized input value represents. + * @param maxInput The float value that the maximum quantized input value represents. + * @param minFilter The float value that the minimum quantized filter value represents. + * @param maxFilter The float value that the maximum quantized filter value represents. + * @param minFreezedOutput The minimum float value of the output tensor. + * @param maxFreezedOutput The maximum float value of the output tensor. + * @param outType The type of the output. + * @param strides List of stride values. + * @param padding The value of the padding attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize} output and operands + * @return a new instance of QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize + */ + public QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize quantizedDepthwiseConv2DWithBiasAndReluAndRequantize( + Operand input, Operand filter, + Operand bias, Operand minInput, Operand maxInput, + Operand minFilter, Operand maxFilter, Operand minFreezedOutput, + Operand maxFreezedOutput, Class outType, List strides, String padding, + QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.Options... options) { + return QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.create(scope, input, filter, bias, minInput, maxInput, minFilter, maxFilter, minFreezedOutput, maxFreezedOutput, outType, strides, padding, options); + } + /** * Quantized Instance normalization. * @@ -1832,6 +2906,34 @@ public Relu6 relu6(Operand features) { return Relu6.create(scope, features); } + /** + * Computes rectified linear 6 gradients for a Relu6 operation. + * + * @param data type for {@code backprops} output + * @param gradients The backpropagated gradients to the corresponding Relu6 operation. + * @param features The features passed as input to the corresponding Relu6 operation, or + * its output; using either one produces the same result. + * @param data type for {@code Relu6Grad} output and operands + * @return a new instance of Relu6Grad + */ + public Relu6Grad relu6Grad(Operand gradients, Operand features) { + return Relu6Grad.create(scope, gradients, features); + } + + /** + * Computes rectified linear gradients for a Relu operation. + * + * @param data type for {@code backprops} output + * @param gradients The backpropagated gradients to the corresponding Relu operation. + * @param features The features passed as input to the corresponding Relu operation, OR + * the outputs of that operation (both work equivalently). + * @param data type for {@code ReluGrad} output and operands + * @return a new instance of ReluGrad + */ + public ReluGrad reluGrad(Operand gradients, Operand features) { + return ReluGrad.create(scope, gradients, features); + } + /** * Computes scaled exponential linear: {@code scale * alpha * (exp(features) - 1)} * if < 0, {@code scale * features} otherwise. @@ -1849,6 +2951,19 @@ public Selu selu(Operand features) { return Selu.create(scope, features); } + /** + * Computes gradients for the scaled exponential linear (Selu) operation. + * + * @param data type for {@code backprops} output + * @param gradients The backpropagated gradients to the corresponding Selu operation. + * @param outputs The outputs of the corresponding Selu operation. + * @param data type for {@code SeluGrad} output and operands + * @return a new instance of SeluGrad + */ + public SeluGrad seluGrad(Operand gradients, Operand outputs) { + return SeluGrad.create(scope, gradients, outputs); + } + /** * Computes softmax activations. * For each batch {@code i} and class {@code j} we have @@ -1894,6 +3009,20 @@ public Softsign softsign(Operand features) { return Softsign.create(scope, features); } + /** + * Computes softsign gradients for a softsign operation. + * + * @param data type for {@code backprops} output + * @param gradients The backpropagated gradients to the corresponding softsign operation. + * @param features The features passed as input to the corresponding softsign operation. + * @param data type for {@code SoftsignGrad} output and operands + * @return a new instance of SoftsignGrad + */ + public SoftsignGrad softsignGrad(Operand gradients, + Operand features) { + return SoftsignGrad.create(scope, gradients, features); + } + /** * SpaceToBatch for 4-D tensors of type T. * This is a legacy version of the more general SpaceToBatchND. @@ -2139,6 +3268,121 @@ public TopK topK(Operand input, return TopK.create(scope, input, k, indexType, options); } + /** + * Perform quantized convolution of quantized Tensor {@code lhs} and quantized Tensor {@code rhs}. to make quantized {@code output}. + * Given quantized {@code lhs} and quantized {@code rhs}, performs quantized dot on {@code lhs} and {@code rhs} to make quantized {@code output}. + *

{@code lhs} and {@code rhs} must be Tensors of same rank, and meet following shape conditions. + *

    + *
  • {@code lhs_feature} % {@code feature_group_count} == 0
    • + *
  • {@code lhs_feature} % {@code rhs_input_feature} == 0
    • + *
  • {@code lhs_feature} / {@code feature_group_count} == {@code rhs_input_feature}
    • + *
  • {@code rhs_output_feature} % {@code feature_group_count} == 0
    • + *
  • {@code lhs_batch} % {@code batch_group_count} == 0
    • + *
  • {@code rhs_output_feature} % {@code batch_group_count} == 0
    • + *
+ *

{@code lhs} and {@code rhs} must be quantized Tensor, where data value is quantized using the formula: + * 

+   *  quantized_data = clip(original_data / scale + zero_point, quantization_min_val, quantization_max_val)
+   *
+ *

{@code output} is also quantized, using the same formula. + * If {@code rhs} is per-tensor quantized, {@code output} must be also per-tensor quantized. + * + * @param data type for {@code output} output + * @param lhs Must be a quantized tensor, rank >= 3. + * @param rhs Must be a quantized tensor, same rank as {@code lhs}. + * @param lhsScales The float value(s) used as scale factors when quantizing the original data that {@code lhs} represents. + * Must be a scalar {@code Tensor} ({@code lhs} supports only per-tensor quantization). + * @param lhsZeroPoints The int32 value(s) used as zero points when quantizing original data that {@code lhs} represents. + * Same shape condition as {@code lhs_scales}. + * @param rhsScales The float value(s) used as scale factors when quantizing the original data that {@code rhs} represents. + * Must be a scalar {@code Tensor} for per-tensor quantization, + * or 1D {@code Tensor} of size {@code rhs.dim_size(kernel_output_feature_dimension)}, for per-channel quantization. + * @param rhsZeroPoints The int32 value(s) used as zero points when quantizing original data that {@code rhs} represents. + * Same shape condition as {@code rhs_scales}. + * @param outputScales The float value(s) to use as scale factors when quantizing original data that {@code output} represents. + * Must be a scalar {@code Tensor} for per-tensor quantization, + * or 1D {@code Tensor} of size {@code rhs.dim_size(kernel_output_feature_dimension)} + *

    + *
  • which is equal to {@code output.dim_size(output_feature_dimension)}, + * for per-channel quantization. + * If {@code rhs} is per-tensor quantized, output must be also per-tensor quantized. + * This means that if {@code rhs_scales} and {@code rhs_zero_points} are scalar {@code Tensor}s, {@code output_scales} and {@code output_zero_points} must be scalar {@code Tensor}s as well.
    • + *
+ * @param outputZeroPoints The int32 value(s) used as zero points when quantizing original data that output represents. + * Same shape condition as {@code output_scales}. + * @param Tout The type of {@code output} {@code Tensor}. + * @param padding string from: {@code "SAME"}, {@code "VALID"}, or {@code "EXPLICIT"}, indicating the type of padding algorithm to use. + * @param lhsQuantizationMinVal The min value of the quantized data stored in {@code lhs}. + * For example, if {@code Tin} is {@code qint8}, this must be set to -127 if narrow range quantized or -128 if not. + * @param lhsQuantizationMaxVal The max value of the quantized data stored in {@code lhs}. + * For example, if {@code Tin} is {@code qint8}, this must be set to 127. + * @param rhsQuantizationMinVal The min value of the quantized data stored in {@code rhs}. + * For example, if {@code Tin} is {@code qint8}, this must be set to -127 if narrow range quantized or -128 if not. + * @param rhsQuantizationMaxVal The max value of the quantized data stored in {@code rhs}. + * For example, if {@code Tin} is {@code qint8}, this must be set to 127. + * @param outputQuantizationMinVal The min value of the quantized data stored in {@code output}. + * For example, if {@code Tout} is {@code qint8}, this must be set to -127 if narrow range quantized or -128 if not. + * @param outputQuantizationMaxVal The max value of the quantized data stored in {@code output}. + * For example, if {@code Tout} is {@code qint8}, this must be set to 127. + * @param options carries optional attribute values + * @param data type for {@code UniformQuantizedConvolution} output and operands + * @param data type for {@code UniformQuantizedConvolution} output and operands + * @return a new instance of UniformQuantizedConvolution + */ + public UniformQuantizedConvolution uniformQuantizedConvolution( + Operand lhs, Operand rhs, Operand lhsScales, Operand lhsZeroPoints, + Operand rhsScales, Operand rhsZeroPoints, Operand outputScales, + Operand outputZeroPoints, Class Tout, String padding, Long lhsQuantizationMinVal, + Long lhsQuantizationMaxVal, Long rhsQuantizationMinVal, Long rhsQuantizationMaxVal, + Long outputQuantizationMinVal, Long outputQuantizationMaxVal, + UniformQuantizedConvolution.Options... options) { + return UniformQuantizedConvolution.create(scope, lhs, rhs, lhsScales, lhsZeroPoints, rhsScales, rhsZeroPoints, outputScales, outputZeroPoints, Tout, padding, lhsQuantizationMinVal, lhsQuantizationMaxVal, rhsQuantizationMinVal, rhsQuantizationMaxVal, outputQuantizationMinVal, outputQuantizationMaxVal, options); + } + + /** + * Perform hybrid quantized convolution of float Tensor {@code lhs} and quantized Tensor {@code rhs}. + * Given float {@code lhs} and quantized {@code rhs}, internally performs quantization on {@code lhs}, + * and then performs quantized convolution on quantized {@code lhs} and {@code rhs}. + *

The internal quantization on {@code lhs} is a quantization to {@code Trhs}, dynamic range, + * per-batch (per-axis along axis {@code dimension_numbers.input_batch_dimension}), asymmetric, + * and not narrow range (the range is [Trhs_MIN, Trhs_MAX]). + *

{@code lhs} and {@code rhs} must be Tensors of same rank, and meet following shape conditions. + *

    + *
  • lhs_feature % feature_group_count == 0
    • + *
  • lhs_feature % rhs_input_feature == 0
    • + *
  • lhs_feature / feature_group_count == rhs_input_feature
    • + *
  • rhs_output_feature % feature_group_count == 0
    • + *
  • lhs_batch % batch_group_count == 0
    • + *
  • rhs_output_feature % batch_group_count == 0
    • + *
+ *

{@code rhs} must be quantized Tensor, where its data value is quantized using the formula: + * quantized_data = clip(original_data / scale + zero_point, quantization_min_val, quantization_max_val). + * + * @param data type for {@code output} output + * @param lhs Must be a non-quantized Tensor of {@code Tlhs}, rank >= 3. + * @param rhs Must be a quantized Tensor of {@code Trhs}, same rank as {@code lhs}. + * @param rhsScales The float value(s) used as scale factors when quantizing the original data that {@code rhs} represents. + * Must be a scalar Tensor for per-tensor quantization, + * or 1D Tensor of size {@code rhs.dim_size(kernel_output_feature_dimension)}, for per-channel quantization. + * @param rhsZeroPoints The int32 value(s) used as zero_point when quantizing original data that {@code rhs} represents. + * Same shape condition as {@code rhs_scales}. + * @param Tout The type of output Tensor. + * @param padding string from: {@code "SAME"}, {@code "VALID"}, or {@code "EXPLICIT"}, indicating the type of padding algorithm to use. + * @param rhsQuantizationMinVal The min value of the quantized data stored in {@code rhs}. + * For example, if {@code Trhs} is qint8, this must be set to -127 if narrow range quantized or -128 if not. + * @param rhsQuantizationMaxVal The max value of the quantized data stored in {@code rhs}. + * For example, if {@code Trhs} is qint8, this must be set to 127. + * @param options carries optional attribute values + * @param data type for {@code UniformQuantizedConvolutionHybrid} output and operands + * @return a new instance of UniformQuantizedConvolutionHybrid + */ + public UniformQuantizedConvolutionHybrid uniformQuantizedConvolutionHybrid( + Operand lhs, Operand rhs, Operand rhsScales, + Operand rhsZeroPoints, Class Tout, String padding, Long rhsQuantizationMinVal, + Long rhsQuantizationMaxVal, UniformQuantizedConvolutionHybrid.Options... options) { + return UniformQuantizedConvolutionHybrid.create(scope, lhs, rhs, rhsScales, rhsZeroPoints, Tout, padding, rhsQuantizationMinVal, rhsQuantizationMaxVal, options); + } + /** * Get the parent {@link Ops} object. */ diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/Ops.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/Ops.java index 91171014690..bc83757228c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/Ops.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/Ops.java @@ -27,6 +27,7 @@ import org.tensorflow.ExecutionEnvironment; import org.tensorflow.Operand; import org.tensorflow.Operation; +import org.tensorflow.Tensor; import org.tensorflow.ndarray.BooleanNdArray; import org.tensorflow.ndarray.ByteNdArray; import org.tensorflow.ndarray.DoubleNdArray; @@ -45,6 +46,10 @@ import org.tensorflow.ndarray.index.Index; import org.tensorflow.op.core.Abort; import org.tensorflow.op.core.All; +import org.tensorflow.op.core.AnonymousHashTable; +import org.tensorflow.op.core.AnonymousMutableDenseHashTable; +import org.tensorflow.op.core.AnonymousMutableHashTable; +import org.tensorflow.op.core.AnonymousMutableHashTableOfTensors; import org.tensorflow.op.core.Any; import org.tensorflow.op.core.ApproxTopK; import org.tensorflow.op.core.AssertThat; @@ -68,11 +73,15 @@ import org.tensorflow.op.core.BooleanMask; import org.tensorflow.op.core.BooleanMaskUpdate; import org.tensorflow.op.core.BroadcastDynamicShape; +import org.tensorflow.op.core.BroadcastGradientArgs; import org.tensorflow.op.core.BroadcastTo; import org.tensorflow.op.core.Bucketize; import org.tensorflow.op.core.Case; import org.tensorflow.op.core.ClipByValue; +import org.tensorflow.op.core.CompositeTensorVariantFromComponents; +import org.tensorflow.op.core.CompositeTensorVariantToComponents; import org.tensorflow.op.core.Concat; +import org.tensorflow.op.core.ConcatOffset; import org.tensorflow.op.core.Constant; import org.tensorflow.op.core.ConsumeMutexLock; import org.tensorflow.op.core.ControlTrigger; @@ -84,6 +93,8 @@ import org.tensorflow.op.core.DeleteSessionTensor; import org.tensorflow.op.core.DestroyResourceOp; import org.tensorflow.op.core.DestroyTemporaryVariable; +import org.tensorflow.op.core.DeviceIndex; +import org.tensorflow.op.core.DummyMemoryCache; import org.tensorflow.op.core.DynamicPartition; import org.tensorflow.op.core.DynamicStitch; import org.tensorflow.op.core.EditDistance; @@ -92,14 +103,20 @@ import org.tensorflow.op.core.EmptyTensorMap; import org.tensorflow.op.core.EncodeProto; import org.tensorflow.op.core.EnsureShape; +import org.tensorflow.op.core.Enter; +import org.tensorflow.op.core.Exit; import org.tensorflow.op.core.ExpandDims; import org.tensorflow.op.core.ExtractVolumePatches; +import org.tensorflow.op.core.FakeParam; +import org.tensorflow.op.core.FileSystemSetConfiguration; import org.tensorflow.op.core.Fill; import org.tensorflow.op.core.Fingerprint; import org.tensorflow.op.core.For; import org.tensorflow.op.core.Function; import org.tensorflow.op.core.Gather; import org.tensorflow.op.core.GatherNd; +import org.tensorflow.op.core.GetElementAtIndex; +import org.tensorflow.op.core.GetOptions; import org.tensorflow.op.core.GetSessionHandle; import org.tensorflow.op.core.GetSessionTensor; import org.tensorflow.op.core.Gradients; @@ -107,6 +124,7 @@ import org.tensorflow.op.core.HashTable; import org.tensorflow.op.core.Helpers; import org.tensorflow.op.core.HistogramFixedWidth; +import org.tensorflow.op.core.HostConst; import org.tensorflow.op.core.Identity; import org.tensorflow.op.core.IdentityN; import org.tensorflow.op.core.If; @@ -118,14 +136,18 @@ import org.tensorflow.op.core.InplaceUpdate; import org.tensorflow.op.core.IsVariableInitialized; import org.tensorflow.op.core.KthOrderStatistic; +import org.tensorflow.op.core.LinSpace; import org.tensorflow.op.core.LookupTableExport; import org.tensorflow.op.core.LookupTableFind; import org.tensorflow.op.core.LookupTableImport; import org.tensorflow.op.core.LookupTableInsert; +import org.tensorflow.op.core.LookupTableRemove; import org.tensorflow.op.core.LookupTableSize; import org.tensorflow.op.core.LoopCond; +import org.tensorflow.op.core.LowerBound; import org.tensorflow.op.core.MakeUnique; import org.tensorflow.op.core.MapClear; +import org.tensorflow.op.core.MapDefun; import org.tensorflow.op.core.MapIncompleteSize; import org.tensorflow.op.core.MapPeek; import org.tensorflow.op.core.MapSize; @@ -136,12 +158,16 @@ import org.tensorflow.op.core.Merge; import org.tensorflow.op.core.Min; import org.tensorflow.op.core.MirrorPad; +import org.tensorflow.op.core.MirrorPadGrad; import org.tensorflow.op.core.MlirPassthroughOp; import org.tensorflow.op.core.MutableDenseHashTable; import org.tensorflow.op.core.MutableHashTable; import org.tensorflow.op.core.MutableHashTableOfTensors; import org.tensorflow.op.core.Mutex; import org.tensorflow.op.core.MutexLock; +import org.tensorflow.op.core.NcclAllReduce; +import org.tensorflow.op.core.NcclBroadcast; +import org.tensorflow.op.core.NcclReduce; import org.tensorflow.op.core.NextIteration; import org.tensorflow.op.core.NoOp; import org.tensorflow.op.core.OneHot; @@ -167,12 +193,17 @@ import org.tensorflow.op.core.Range; import org.tensorflow.op.core.Rank; import org.tensorflow.op.core.ReadVariableOp; +import org.tensorflow.op.core.Recv; import org.tensorflow.op.core.ReduceAll; import org.tensorflow.op.core.ReduceAny; import org.tensorflow.op.core.ReduceMax; import org.tensorflow.op.core.ReduceMin; import org.tensorflow.op.core.ReduceProd; import org.tensorflow.op.core.ReduceSum; +import org.tensorflow.op.core.RefEnter; +import org.tensorflow.op.core.RefExit; +import org.tensorflow.op.core.RefIdentity; +import org.tensorflow.op.core.RefMerge; import org.tensorflow.op.core.RefNextIteration; import org.tensorflow.op.core.RefSelect; import org.tensorflow.op.core.RefSwitch; @@ -206,12 +237,15 @@ import org.tensorflow.op.core.ScatterMul; import org.tensorflow.op.core.ScatterNd; import org.tensorflow.op.core.ScatterNdAdd; +import org.tensorflow.op.core.ScatterNdMax; +import org.tensorflow.op.core.ScatterNdMin; import org.tensorflow.op.core.ScatterNdNonAliasingAdd; import org.tensorflow.op.core.ScatterNdSub; import org.tensorflow.op.core.ScatterNdUpdate; import org.tensorflow.op.core.ScatterSub; import org.tensorflow.op.core.ScatterUpdate; import org.tensorflow.op.core.Select; +import org.tensorflow.op.core.Send; import org.tensorflow.op.core.SetDiff1d; import org.tensorflow.op.core.SetSize; import org.tensorflow.op.core.ShapeN; @@ -224,6 +258,10 @@ import org.tensorflow.op.core.SplitV; import org.tensorflow.op.core.Squeeze; import org.tensorflow.op.core.Stack; +import org.tensorflow.op.core.StackClose; +import org.tensorflow.op.core.StackCreate; +import org.tensorflow.op.core.StackPop; +import org.tensorflow.op.core.StackPush; import org.tensorflow.op.core.Stage; import org.tensorflow.op.core.StageClear; import org.tensorflow.op.core.StagePeek; @@ -232,8 +270,10 @@ import org.tensorflow.op.core.StatefulIf; import org.tensorflow.op.core.StatefulPartitionedCall; import org.tensorflow.op.core.StatefulWhile; +import org.tensorflow.op.core.StatelessCase; import org.tensorflow.op.core.StatelessIf; import org.tensorflow.op.core.StatelessWhile; +import org.tensorflow.op.core.StochasticCastToInt; import org.tensorflow.op.core.StopGradient; import org.tensorflow.op.core.StridedSlice; import org.tensorflow.op.core.StridedSliceAssign; @@ -241,6 +281,7 @@ import org.tensorflow.op.core.StridedSliceHelper; import org.tensorflow.op.core.Sum; import org.tensorflow.op.core.SwitchCond; +import org.tensorflow.op.core.SyncDevice; import org.tensorflow.op.core.TemporaryVariable; import org.tensorflow.op.core.TensorArray; import org.tensorflow.op.core.TensorArrayClose; @@ -290,11 +331,13 @@ import org.tensorflow.op.core.TopKWithUnique; import org.tensorflow.op.core.Unbatch; import org.tensorflow.op.core.UnbatchGrad; +import org.tensorflow.op.core.UniformQuantizedClipByValue; import org.tensorflow.op.core.Unique; import org.tensorflow.op.core.UniqueWithCounts; import org.tensorflow.op.core.UnravelIndex; import org.tensorflow.op.core.Unstack; import org.tensorflow.op.core.Unstage; +import org.tensorflow.op.core.UpperBound; import org.tensorflow.op.core.VarHandleOp; import org.tensorflow.op.core.VarIsInitializedOp; import org.tensorflow.op.core.Variable; @@ -348,68 +391,80 @@ public final class Ops { public final NnOps nn; - public final SummaryOps summary; - - public final ImageOps image; - - public final RaggedOps ragged; + public final ClusterOps cluster; public final DataOps data; - public final ShapeOps shape; - - public final IoOps io; - - public final DtypesOps dtypes; - - public final XlaOps xla; - - public final LinalgOps linalg; + public final MathOps math; public final RandomOps random; public final StringsOps strings; - public final SparseOps sparse; - - public final TpuOps tpu; - public final BitwiseOps bitwise; - public final MathOps math; + public final DebuggingOps debugging; + + public final CollectiveOps collective; public final AudioOps audio; + public final DistributeOps distribute; + public final SignalOps signal; public final TrainOps train; public final QuantizationOps quantization; + public final SummaryOps summary; + + public final ImageOps image; + + public final RaggedOps ragged; + + public final ShapeOps shape; + + public final IoOps io; + + public final DtypesOps dtypes; + + public final LinalgOps linalg; + + public final XlaOps xla; + + public final SparseOps sparse; + + public final TpuOps tpu; + private final Scope scope; Ops(Scope scope) { this.scope = scope; nn = new NnOps(this); + cluster = new ClusterOps(this); + data = new DataOps(this); + math = new MathOps(this); + random = new RandomOps(this); + strings = new StringsOps(this); + bitwise = new BitwiseOps(this); + debugging = new DebuggingOps(this); + collective = new CollectiveOps(this); + audio = new AudioOps(this); + distribute = new DistributeOps(this); + signal = new SignalOps(this); + train = new TrainOps(this); + quantization = new QuantizationOps(this); summary = new SummaryOps(this); image = new ImageOps(this); ragged = new RaggedOps(this); - data = new DataOps(this); shape = new ShapeOps(this); io = new IoOps(this); dtypes = new DtypesOps(this); - xla = new XlaOps(this); linalg = new LinalgOps(this); - random = new RandomOps(this); - strings = new StringsOps(this); + xla = new XlaOps(this); sparse = new SparseOps(this); tpu = new TpuOps(this); - bitwise = new BitwiseOps(this); - math = new MathOps(this); - audio = new AudioOps(this); - signal = new SignalOps(this); - train = new TrainOps(this); - quantization = new QuantizationOps(this); } /** @@ -442,6 +497,105 @@ public All all(Operand input, Operand axis, All.Option return All.create(scope, input, axis, options); } + /** + * Creates a uninitialized anonymous hash table. + * This op creates a new anonymous hash table (as a resource) everytime + * it is executed, with the specified dtype of its keys and values, + * returning the resource handle. Before using the table you will have + * to initialize it. After initialization the table will be + * immutable. The table is anonymous in the sense that it can only be + * accessed by the returned resource handle (e.g. it cannot be looked up + * by a name in a resource manager). The table will be automatically + * deleted when all resource handles pointing to it are gone. + * + * @param keyDtype Type of the table keys. + * @param valueDtype Type of the table values. + * @param data type for {@code AnonymousHashTable} output and operands + * @param data type for {@code AnonymousHashTable} output and operands + * @return a new instance of AnonymousHashTable + */ + public AnonymousHashTable anonymousHashTable(Class keyDtype, + Class valueDtype) { + return AnonymousHashTable.create(scope, keyDtype, valueDtype); + } + + /** + * Creates an empty anonymous mutable hash table that uses tensors as the backing store. + * This op creates a new anonymous mutable hash table (as a resource) everytime + * it is executed, with the specified dtype of its keys and values, + * returning the resource handle. Each value must be a scalar. + * Data can be inserted into the table using + * the insert operations. It does not support the initialization operation. + *

It uses "open addressing" with quadratic reprobing to resolve + * collisions. + *

The table is anonymous in the sense that it can only be + * accessed by the returned resource handle (e.g. it cannot be looked up + * by a name in a resource manager). The table will be automatically + * deleted when all resource handles pointing to it are gone. + * + * @param emptyKey The key used to represent empty key buckets internally. Must not + * be used in insert or lookup operations. + * @param deletedKey The deletedKey value + * @param valueDtype Type of the table values. + * @param options carries optional attribute values + * @param data type for {@code AnonymousMutableDenseHashTable} output and operands + * @param data type for {@code AnonymousMutableDenseHashTable} output and operands + * @return a new instance of AnonymousMutableDenseHashTable + */ + public AnonymousMutableDenseHashTable anonymousMutableDenseHashTable( + Operand emptyKey, Operand deletedKey, Class valueDtype, + AnonymousMutableDenseHashTable.Options... options) { + return AnonymousMutableDenseHashTable.create(scope, emptyKey, deletedKey, valueDtype, options); + } + + /** + * Creates an empty anonymous mutable hash table. + * This op creates a new anonymous mutable hash table (as a resource) everytime + * it is executed, with the specified dtype of its keys and values, + * returning the resource handle. Each value must be a scalar. + * Data can be inserted into the table using + * the insert operations. It does not support the initialization operation. + * The table is anonymous in the sense that it can only be + * accessed by the returned resource handle (e.g. it cannot be looked up + * by a name in a resource manager). The table will be automatically + * deleted when all resource handles pointing to it are gone. + * + * @param keyDtype Type of the table keys. + * @param valueDtype Type of the table values. + * @param data type for {@code AnonymousMutableHashTable} output and operands + * @param data type for {@code AnonymousMutableHashTable} output and operands + * @return a new instance of AnonymousMutableHashTable + */ + public AnonymousMutableHashTable anonymousMutableHashTable( + Class keyDtype, Class valueDtype) { + return AnonymousMutableHashTable.create(scope, keyDtype, valueDtype); + } + + /** + * Creates an empty anonymous mutable hash table of vector values. + * This op creates a new anonymous mutable hash table (as a resource) everytime + * it is executed, with the specified dtype of its keys and values, + * returning the resource handle. Each value must be a vector. + * Data can be inserted into the table using + * the insert operations. It does not support the initialization operation. + * The table is anonymous in the sense that it can only be + * accessed by the returned resource handle (e.g. it cannot be looked up + * by a name in a resource manager). The table will be automatically + * deleted when all resource handles pointing to it are gone. + * + * @param keyDtype Type of the table keys. + * @param valueDtype Type of the table values. + * @param options carries optional attribute values + * @param data type for {@code AnonymousMutableHashTableOfTensors} output and operands + * @param data type for {@code AnonymousMutableHashTableOfTensors} output and operands + * @return a new instance of AnonymousMutableHashTableOfTensors + */ + public AnonymousMutableHashTableOfTensors anonymousMutableHashTableOfTensors( + Class keyDtype, Class valueDtype, + AnonymousMutableHashTableOfTensors.Options... options) { + return AnonymousMutableHashTableOfTensors.create(scope, keyDtype, valueDtype, options); + } + /** * Computes the "logical or" of elements across dimensions of a tensor. * Reduces {@code input} along the dimensions given in {@code axis}. Unless @@ -477,72 +631,72 @@ public ApproxTopK approxTopK(Operand input, Long k, } /** - * Creates a constant of {@code String} elements, using the default UTF-8 charset. + * Creates a constant of {@code long} elements. * * @param data An array containing the values to put into the new constant. - * @return the {@code String} constant + * @return a long constant */ - public Constant array(String... data) { + public Constant array(long... data) { return Constant.arrayOf(scope, data); } /** - * Creates a constant of {@code int} elements. + * Creates a constant of {@code float} elements. * * @param data An array containing the values to put into the new constant. * @return a float constant */ - public Constant array(int... data) { + public Constant array(float... data) { return Constant.arrayOf(scope, data); } /** - * Creates a constant of {@code double} elements. + * Creates a constant of {@code String} elements, using the default UTF-8 charset. * * @param data An array containing the values to put into the new constant. - * @return a double constant + * @return the {@code String} constant */ - public Constant array(double... data) { + public Constant array(String... data) { return Constant.arrayOf(scope, data); } /** - * Creates a constant of {@code long} elements. + * Creates a constant of {@code int} elements. * * @param data An array containing the values to put into the new constant. - * @return a long constant + * @return a float constant */ - public Constant array(long... data) { + public Constant array(int... data) { return Constant.arrayOf(scope, data); } /** - * Creates a constant of {@code byte} elements. + * Creates a constant of {@code double} elements. * * @param data An array containing the values to put into the new constant. - * @return a byte constant + * @return a double constant */ - public Constant array(byte... data) { + public Constant array(double... data) { return Constant.arrayOf(scope, data); } /** - * Creates a constant of {@code boolean} elements. + * Creates a constant of {@code byte} elements. * * @param data An array containing the values to put into the new constant. - * @return a boolean constant + * @return a byte constant */ - public Constant array(boolean... data) { + public Constant array(byte... data) { return Constant.arrayOf(scope, data); } /** - * Creates a constant of {@code float} elements. + * Creates a constant of {@code boolean} elements. * * @param data An array containing the values to put into the new constant. - * @return a float constant + * @return a boolean constant */ - public Constant array(float... data) { + public Constant array(boolean... data) { return Constant.arrayOf(scope, data); } @@ -1149,6 +1303,21 @@ public BroadcastDynamicShape broadcastDynamicShape(Operan return BroadcastDynamicShape.create(scope, s0, s1); } + /** + * Return the reduction indices for computing gradients of s0 op s1 with broadcast. + * This is typically used by gradient computations for a broadcasting operation. + * + * @param data type for {@code r0} output + * @param s0 The s0 value + * @param s1 The s1 value + * @param data type for {@code BroadcastGradientArgs} output and operands + * @return a new instance of BroadcastGradientArgs + */ + public BroadcastGradientArgs broadcastGradientArgs(Operand s0, + Operand s1) { + return BroadcastGradientArgs.create(scope, s0, s1); + } + /** * Broadcast an array for a compatible shape. * Broadcasting is the process of making arrays to have compatible shapes @@ -1303,6 +1472,39 @@ public ClipByValue clipByValue(Operand t, Operand cli return ClipByValue.create(scope, t, clipValueMin, clipValueMax); } + /** + * Encodes an {@code ExtensionType} value into a {@code variant} scalar Tensor. + * Returns a scalar variant tensor containing a single {@code CompositeTensorVariant} + * with the specified Tensor components and TypeSpec. + * + * @param components The component tensors for the extension type value. + * @param metadata String serialization for the TypeSpec. (Note: the encoding for the TypeSpec + * may change in future versions of TensorFlow.) + * @return a new instance of CompositeTensorVariantFromComponents + */ + public CompositeTensorVariantFromComponents compositeTensorVariantFromComponents( + Iterable> components, String metadata) { + return CompositeTensorVariantFromComponents.create(scope, components, metadata); + } + + /** + * Decodes a {@code variant} scalar Tensor into an {@code ExtensionType} value. + * Returns the Tensor components encoded in a {@code CompositeTensorVariant}. + *

Raises an error if {@code type_spec_proto} doesn't match the TypeSpec + * in {@code encoded}. + * + * @param encoded A scalar {@code variant} Tensor containing an encoded ExtensionType value. + * @param metadata String serialization for the TypeSpec. Must be compatible with the + * {@code TypeSpec} contained in {@code encoded}. (Note: the encoding for the TypeSpec + * may change in future versions of TensorFlow.) + * @param Tcomponents Expected dtypes for components. + * @return a new instance of CompositeTensorVariantToComponents + */ + public CompositeTensorVariantToComponents compositeTensorVariantToComponents( + Operand encoded, String metadata, List> Tcomponents) { + return CompositeTensorVariantToComponents.create(scope, encoded, metadata, Tcomponents); + } + /** * Concatenates tensors along one dimension. * @@ -1319,6 +1521,34 @@ public Concat concat(Iterable> values, return Concat.create(scope, values, axis); } + /** + * Computes offsets of concat inputs within its output. + * For example: + *

+ *
+ *
+ *

x = [2, 2, 7] + * y = [2, 3, 7] + * z = [2, 9, 7] + * offsets = concat_offset(1, [x, y, z]) + * [list(off.numpy()) for off in offsets] + * [[0, 0, 0], [0, 2, 0], [0, 5, 0]] + *

+ *
+ *
+ *

This is typically used by gradient computations for a concat operation. + * + * @param data type for {@code offset} output + * @param concatDim The dimension along which to concatenate. + * @param shape The {@code N} int32 or int64 vectors representing shape of tensors being concatenated. + * @param data type for {@code ConcatOffset} output and operands + * @return a new instance of ConcatOffset + */ + public ConcatOffset concatOffset(Operand concatDim, + Iterable> shape) { + return ConcatOffset.create(scope, concatDim, shape); + } + /** * Creates a constant containing a single {@code int} element. * @@ -1351,17 +1581,6 @@ public Constant constant(byte[][][][][] data) { return Constant.tensorOf(scope, data); } - /** - * Creates a constant of {@code String} elements that is a copy of a given n-dimensional array, - * using the default UTF-8 encoding. - * - * @param data an n-dimensional array of {@code String} elements. - * @return a string constant - */ - public Constant constant(NdArray data) { - return Constant.tensorOf(scope, data); - } - /** * Creates a rank-4 constant of {@code int} elements. * @@ -1383,17 +1602,6 @@ public Constant constant(byte data) { return Constant.scalarOf(scope, data); } - /** - * Creates a rank-2 constant of {@code long} elements. - * - * @param data An array containing the values to put into the new constant. The dimensions of the - * new constant will match those of the array. - * @return a long constant - */ - public Constant constant(long[][] data) { - return Constant.tensorOf(scope, data); - } - /** * Creates a rank-6 constant of {@code float} elements. * @@ -1427,28 +1635,6 @@ public Constant constant(boolean[][][][] data) { return Constant.tensorOf(scope, data); } - /** - * Creates a rank-3 constant of {@code float} elements. - * - * @param data An array containing the values to put into the new constant. The dimensions of the - * new constant will match those of the array. - * @return a float constant - */ - public Constant constant(float[][][] data) { - return Constant.tensorOf(scope, data); - } - - /** - * Creates a rank-5 constant of {@code float} elements. - * - * @param data An array containing the values to put into the new constant. The dimensions of the - * new constant will match those of the array. - * @return a float constant - */ - public Constant constant(float[][][][][] data) { - return Constant.tensorOf(scope, data); - } - /** * Creates a rank-5 constant of {@code long} elements. * @@ -1472,217 +1658,304 @@ public Constant constant(int[] data) { } /** - * Creates a rank-2 constant of {@code float} elements. + * Creates a rank-2 constant of {@code boolean} elements. * * @param data An array containing the values to put into the new constant. The dimensions of the * new constant will match those of the array. - * @return a float constant + * @return a boolean constant */ - public Constant constant(float[][] data) { + public Constant constant(boolean[][] data) { return Constant.tensorOf(scope, data); } /** - * Creates a rank-2 constant of {@code boolean} elements. + * Creates a constant of {@code boolean} elements that is a copy of a given n-dimensional array. * - * @param data An array containing the values to put into the new constant. The dimensions of the - * new constant will match those of the array. + * @param data an n-dimensional array of {@code boolean} elements. * @return a boolean constant */ - public Constant constant(boolean[][] data) { + public Constant constant(BooleanNdArray data) { return Constant.tensorOf(scope, data); } /** - * Creates a constant containing a single {@code double} element. + * Creates a rank-1 constant of {@code double} elements. * - * @param data The value to put into the new constant. + * @param data An array containing the values to put into the new constant. The dimensions of the + * new constant will match those of the array. * @return a double constant */ - public Constant constant(double data) { - return Constant.scalarOf(scope, data); + public Constant constant(double[] data) { + return Constant.vectorOf(scope, data); } /** - * Creates a constant containing a single {@code boolean} element. + * Creates a constant of {@code long} elements that is a copy of a given n-dimensional array. * - * @param data The value to put into the new constant. - * @return a boolean constant + * @param data an n-dimensional array of {@code long} elements. + * @return a long constant */ - public Constant constant(boolean data) { - return Constant.scalarOf(scope, data); + public Constant constant(LongNdArray data) { + return Constant.tensorOf(scope, data); } /** - * Creates a constant containing a single {@code long} element. + * Creates a rank-3 constant of {@code long} elements. * - * @param data The value to put into the new constant. + * @param data An array containing the values to put into the new constant. The dimensions of the + * new constant will match those of the array. * @return a long constant */ - public Constant constant(long data) { - return Constant.scalarOf(scope, data); + public Constant constant(long[][][] data) { + return Constant.tensorOf(scope, data); } /** - * Creates a {@code String} constant using the default, UTF-8 encoding. + * Creates a rank-1 constant of {@code byte} elements. * - * @param data The string to put into the new constant. - * @return a string constant + * @param data An array containing the values to put into the new constant. The dimensions of the + * new constant will match those of the array. + * @return a byte constant */ - public Constant constant(String data) { - return Constant.scalarOf(scope, data); + public Constant constant(byte[] data) { + return Constant.vectorOf(scope, data); } /** - * Creates a constant of {@code boolean} elements that is a copy of a given n-dimensional array. + * Creates a constant of {@code float} elements that is a copy of a given n-dimensional array. * - * @param data an n-dimensional array of {@code boolean} elements. - * @return a boolean constant + * @param data an n-dimensional array of {@code float} elements. + * @return a float constant */ - public Constant constant(BooleanNdArray data) { + public Constant constant(FloatNdArray data) { return Constant.tensorOf(scope, data); } /** - * Creates a rank-1 constant of {@code double} elements. + * Creates a rank-5 constant of {@code int} elements. * * @param data An array containing the values to put into the new constant. The dimensions of the * new constant will match those of the array. - * @return a double constant + * @return an integer constant */ - public Constant constant(double[] data) { - return Constant.vectorOf(scope, data); + public Constant constant(int[][][][][] data) { + return Constant.tensorOf(scope, data); } /** - * Creates a constant of {@code long} elements that is a copy of a given n-dimensional array. + * Creates a rank-5 constant of {@code double} elements. * - * @param data an n-dimensional array of {@code long} elements. - * @return a long constant + * @param data An array containing the values to put into the new constant. The dimensions of the + * new constant will match those of the array. + * @return a double constant */ - public Constant constant(LongNdArray data) { + public Constant constant(double[][][][][] data) { return Constant.tensorOf(scope, data); } /** - * Creates a rank-1 constant of {@code float} elements. + * Creates a rank-5 constant of {@code boolean} elements. * * @param data An array containing the values to put into the new constant. The dimensions of the * new constant will match those of the array. + * @return a boolean constant + */ + public Constant constant(boolean[][][][][] data) { + return Constant.tensorOf(scope, data); + } + + /** + * Creates a constant containing a single {@code float} element. + * + * @param data The value to put into the new constant. * @return a float constant */ - public Constant constant(float[] data) { - return Constant.vectorOf(scope, data); + public Constant constant(float data) { + return Constant.scalarOf(scope, data); } /** - * Creates a rank-3 constant of {@code long} elements. + * Creates a rank-2 constant of {@code byte} elements. * * @param data An array containing the values to put into the new constant. The dimensions of the * new constant will match those of the array. - * @return a long constant + * @return a byte constant */ - public Constant constant(long[][][] data) { + public Constant constant(byte[][] data) { return Constant.tensorOf(scope, data); } /** - * Creates a rank-3 constant of {@code boolean} elements. + * Creates a rank-2 constant of {@code double} elements. * * @param data An array containing the values to put into the new constant. The dimensions of the * new constant will match those of the array. - * @return a boolean constant + * @return a double constant */ - public Constant constant(boolean[][][] data) { + public Constant constant(double[][] data) { return Constant.tensorOf(scope, data); } /** - * Creates a rank-1 constant of {@code byte} elements. + * Creates a rank-3 constant of {@code byte} elements. * * @param data An array containing the values to put into the new constant. The dimensions of the * new constant will match those of the array. * @return a byte constant */ - public Constant constant(byte[] data) { - return Constant.vectorOf(scope, data); + public Constant constant(byte[][][] data) { + return Constant.tensorOf(scope, data); } /** - * Creates a rank-3 constant of {@code int} elements. + * Creates a constant of {@code String} elements that is a copy of a given n-dimensional array, + * using the default UTF-8 encoding. + * + * @param data an n-dimensional array of {@code String} elements. + * @return a string constant + */ + public Constant constant(NdArray data) { + return Constant.tensorOf(scope, data); + } + + /** + * Creates a rank-2 constant of {@code long} elements. * * @param data An array containing the values to put into the new constant. The dimensions of the * new constant will match those of the array. - * @return an integer constant + * @return a long constant */ - public Constant constant(int[][][] data) { + public Constant constant(long[][] data) { return Constant.tensorOf(scope, data); } /** - * Creates a constant of {@code int} elements that is a copy of a given n-dimensional array. + * Creates a rank-3 constant of {@code float} elements. * - * @param data an n-dimensional array of {@code int} elements. - * @return an integer constant + * @param data An array containing the values to put into the new constant. The dimensions of the + * new constant will match those of the array. + * @return a float constant */ - public Constant constant(IntNdArray data) { + public Constant constant(float[][][] data) { return Constant.tensorOf(scope, data); } /** - * Creates a rank-1 constant of {@code long} elements. + * Creates a rank-5 constant of {@code float} elements. * * @param data An array containing the values to put into the new constant. The dimensions of the * new constant will match those of the array. - * @return a long constant + * @return a float constant */ - public Constant constant(long[] data) { - return Constant.vectorOf(scope, data); + public Constant constant(float[][][][][] data) { + return Constant.tensorOf(scope, data); } /** - * Creates a constant of {@code float} elements that is a copy of a given n-dimensional array. + * Creates a rank-2 constant of {@code float} elements. * - * @param data an n-dimensional array of {@code float} elements. + * @param data An array containing the values to put into the new constant. The dimensions of the + * new constant will match those of the array. * @return a float constant */ - public Constant constant(FloatNdArray data) { + public Constant constant(float[][] data) { return Constant.tensorOf(scope, data); } /** - * Creates a rank-5 constant of {@code int} elements. + * Creates a constant containing a single {@code double} element. + * + * @param data The value to put into the new constant. + * @return a double constant + */ + public Constant constant(double data) { + return Constant.scalarOf(scope, data); + } + + /** + * Creates a constant containing a single {@code boolean} element. + * + * @param data The value to put into the new constant. + * @return a boolean constant + */ + public Constant constant(boolean data) { + return Constant.scalarOf(scope, data); + } + + /** + * Creates a constant containing a single {@code long} element. + * + * @param data The value to put into the new constant. + * @return a long constant + */ + public Constant constant(long data) { + return Constant.scalarOf(scope, data); + } + + /** + * Creates a {@code String} constant using the default, UTF-8 encoding. + * + * @param data The string to put into the new constant. + * @return a string constant + */ + public Constant constant(String data) { + return Constant.scalarOf(scope, data); + } + + /** + * Creates a rank-1 constant of {@code float} elements. * * @param data An array containing the values to put into the new constant. The dimensions of the * new constant will match those of the array. - * @return an integer constant + * @return a float constant */ - public Constant constant(int[][][][][] data) { - return Constant.tensorOf(scope, data); + public Constant constant(float[] data) { + return Constant.vectorOf(scope, data); } /** - * Creates a rank-5 constant of {@code double} elements. + * Creates a rank-3 constant of {@code boolean} elements. * * @param data An array containing the values to put into the new constant. The dimensions of the * new constant will match those of the array. - * @return a double constant + * @return a boolean constant */ - public Constant constant(double[][][][][] data) { + public Constant constant(boolean[][][] data) { return Constant.tensorOf(scope, data); } /** - * Creates a rank-5 constant of {@code boolean} elements. + * Creates a rank-3 constant of {@code int} elements. * * @param data An array containing the values to put into the new constant. The dimensions of the * new constant will match those of the array. - * @return a boolean constant + * @return an integer constant */ - public Constant constant(boolean[][][][][] data) { + public Constant constant(int[][][] data) { + return Constant.tensorOf(scope, data); + } + + /** + * Creates a constant of {@code int} elements that is a copy of a given n-dimensional array. + * + * @param data an n-dimensional array of {@code int} elements. + * @return an integer constant + */ + public Constant constant(IntNdArray data) { return Constant.tensorOf(scope, data); } + /** + * Creates a rank-1 constant of {@code long} elements. + * + * @param data An array containing the values to put into the new constant. The dimensions of the + * new constant will match those of the array. + * @return a long constant + */ + public Constant constant(long[] data) { + return Constant.vectorOf(scope, data); + } + /** * Creates a rank-6 constant of {@code int} elements. * @@ -1748,16 +2021,6 @@ public Constant constant(boolean[] data) { return Constant.vectorOf(scope, data); } - /** - * Creates a constant containing a single {@code float} element. - * - * @param data The value to put into the new constant. - * @return a float constant - */ - public Constant constant(float data) { - return Constant.scalarOf(scope, data); - } - /** * Creates a rank-4 constant of {@code byte} elements. * @@ -1812,39 +2075,6 @@ public Constant constant(long[][][][] data) { return Constant.tensorOf(scope, data); } - /** - * Creates a rank-2 constant of {@code byte} elements. - * - * @param data An array containing the values to put into the new constant. The dimensions of the - * new constant will match those of the array. - * @return a byte constant - */ - public Constant constant(byte[][] data) { - return Constant.tensorOf(scope, data); - } - - /** - * Creates a rank-2 constant of {@code double} elements. - * - * @param data An array containing the values to put into the new constant. The dimensions of the - * new constant will match those of the array. - * @return a double constant - */ - public Constant constant(double[][] data) { - return Constant.tensorOf(scope, data); - } - - /** - * Creates a rank-3 constant of {@code byte} elements. - * - * @param data An array containing the values to put into the new constant. The dimensions of the - * new constant will match those of the array. - * @return a byte constant - */ - public Constant constant(byte[][][] data) { - return Constant.tensorOf(scope, data); - } - /** * Creates a rank-4 constant of {@code double} elements. * @@ -1952,26 +2182,26 @@ public Constant constant(Shape shape, IntDataBuffer data) { } /** - * Create a {@link TInt64} constant with data from the given buffer. + * Create a {@link TFloat64} constant with data from the given buffer. * * @param shape the tensor shape. * @param data a buffer containing the tensor data. - * @return a long constant + * @return a double constant * @throws IllegalArgumentException If the tensor shape is not compatible with the buffer */ - public Constant constant(Shape shape, LongDataBuffer data) { + public Constant constant(Shape shape, DoubleDataBuffer data) { return Constant.tensorOf(scope, shape, data); } /** - * Create a {@link TFloat64} constant with data from the given buffer. + * Create a {@link TInt64} constant with data from the given buffer. * * @param shape the tensor shape. * @param data a buffer containing the tensor data. - * @return a double constant + * @return a long constant * @throws IllegalArgumentException If the tensor shape is not compatible with the buffer */ - public Constant constant(Shape shape, DoubleDataBuffer data) { + public Constant constant(Shape shape, LongDataBuffer data) { return Constant.tensorOf(scope, shape, data); } @@ -2264,6 +2494,29 @@ public DestroyTemporaryVariable destroyTemporaryVariable(Op return DestroyTemporaryVariable.create(scope, ref, varName); } + /** + * Return the index of device the op runs. + * Given a list of device names, this operation returns the index of the device + * this op runs. The length of the list is returned in two cases: + * (1) Device does not exist in the given device list. + * (2) It is in XLA compilation. + * + * @param deviceNames The value of the deviceNames attribute + * @return a new instance of DeviceIndex + */ + public DeviceIndex deviceIndex(List deviceNames) { + return DeviceIndex.create(scope, deviceNames); + } + + /** + * The DummyMemoryCache operation + * + * @return a new instance of DummyMemoryCache + */ + public DummyMemoryCache dummyMemoryCache() { + return DummyMemoryCache.create(scope); + } + /** * Partitions {@code data} into {@code num_partitions} tensors using indices from {@code partitions}. * For each index tuple {@code js} of size {@code partitions.ndim}, the slice {@code data[js, ...]} @@ -2526,13 +2779,46 @@ public EncodeProto encodeProto(Operand sizes, Iterable> value * Returns the input tensor otherwise. * * @param data type for {@code output} output - * @param input A tensor, whose shape is to be validated. - * @param shape The expected (possibly partially specified) shape of the input tensor. - * @param data type for {@code EnsureShape} output and operands - * @return a new instance of EnsureShape + * @param input A tensor, whose shape is to be validated. + * @param shape The expected (possibly partially specified) shape of the input tensor. + * @param data type for {@code EnsureShape} output and operands + * @return a new instance of EnsureShape + */ + public EnsureShape ensureShape(Operand input, Shape shape) { + return EnsureShape.create(scope, input, shape); + } + + /** + * Creates or finds a child frame, and makes {@code data} available to the child frame. + * This op is used together with {@code Exit} to create loops in the graph. + * The unique {@code frame_name} is used by the {@code Executor} to identify frames. If + * {@code is_constant} is true, {@code output} is a constant in the child frame; otherwise + * it may be changed in the child frame. At most {@code parallel_iterations} iterations + * are run in parallel in the child frame. + * + * @param data type for {@code output} output + * @param data The tensor to be made available to the child frame. + * @param frameName The name of the child frame. + * @param options carries optional attribute values + * @param data type for {@code Enter} output and operands + * @return a new instance of Enter + */ + public Enter enter(Operand data, String frameName, + Enter.Options... options) { + return Enter.create(scope, data, frameName, options); + } + + /** + * Exits the current frame to its parent frame. + * Exit makes its input {@code data} available to the parent frame. + * + * @param data type for {@code output} output + * @param data The tensor to be made available to the parent frame. + * @param data type for {@code Exit} output and operands + * @return a new instance of Exit */ - public EnsureShape ensureShape(Operand input, Shape shape) { - return EnsureShape.create(scope, input, shape); + public Exit exit(Operand data) { + return Exit.create(scope, data); } /** @@ -2596,6 +2882,38 @@ public ExtractVolumePatches extractVolumePatches(Operand< return ExtractVolumePatches.create(scope, input, ksizes, strides, padding); } + /** + * This op is used as a placeholder in If branch functions. It doesn't provide a + * valid output when run, so must either be removed (e.g. replaced with a + * function input) or guaranteed not to be used (e.g. if mirroring an + * intermediate output needed for the gradient computation of the other branch). + * + * @param data type for {@code output} output + * @param dtype The type of the output. + * @param shape 

+   *  The purported shape of the output. This is only used for shape inference;
+   *  the output will not necessarily have this shape. Can be a partial shape.
+   *
+ * @param data type for {@code FakeParam} output and operands + * @return a new instance of FakeParam + */ + public FakeParam fakeParam(Class dtype, Shape shape) { + return FakeParam.create(scope, dtype, shape); + } + + /** + * Set configuration of the file system. + * + * @param scheme File system scheme. + * @param key The name of the configuration option. + * @param value The value of the configuration option. + * @return a new instance of FileSystemSetConfiguration + */ + public FileSystemSetConfiguration fileSystemSetConfiguration(Operand scheme, + Operand key, Operand value) { + return FileSystemSetConfiguration.create(scope, scheme, key, value); + } + /** * Creates a tensor filled with a scalar value. * This operation creates a tensor of shape {@code dims} and fills it with {@code value}. @@ -2830,6 +3148,30 @@ public GatherNd gatherNd(Operand params, return GatherNd.create(scope, params, indices); } + /** + * Gets the element at the specified index in a dataset. + * + * @param dataset The dataset value + * @param index The index value + * @param outputTypes The value of the outputTypes attribute + * @param outputShapes The value of the outputShapes attribute + * @return a new instance of GetElementAtIndex + */ + public GetElementAtIndex getElementAtIndex(Operand dataset, + Operand index, List> outputTypes, List outputShapes) { + return GetElementAtIndex.create(scope, dataset, index, outputTypes, outputShapes); + } + + /** + * Returns the {@code tf.data.Options} attached to {@code input_dataset}. + * + * @param inputDataset A variant tensor representing the input dataset. + * @return a new instance of GetOptions + */ + public GetOptions getOptions(Operand inputDataset) { + return GetOptions.create(scope, inputDataset); + } + /** * Store the input tensor in the state of the current session. * @@ -2999,6 +3341,19 @@ public HistogramFixedWidth histogramFi return HistogramFixedWidth.create(scope, values, valueRange, nbins, dtype); } + /** + * Returns a constant tensor on the host. Only for writing C++ tests. + * + * @param data type for {@code output} output + * @param value Attr {@code value} is the tensor to return. + * @param dtype The value of the dtype attribute + * @param data type for {@code HostConst} output and operands + * @return a new instance of HostConst + */ + public HostConst hostConst(Tensor value, Class dtype) { + return HostConst.create(scope, value, dtype); + } + /** * Return a tensor with the same shape and contents as the input tensor or value. * @@ -3213,6 +3568,28 @@ public KthOrderStatistic kthOrderStatistic(Operand input, Long k) { return KthOrderStatistic.create(scope, input, k); } + /** + * Generates values in an interval. + * A sequence of {@code num} evenly-spaced values are generated beginning at {@code start}. + * If {@code num > 1}, the values in the sequence increase by + * {@code (stop - start) / (num - 1)}, so that the last one is exactly {@code stop}. + *

For example: + * 

+   *  tf.linspace(10.0, 12.0, 3, name="linspace") => [ 10.0  11.0  12.0]
+   *
+ * + * @param data type for {@code output} output + * @param start 0-D tensor. First entry in the range. + * @param stop 0-D tensor. Last entry in the range. + * @param num 0-D tensor. Number of values to generate. + * @param data type for {@code LinSpace} output and operands + * @return a new instance of LinSpace + */ + public LinSpace linSpace(Operand start, Operand stop, + Operand num) { + return LinSpace.create(scope, start, stop, num); + } + /** * Outputs all keys and values in the table. * @@ -3279,6 +3656,20 @@ public LookupTableInsert lookupTableInsert(Operand tableHandle, return LookupTableInsert.create(scope, tableHandle, keys, values); } + /** + * Removes keys and its associated values from a table. + * The tensor {@code keys} must of the same type as the keys of the table. Keys not + * already in the table are silently ignored. + * + * @param tableHandle Handle to the table. + * @param keys Any shape. Keys of the elements to remove. + * @return a new instance of LookupTableRemove + */ + public LookupTableRemove lookupTableRemove(Operand tableHandle, + Operand keys) { + return LookupTableRemove.create(scope, tableHandle, keys); + } + /** * Computes the number of elements in the given table. * @@ -3301,6 +3692,64 @@ public LoopCond loopCond(Operand input) { return LoopCond.create(scope, input); } + /** + * Applies lower_bound(sorted_search_values, values) along each row. + * Each set of rows with the same index in (sorted_inputs, values) is treated + * independently. The resulting row is the equivalent of calling + * {@code np.searchsorted(sorted_inputs, values, side='left')}. + *

The result is not a global index to the entire + * {@code Tensor}, but rather just the index in the last dimension. + *

A 2-D example: + * sorted_sequence = [[0, 3, 9, 9, 10], + * [1, 2, 3, 4, 5]] + * values = [[2, 4, 9], + * [0, 2, 6]] + *

result = LowerBound(sorted_sequence, values) + *

result == [[1, 2, 2], + * [0, 1, 5]] + * + * @param data type for {@code output} output + * @param sortedInputs 2-D Tensor where each row is ordered. + * @param values 2-D Tensor with the same numbers of rows as {@code sorted_search_values}. Contains + * the values that will be searched for in {@code sorted_search_values}. + * @param data type for {@code LowerBound} output and operands + * @return a new instance of LowerBound, with default output types + */ + public LowerBound lowerBound(Operand sortedInputs, + Operand values) { + return LowerBound.create(scope, sortedInputs, values); + } + + /** + * Applies lower_bound(sorted_search_values, values) along each row. + * Each set of rows with the same index in (sorted_inputs, values) is treated + * independently. The resulting row is the equivalent of calling + * {@code np.searchsorted(sorted_inputs, values, side='left')}. + *

The result is not a global index to the entire + * {@code Tensor}, but rather just the index in the last dimension. + *

A 2-D example: + * sorted_sequence = [[0, 3, 9, 9, 10], + * [1, 2, 3, 4, 5]] + * values = [[2, 4, 9], + * [0, 2, 6]] + *

result = LowerBound(sorted_sequence, values) + *

result == [[1, 2, 2], + * [0, 1, 5]] + * + * @param data type for {@code output} output + * @param sortedInputs 2-D Tensor where each row is ordered. + * @param values 2-D Tensor with the same numbers of rows as {@code sorted_search_values}. Contains + * the values that will be searched for in {@code sorted_search_values}. + * @param outType The value of the outType attribute + * @param data type for {@code LowerBound} output and operands + * @param data type for {@code LowerBound} output and operands + * @return a new instance of LowerBound + */ + public LowerBound lowerBound(Operand sortedInputs, + Operand values, Class outType) { + return LowerBound.create(scope, sortedInputs, values, outType); + } + /** * Make all elements in the non-Batch dimension unique, but "close" to * their initial value. Never returns a sub-normal number. Never returns @@ -3326,6 +3775,38 @@ public MapClear mapClear(List> dtypes, MapClear.Options.. return MapClear.create(scope, dtypes, options); } + /** + * Maps a function on the list of tensors unpacked from arguments on dimension 0. + * The function given by {@code f} is assumed to be stateless, and is executed + * concurrently on all the slices; up to batch_size (i.e. the size of the 0th + * dimension of each argument) functions will be scheduled at once. + *

The {@code max_intra_op_parallelism} attr, which defaults to 1, can be used to + * limit the intra op parallelism. To limit inter-op parallelism, a user can + * set a private threadpool on the dataset using {@code tf.data.Options}'s + * {@code ThreadingOptions}. + *

Note that this op is not exposed to users directly, but is invoked in tf.data + * rewrites. + * + * @param arguments 

+   *  A list of tensors whose types are `Targuments`, corresponding to the inputs
+   *  the function should be mapped over.
+   *
+ * @param capturedInputs 
+   *  A list of tensors whose types are `Tcaptured`, corresponding to the captured
+   *  inputs of the defun.
+   *
+ * @param outputTypes A list of types. + * @param outputShapes A list of shapes. + * @param f The value of the f attribute + * @param options carries optional attribute values + * @return a new instance of MapDefun + */ + public MapDefun mapDefun(Iterable> arguments, Iterable> capturedInputs, + List> outputTypes, List outputShapes, ConcreteFunction f, + MapDefun.Options... options) { + return MapDefun.create(scope, arguments, capturedInputs, outputTypes, outputShapes, f, options); + } + /** * Op returns the number of incomplete elements in the underlying container. * @@ -3510,6 +3991,36 @@ public MirrorPad mirrorPad(Operand input, return MirrorPad.create(scope, input, paddings, mode); } + /** + * Gradient op for {@code MirrorPad} op. This op folds a mirror-padded tensor. + * This operation folds the padded areas of {@code input} by {@code MirrorPad} according to the + * {@code paddings} you specify. {@code paddings} must be the same as {@code paddings} argument + * given to the corresponding {@code MirrorPad} op. + *

The folded size of each dimension D of the output is: + *

{@code input.dim_size(D) - paddings(D, 0) - paddings(D, 1)} + *

For example: + * 

+   *  # 't' is [[1, 2, 3], [4, 5, 6], [7, 8, 9]].
+   *  # 'paddings' is [[0, 1]], [0, 1]].
+   *  # 'mode' is SYMMETRIC.
+   *  # rank of 't' is 2.
+   *  pad(t, paddings) ==> [[ 1,  5]
+   *                        [11, 28]]
+   *
+ * + * @param data type for {@code output} output + * @param input The input tensor to be folded. + * @param paddings A two-column matrix specifying the padding sizes. The number of + * rows must be the same as the rank of {@code input}. + * @param mode The mode used in the {@code MirrorPad} op. + * @param data type for {@code MirrorPadGrad} output and operands + * @return a new instance of MirrorPadGrad + */ + public MirrorPadGrad mirrorPadGrad(Operand input, + Operand paddings, String mode) { + return MirrorPadGrad.create(scope, input, paddings, mode); + } + /** * Wraps an arbitrary MLIR computation expressed as a module with a main() function. * This operation does not have an associated kernel and is not intended to be @@ -3663,6 +4174,77 @@ public MutexLock mutexLock(Operand mutex) { return MutexLock.create(scope, mutex); } + /** + * Outputs a tensor containing the reduction across all input tensors. + * Outputs a tensor containing the reduction across all input tensors passed to ops + * within the same `shared_name. + *

The graph should be constructed so if one op runs with shared_name value {@code c}, + * then {@code num_devices} ops will run with shared_name value {@code c}. Failure to do so + * will cause the graph execution to fail to complete. + *

input: the input to the reduction + * data: the value of the reduction across all {@code num_devices} devices. + * reduction: the reduction operation to perform. + * num_devices: The number of devices participating in this reduction. + * shared_name: Identifier that shared between ops of the same reduction. + * + * @param data type for {@code data} output + * @deprecated use {@link org.tensorflow.op.distribute.NcclAllReduce} instead + * @param input The input value + * @param reduction The value of the reduction attribute + * @param numDevices The value of the numDevices attribute + * @param sharedName The value of the sharedName attribute + * @param data type for {@code NcclAllReduce} output and operands + * @return a new instance of NcclAllReduce + */ + @Deprecated + public NcclAllReduce ncclAllReduce(Operand input, String reduction, + Long numDevices, String sharedName) { + return NcclAllReduce.create(scope, input, reduction, numDevices, sharedName); + } + + /** + * Sends {@code input} to all devices that are connected to the output. + * Sends {@code input} to all devices that are connected to the output. + *

The graph should be constructed so that all ops connected to the output have a + * valid device assignment, and the op itself is assigned one of these devices. + *

input: The input to the broadcast. + * output: The same as input. + * shape: The shape of the input tensor. + * + * @param data type for {@code output} output + * @deprecated use {@link org.tensorflow.op.distribute.NcclBroadcast} instead + * @param input The input value + * @param shape The value of the shape attribute + * @param data type for {@code NcclBroadcast} output and operands + * @return a new instance of NcclBroadcast + */ + @Deprecated + public NcclBroadcast ncclBroadcast(Operand input, Shape shape) { + return NcclBroadcast.create(scope, input, shape); + } + + /** + * Reduces {@code input} from {@code num_devices} using {@code reduction} to a single device. + * Reduces {@code input} from {@code num_devices} using {@code reduction} to a single device. + *

The graph should be constructed so that all inputs have a valid device + * assignment, and the op itself is assigned one of these devices. + *

input: The input to the reduction. + * data: the value of the reduction across all {@code num_devices} devices. + * reduction: the reduction operation to perform. + * + * @param data type for {@code data} output + * @deprecated use {@link org.tensorflow.op.distribute.NcclReduce} instead + * @param input The input value + * @param reduction The value of the reduction attribute + * @param data type for {@code NcclReduce} output and operands + * @return a new instance of NcclReduce + */ + @Deprecated + public NcclReduce ncclReduce(Iterable> input, + String reduction) { + return NcclReduce.create(scope, input, reduction); + } + /** * Makes its input available to the next iteration. * @@ -4214,6 +4796,24 @@ public ReadVariableOp readVariableOp(Operand data type for {@code tensor} output + * @param tensorType The value of the tensorType attribute + * @param tensorName The name of the tensor to receive. + * @param sendDevice The name of the device sending the tensor. + * @param sendDeviceIncarnation The current incarnation of send_device. + * @param recvDevice The name of the device receiving the tensor. + * @param options carries optional attribute values + * @param data type for {@code Recv} output and operands + * @return a new instance of Recv + */ + public Recv recv(Class tensorType, String tensorName, String sendDevice, + Long sendDeviceIncarnation, String recvDevice, Recv.Options... options) { + return Recv.create(scope, tensorType, tensorName, sendDevice, sendDeviceIncarnation, recvDevice, options); + } + /** * Computes the "logical and" of elements across dimensions of a tensor. * Reduces {@code input} along the dimensions given in {@code axis}. Unless @@ -4330,6 +4930,66 @@ public ReduceSum reduceSum(Operand input, Operand data type for {@code output} output + * @param data The tensor to be made available to the child frame. + * @param frameName The name of the child frame. + * @param options carries optional attribute values + * @param data type for {@code RefEnter} output and operands + * @return a new instance of RefEnter + */ + public RefEnter refEnter(Operand data, String frameName, + RefEnter.Options... options) { + return RefEnter.create(scope, data, frameName, options); + } + + /** + * Exits the current frame to its parent frame. + * Exit makes its input {@code data} available to the parent frame. + * + * @param data type for {@code output} output + * @param data The tensor to be made available to the parent frame. + * @param data type for {@code RefExit} output and operands + * @return a new instance of RefExit + */ + public RefExit refExit(Operand data) { + return RefExit.create(scope, data); + } + + /** + * Return the same ref tensor as the input ref tensor. + * + * @param data type for {@code output} output + * @param input The input value + * @param data type for {@code RefIdentity} output and operands + * @return a new instance of RefIdentity + */ + public RefIdentity refIdentity(Operand input) { + return RefIdentity.create(scope, input); + } + + /** + * Forwards the value of an available tensor from {@code inputs} to {@code output}. + * {@code Merge} waits for at least one of the tensors in {@code inputs} to become available. + * It is usually combined with {@code Switch} to implement branching. + *

{@code Merge} forwards the first tensor for become available to {@code output}, and sets + * {@code value_index} to its index in {@code inputs}. + * + * @param data type for {@code output} output + * @param inputs The input tensors, exactly one of which will become available. + * @param data type for {@code RefMerge} output and operands + * @return a new instance of RefMerge + */ + public RefMerge refMerge(Iterable> inputs) { + return RefMerge.create(scope, inputs); + } + /** * Makes its input available to the next iteration. * @@ -5390,6 +6050,42 @@ public ScatterNdAdd scatterNdAdd(Operand ref, return ScatterNdAdd.create(scope, ref, indices, updates, options); } + /** + * Computes element-wise maximum. + * + * @param data type for {@code output_ref} output + * @param ref A mutable Tensor. Should be from a Variable node. + * @param indices A Tensor. Must be one of the following types: int32, int64. + * A tensor of indices into ref. + * @param updates A Tensor. Must have the same type as ref. A tensor of updated values + * to add to ref. + * @param options carries optional attribute values + * @param data type for {@code ScatterNdMax} output and operands + * @return a new instance of ScatterNdMax + */ + public ScatterNdMax scatterNdMax(Operand ref, + Operand indices, Operand updates, ScatterNdMax.Options... options) { + return ScatterNdMax.create(scope, ref, indices, updates, options); + } + + /** + * Computes element-wise minimum. + * + * @param data type for {@code output_ref} output + * @param ref A mutable Tensor. Should be from a Variable node. + * @param indices A Tensor. Must be one of the following types: int32, int64. + * A tensor of indices into ref. + * @param updates A Tensor. Must have the same type as ref. A tensor of updated values + * to add to ref. + * @param options carries optional attribute values + * @param data type for {@code ScatterNdMin} output and operands + * @return a new instance of ScatterNdMin + */ + public ScatterNdMin scatterNdMin(Operand ref, + Operand indices, Operand updates, ScatterNdMin.Options... options) { + return ScatterNdMin.create(scope, ref, indices, updates, options); + } + /** * Applies sparse addition to {@code input} using individual values or slices * from {@code updates} according to indices {@code indices}. The updates are non-aliasing: @@ -5608,6 +6304,22 @@ public Select select(Operand condition, Operand t return Select.create(scope, condition, t, e); } + /** + * Sends the named tensor from send_device to recv_device. + * + * @param tensor The tensor to send. + * @param tensorName The name of the tensor to send. + * @param sendDevice The name of the device sending the tensor. + * @param sendDeviceIncarnation The current incarnation of send_device. + * @param recvDevice The name of the device receiving the tensor. + * @param options carries optional attribute values + * @return a new instance of Send + */ + public Send send(Operand tensor, String tensorName, String sendDevice, + Long sendDeviceIncarnation, String recvDevice, Send.Options... options) { + return Send.create(scope, tensor, tensorName, sendDevice, sendDeviceIncarnation, recvDevice, options); + } + /** * Computes the difference between two lists of numbers or strings. * Given a list {@code x} and a list {@code y}, this operation returns a list {@code out} that @@ -6047,6 +6759,60 @@ public Stack stack(Iterable> values, Stack.Optio return Stack.create(scope, values, options); } + /** + * Delete the stack from its resource container. + * + * @param handle The handle to a stack. + * @return a new instance of StackClose + */ + public StackClose stackClose(Operand handle) { + return StackClose.create(scope, handle); + } + + /** + * A stack that produces elements in first-in last-out order. + * + * @param maxSize The maximum size of the stack if non-negative. If negative, the stack + * size is unlimited. + * @param elemType The type of the elements on the stack. + * @param options carries optional attribute values + * @param data type for {@code StackV2} output and operands + * @return a new instance of StackCreate + */ + public StackCreate stackCreate(Operand maxSize, Class elemType, + StackCreate.Options... options) { + return StackCreate.create(scope, maxSize, elemType, options); + } + + /** + * Pop the element at the top of the stack. + * + * @param data type for {@code elem} output + * @param handle The handle to a stack. + * @param elemType The type of the elem that is popped. + * @param data type for {@code StackPopV2} output and operands + * @return a new instance of StackPop + */ + public StackPop stackPop(Operand handle, + Class elemType) { + return StackPop.create(scope, handle, elemType); + } + + /** + * Push an element onto the stack. + * + * @param data type for {@code output} output + * @param handle The handle to a stack. + * @param elem The tensor to be pushed onto the stack. + * @param options carries optional attribute values + * @param data type for {@code StackPushV2} output and operands + * @return a new instance of StackPush + */ + public StackPush stackPush(Operand handle, Operand elem, + StackPush.Options... options) { + return StackPush.create(scope, handle, elem, options); + } + /** * Stage values similar to a lightweight Enqueue. * The basic functionality of this Op is similar to a queue with many @@ -6212,6 +6978,44 @@ public StatefulWhile statefulWhile(Iterable> input, ConcreteFunction return StatefulWhile.create(scope, input, cond, body, options); } + /** + * An n-way switch statement which calls a single branch function. + * 

+   *  An n-way switch statement, implementing the following:
+   *  ```
+   *  switch (branch_index) {
+   *    case 0:
+   *      output = branches[0](input);
+   *      break;
+   *    case 1:
+   *      output = branches[1](input);
+   *      break;
+   *    ...
+   *    case [[nbranches-1]]:
+   *    default:
+   *      output = branches[nbranches-1](input);
+   *      break;
+   *  }
+   *  ```
+   *
+   *  This should only be used when the none of branches has stateful ops.
+   *
+ * + * @param branchIndex The branch selector, an int32 Tensor. + * @param input A list of input tensors passed to the branch function. + * @param Tout A list of output types. + * @param branches 
+   *    A list of functions each of which takes 'inputs' and returns a list of
+   *    tensors, whose types are the same as what every other branch returns.
+   *
+ * @param options carries optional attribute values + * @return a new instance of StatelessCase + */ + public StatelessCase statelessCase(Operand branchIndex, Iterable> input, + List> Tout, List branches, Case.Options... options) { + return StatelessCase.create(scope, branchIndex, input, Tout, branches, options); + } + /** * output = cond ? then_branch(input) : else_branch(input) * @@ -6274,6 +7078,26 @@ public StatelessWhile statelessWhile(Iterable> input, ConcreteFunctio return StatelessWhile.create(scope, input, cond, body, options); } + /** + * Stochastically cast a given tensor from floats to ints. + * The values are cast with a deterministic pseudo-random tensor from a uniform distribution generated from user given key, counter, algorithm. Values will saturate if out of the specified integer type range, and will become zero if inputs are NaN. + *

The outputs are a deterministic function of {@code input}, {@code key}, {@code counter}, {@code alg}. + * + * @param data type for {@code output} output + * @param input The operand to stochastically cast to int. + * @param key Key for the counter-based RNG algorithm (shape uint64[1]). + * @param counter Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used. + * @param alg The RNG algorithm (shape int32[]). + * @param Tout The type of the output. + * @param data type for {@code StochasticCastToInt} output and operands + * @return a new instance of StochasticCastToInt + */ + public StochasticCastToInt stochasticCastToInt( + Operand input, Operand key, + Operand counter, Operand alg, Class Tout) { + return StochasticCastToInt.create(scope, input, key, counter, alg, Tout); + } + /** * Stops gradient computation. * When executed in a graph, this op outputs its input tensor as-is. @@ -6624,6 +7448,17 @@ public SwitchCond switchCond(Operand data, Operand UnbatchGrad unbatchGrad(Operand originalInput, return UnbatchGrad.create(scope, originalInput, batchIndex, grad, id, options); } + /** + * Perform clip by value on the quantized Tensor {@code operand}. + * Given quantized {@code operand} which was quantized using {@code scales} and {@code zero_points}, performs clip by value using {@code min} and {@code max} values. + * If quantization_axis is -1 (per-tensor quantized), the entire operand is clipped using scalar min, max. + * Otherwise (per-channel quantized), the clipping is also done per-channel. + * + * @param data type for {@code output} output + * @param operand Must be a Tensor of T. + * @param min The min value(s) to clip operand. Must be a Tensor of T. + * Must be a scalar Tensor if quantization_axis is -1 (per-tensor quantization), otherwise 1D Tensor of size (operand.dim_size(quantization_axis),) (per-axis quantization). + * @param max The min value(s) to clip operand. Must be a Tensor of T. + * Must be a scalar Tensor if quantization_axis is -1 (per-tensor quantization), otherwise 1D Tensor of size (operand.dim_size(quantization_axis),) (per-axis quantization). + * @param scales The float value(s) used as scale(s) when quantizing {@code operand}, {@code min} and {@code max}. + * Must be a scalar Tensor if quantization_axis is -1 (per-tensor quantization), otherwise 1D Tensor of size (operand.dim_size(quantization_axis),) (per-axis quantization). + * @param zeroPoints The int32 value(s) used as zero_point(s) when quantizing {@code operand}, {@code min} and {@code max}. + * Same shape condition as scales. + * @param quantizationMinVal The quantization min value that was used when operand was quantized. + * @param quantizationMaxVal The quantization max value that was used when operand was quantized. + * @param options carries optional attribute values + * @param data type for {@code UniformQuantizedClipByValue} output and operands + * @return a new instance of UniformQuantizedClipByValue + */ + public UniformQuantizedClipByValue uniformQuantizedClipByValue( + Operand operand, Operand min, Operand max, Operand scales, + Operand zeroPoints, Long quantizationMinVal, Long quantizationMaxVal, + UniformQuantizedClipByValue.Options... options) { + return UniformQuantizedClipByValue.create(scope, operand, min, max, scales, zeroPoints, quantizationMinVal, quantizationMaxVal, options); + } + /** * Finds unique elements along an axis of a tensor. * This operation either returns a tensor {@code y} containing unique elements @@ -8020,6 +8884,64 @@ public Unstage unstage(List> dtypes, Unstage.Options... o return Unstage.create(scope, dtypes, options); } + /** + * Applies upper_bound(sorted_search_values, values) along each row. + * Each set of rows with the same index in (sorted_inputs, values) is treated + * independently. The resulting row is the equivalent of calling + * {@code np.searchsorted(sorted_inputs, values, side='right')}. + *

The result is not a global index to the entire + * {@code Tensor}, but rather just the index in the last dimension. + *

A 2-D example: + * sorted_sequence = [[0, 3, 9, 9, 10], + * [1, 2, 3, 4, 5]] + * values = [[2, 4, 9], + * [0, 2, 6]] + *

result = UpperBound(sorted_sequence, values) + *

result == [[1, 2, 4], + * [0, 2, 5]] + * + * @param data type for {@code output} output + * @param sortedInputs 2-D Tensor where each row is ordered. + * @param values 2-D Tensor with the same numbers of rows as {@code sorted_search_values}. Contains + * the values that will be searched for in {@code sorted_search_values}. + * @param data type for {@code UpperBound} output and operands + * @return a new instance of UpperBound, with default output types + */ + public UpperBound upperBound(Operand sortedInputs, + Operand values) { + return UpperBound.create(scope, sortedInputs, values); + } + + /** + * Applies upper_bound(sorted_search_values, values) along each row. + * Each set of rows with the same index in (sorted_inputs, values) is treated + * independently. The resulting row is the equivalent of calling + * {@code np.searchsorted(sorted_inputs, values, side='right')}. + *

The result is not a global index to the entire + * {@code Tensor}, but rather just the index in the last dimension. + *

A 2-D example: + * sorted_sequence = [[0, 3, 9, 9, 10], + * [1, 2, 3, 4, 5]] + * values = [[2, 4, 9], + * [0, 2, 6]] + *

result = UpperBound(sorted_sequence, values) + *

result == [[1, 2, 4], + * [0, 2, 5]] + * + * @param data type for {@code output} output + * @param sortedInputs 2-D Tensor where each row is ordered. + * @param values 2-D Tensor with the same numbers of rows as {@code sorted_search_values}. Contains + * the values that will be searched for in {@code sorted_search_values}. + * @param outType The value of the outType attribute + * @param data type for {@code UpperBound} output and operands + * @param data type for {@code UpperBound} output and operands + * @return a new instance of UpperBound + */ + public UpperBound upperBound(Operand sortedInputs, + Operand values, Class outType) { + return UpperBound.create(scope, sortedInputs, values, outType); + } + /** * Creates a handle to a Variable resource. * diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/QuantizationOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/QuantizationOps.java index e0302dd0252..54527d187d0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/QuantizationOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/QuantizationOps.java @@ -32,8 +32,15 @@ import org.tensorflow.op.quantization.QuantizeAndDequantizeV4Grad; import org.tensorflow.op.quantization.QuantizeDownAndShrinkRange; import org.tensorflow.op.quantization.QuantizedConcat; +import org.tensorflow.op.quantization.QuantizedMatMulWithBiasAndDequantize; +import org.tensorflow.op.quantization.QuantizedMatMulWithBiasAndRequantize; import org.tensorflow.op.quantization.RequantizationRange; import org.tensorflow.op.quantization.Requantize; +import org.tensorflow.op.quantization.UniformDequantize; +import org.tensorflow.op.quantization.UniformQuantize; +import org.tensorflow.op.quantization.UniformQuantizedDot; +import org.tensorflow.op.quantization.UniformQuantizedDotHybrid; +import org.tensorflow.op.quantization.UniformRequantize; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -607,6 +614,60 @@ public QuantizedConcat quantizedConcat(Operand conc return QuantizedConcat.create(scope, concatDim, values, inputMins, inputMaxes); } + /** + * The QuantizedMatMulWithBiasAndDequantize operation + * + * @param data type for {@code out} output + * @param a The a value + * @param b The b value + * @param bias The bias value + * @param minA The minA value + * @param maxA The maxA value + * @param minB The minB value + * @param maxB The maxB value + * @param minFreezedOutput The minFreezedOutput value + * @param maxFreezedOutput The maxFreezedOutput value + * @param Toutput The value of the Toutput attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedMatMulWithBiasAndDequantize} output and operands + * @return a new instance of QuantizedMatMulWithBiasAndDequantize + */ + public QuantizedMatMulWithBiasAndDequantize quantizedMatMulWithBiasAndDequantize( + Operand a, Operand b, Operand bias, + Operand minA, Operand maxA, Operand minB, + Operand maxB, Operand minFreezedOutput, + Operand maxFreezedOutput, Class Toutput, + QuantizedMatMulWithBiasAndDequantize.Options... options) { + return QuantizedMatMulWithBiasAndDequantize.create(scope, a, b, bias, minA, maxA, minB, maxB, minFreezedOutput, maxFreezedOutput, Toutput, options); + } + + /** + * The QuantizedMatMulWithBiasAndRequantize operation + * + * @param data type for {@code out} output + * @param a The a value + * @param b The b value + * @param bias The bias value + * @param minA The minA value + * @param maxA The maxA value + * @param minB The minB value + * @param maxB The maxB value + * @param minFreezedOutput The minFreezedOutput value + * @param maxFreezedOutput The maxFreezedOutput value + * @param Toutput The value of the Toutput attribute + * @param options carries optional attribute values + * @param data type for {@code QuantizedMatMulWithBiasAndRequantize} output and operands + * @return a new instance of QuantizedMatMulWithBiasAndRequantize + */ + public QuantizedMatMulWithBiasAndRequantize quantizedMatMulWithBiasAndRequantize( + Operand a, Operand b, Operand bias, + Operand minA, Operand maxA, Operand minB, + Operand maxB, Operand minFreezedOutput, + Operand maxFreezedOutput, Class Toutput, + QuantizedMatMulWithBiasAndRequantize.Options... options) { + return QuantizedMatMulWithBiasAndRequantize.create(scope, a, b, bias, minA, maxA, minB, maxB, minFreezedOutput, maxFreezedOutput, Toutput, options); + } + /** * Computes a range that covers the actual values present in a quantized tensor. * Given a quantized tensor described by {@code (input, input_min, input_max)}, outputs a @@ -649,6 +710,203 @@ public Requantize requantize(Operand i return Requantize.create(scope, input, inputMin, inputMax, requestedOutputMin, requestedOutputMax, outType); } + /** + * Perform dequantization on the quantized Tensor {@code input}. + * Given quantized {@code input} which was quantized using {@code scales} and {@code zero_points}, performs dequantization using the formula: + * dequantized_data = (quantized_data - zero_point) * scale. + * + * @param data type for {@code output} output + * @param input Must be a Tensor of Tin. + * @param scales The float value(s) used as scale(s) when quantizing original data that input represents. + * Must be a scalar Tensor if quantization_axis is -1 (per-tensor quantization), otherwise 1D Tensor of size (input.dim_size(quantization_axis),) (per-axis quantization). + * @param zeroPoints The int32 value(s) used as zero_point(s) when quantizing original data that input represents. + * Same shape condition as scales. + * @param Tout The type of output Tensor. A tf.DType from: tf.qint8, tf.qint32 + * @param quantizationMinVal The quantization min value that was used when input was quantized. + * The purpose of this attribute is typically (but not limited to) to indicate narrow range, where this is set to: + * {@code (Tin lowest) + 1} if narrow range, and {@code (Tin lowest)} otherwise. + * For example, if Tin is qint8, this is set to -127 if narrow range quantized or -128 if not. + * @param quantizationMaxVal The quantization max value that was used when input was quantized. + * The purpose of this attribute is typically (but not limited to) indicate narrow range, where this is set to: + * {@code (Tout max)} for both narrow range and not narrow range. + * For example, if Tin is qint8, this is set to 127. + * @param options carries optional attribute values + * @param data type for {@code UniformDequantize} output and operands + * @return a new instance of UniformDequantize + */ + public UniformDequantize uniformDequantize( + Operand input, Operand scales, Operand zeroPoints, + Class Tout, Long quantizationMinVal, Long quantizationMaxVal, + UniformDequantize.Options... options) { + return UniformDequantize.create(scope, input, scales, zeroPoints, Tout, quantizationMinVal, quantizationMaxVal, options); + } + + /** + * Perform quantization on Tensor {@code input}. + * Given {@code input}, {@code scales} and {@code zero_points}, performs quantization using the formula: + * quantized_data = floor(input_data * (1.0f / scale) + 0.5f) + zero_point + * + * @param data type for {@code output} output + * @param input Must be a Tensor of Tin. + * @param scales The float value(s) to use as scale(s) to quantize {@code input}. + * Must be a scalar Tensor if quantization_axis is -1 (per-tensor quantization), otherwise 1D Tensor of size (input.dim_size(quantization_axis),) (per-axis quantization). + * @param zeroPoints The int32 value(s) to use as zero_point(s) to quantize {@code input}. + * Same shape condition as scales. + * @param Tout The type of output Tensor. A tf.DType from: tf.float32 + * @param quantizationMinVal The quantization min value to quantize {@code input}. + * The purpose of this attribute is typically (but not limited to) to indicate narrow range, where this is set to: + * {@code (Tin lowest) + 1} if narrow range, and {@code (Tin lowest)} otherwise. + * For example, if Tin is qint8, this is set to -127 if narrow range quantized or -128 if not. + * @param quantizationMaxVal The quantization max value to quantize {@code input}. + * The purpose of this attribute is typically (but not limited to) indicate narrow range, where this is set to: + * {@code (Tout max)} for both narrow range and not narrow range. + * For example, if Tin is qint8, this is set to 127. + * @param options carries optional attribute values + * @param data type for {@code UniformQuantize} output and operands + * @return a new instance of UniformQuantize + */ + public UniformQuantize uniformQuantize(Operand input, + Operand scales, Operand zeroPoints, Class Tout, Long quantizationMinVal, + Long quantizationMaxVal, UniformQuantize.Options... options) { + return UniformQuantize.create(scope, input, scales, zeroPoints, Tout, quantizationMinVal, quantizationMaxVal, options); + } + + /** + * Perform quantized dot of quantized Tensor {@code lhs} and quantized Tensor {@code rhs} to make quantized {@code output}. + * Given quantized {@code lhs} and quantized {@code rhs}, performs quantized dot on {@code lhs} and {@code rhs} to make quantized {@code output}. + * {@code lhs} and {@code rhs} must be 2D Tensors and the lhs.dim_size(1) must match rhs.dim_size(0). + * {@code lhs} and {@code rhs} must be quantized Tensor, where data value is quantized using the formula: + * quantized_data = clip(original_data / scale + zero_point, quantization_min_val, quantization_max_val). + * {@code output} is also quantized, using the same formula. + * If {@code rhs} is per-tensor quantized, {@code output} must be also per-tensor quantized. + * + * @param data type for {@code output} output + * @param lhs Must be a 2D Tensor of Tin. + * @param rhs Must be a 2D Tensor of Tin. + * @param lhsScales The float value(s) used as scale when quantizing original data that lhs represents. + * Must be a scalar Tensor (lhs supports only per-tensor quantization). + * @param lhsZeroPoints The int32 value(s) used as zero_point when quantizing original data that lhs represents. + * Same shape condition as lhs_scales. + * @param rhsScales The float value(s) used as scale when quantizing original data that rhs represents. + * Must be a scalar Tensor (per-tensor quantization) or 1D Tensor of size (rhs.dim_size(1),) (per-channel quantization). + * @param rhsZeroPoints The int32 value(s) used as zero_point when quantizing original data that rhs represents. + * Same shape condition as rhs_scales. + * @param outputScales The float value(s) to use as scales when quantizing original data that output represents. + * Must be a scalar Tensor (per-tensor quantization) or 1D Tensor of size (output.dim_size(1),) (per-channel quantization). + * If rhs is per-tensor quantized, output must be also per-tensor quantized. + * This means that if rhs_scales and rhs_zero_points are scalar Tensors, output_scales and output_zero_points must be scalar Tensors as well. + * @param outputZeroPoints The int32 value(s) used as zero_point when quantizing original data that output represents. + * Same shape condition as rhs_scales. + * @param Tout The type of output Tensor. + * @param lhsQuantizationMinVal The min value of the quantized data stored in lhs. + * For example, if Tin is qint8, this must be set to -127 if narrow range quantized or -128 if not. + * @param lhsQuantizationMaxVal The max value of the quantized data stored in rhs. + * For example, if Tin is qint8, this must be set to 127. + * @param rhsQuantizationMinVal The min value of the quantized data stored in rhs. + * For example, if Trhs is qint8, this must be set to -127 if narrow range quantized or -128 if not. + * @param rhsQuantizationMaxVal The max value of the quantized data stored in rhs. + * For example, if Trhs is qint8, this must be set to 127. + * @param outputQuantizationMinVal The min value of the quantized data stored in output. + * For example, if Tout is qint8, this must be set to -127 if narrow range quantized or -128 if not. + * @param outputQuantizationMaxVal The max value of the quantized data stored in output. + * For example, if Tout is qint8, this must be set to 127. + * @param options carries optional attribute values + * @param data type for {@code UniformQuantizedDot} output and operands + * @param data type for {@code UniformQuantizedDot} output and operands + * @return a new instance of UniformQuantizedDot + */ + public UniformQuantizedDot uniformQuantizedDot( + Operand lhs, Operand rhs, Operand lhsScales, Operand lhsZeroPoints, + Operand rhsScales, Operand rhsZeroPoints, Operand outputScales, + Operand outputZeroPoints, Class Tout, Long lhsQuantizationMinVal, + Long lhsQuantizationMaxVal, Long rhsQuantizationMinVal, Long rhsQuantizationMaxVal, + Long outputQuantizationMinVal, Long outputQuantizationMaxVal, + UniformQuantizedDot.Options... options) { + return UniformQuantizedDot.create(scope, lhs, rhs, lhsScales, lhsZeroPoints, rhsScales, rhsZeroPoints, outputScales, outputZeroPoints, Tout, lhsQuantizationMinVal, lhsQuantizationMaxVal, rhsQuantizationMinVal, rhsQuantizationMaxVal, outputQuantizationMinVal, outputQuantizationMaxVal, options); + } + + /** + * Perform hybrid quantized dot of float Tensor {@code lhs} and quantized Tensor {@code rhs}. + * Given float {@code lhs} and quantized {@code rhs}, internally performs quantization on {@code lhs}, and then performs quantized dot on quantized lhs and {@code rhs}. + * The internal quantization on {@code lhs} is a quantization to qint8, dynamic range, per-batch (per-axis along axis 0), asymmetric, and not narrow range (the range is [-128, 127]). + * {@code lhs} and {@code rhs} must be 2D Tensors and the lhs.dim_size(1) must match rhs.dim_size(0). + * {@code rhs} must be quantized Tensor, where its data value is quantized using the formula: + * quantized_data = clip(original_data / scale + zero_point, quantization_min_val, quantization_max_val). + * + * @param data type for {@code output} output + * @param lhs Must be a 2D Tensor of Tlhs. + * @param rhs Must be a 2D Tensor of Trhs. + * @param rhsScales The float value(s) used as scale when quantizing original data that rhs represents. + * Must be a scalar Tensor (per-tensor quantization) or 1D Tensor of size (rhs.dim_size(1),) (per-channel quantization). + * @param rhsZeroPoints The int32 value(s) used as zero_point when quantizing original data that rhs represents. + * Same shape condition as rhs_scales. + * @param Tout The type of output Tensor. + * @param rhsQuantizationMinVal The min value of the quantized data stored in rhs. + * For example, if Trhs is qint8, this must be set to -127 if narrow range quantized or -128 if not. + * @param rhsQuantizationMaxVal The max value of the quantized data stored in rhs. + * For example, if Trhs is qint8, this must be set to 127. + * @param options carries optional attribute values + * @param data type for {@code UniformQuantizedDotHybrid} output and operands + * @return a new instance of UniformQuantizedDotHybrid + */ + public UniformQuantizedDotHybrid uniformQuantizedDotHybrid( + Operand lhs, Operand rhs, Operand rhsScales, + Operand rhsZeroPoints, Class Tout, Long rhsQuantizationMinVal, + Long rhsQuantizationMaxVal, UniformQuantizedDotHybrid.Options... options) { + return UniformQuantizedDotHybrid.create(scope, lhs, rhs, rhsScales, rhsZeroPoints, Tout, rhsQuantizationMinVal, rhsQuantizationMaxVal, options); + } + + /** + * Given quantized tensor {@code input}, requantize it with new quantization parameters. + * Given quantized tensor {@code input}, which was quantized using {input_scales, input_zero_points, input_quantization_axis, input_quantization_min_val, input_quantization_max_val}, + * requantize it to a tensor, which is quantized using {output_scales, output_zero_points, output_quantization_axis, output_quantization_min_val, output_quantization_max_val}. + * The requantization is done by using the formula: + * output_quantized_data = clip( + * (input_quantized_data - input_zero_point) * (input_scale / output_scale) + output_zero_point, + * output_quantization_min_val, + * output_quantization_max_val) + *

Per-tensor and per-axis quantization supported cases are followings: + *

    + *
  • per-tensor -> per-tensor
    • + *
  • per-tensor -> per-axis
    • + *
  • per-axis -> per-axis where input_quantization_axis equals output_quantization_axis. + * i.e. At least one among input_quantization_axis and output_quantization_axis must be -1, or two must be equal.
    • + *
+ * + * @param data type for {@code output} output + * @param input Must be a Tensor of Tin. + * @param inputScales The float value(s) used as scale(s) when quantizing original data that {@code input} represents. + * Must be a scalar Tensor if quantization_axis is -1 (per-tensor quantization), otherwise 1D Tensor of size (input.dim_size(quantization_axis),) (per-axis quantization). + * @param inputZeroPoints The int32 value(s) used as zero_point(s) when quantizing original data that {@code input} represents. + * Same shape condition as scales. + * @param outputScales The float value(s) to use as new scale(s) to quantize original data that {@code input} represents. + * Must be a scalar Tensor if quantization_axis is -1 (per-tensor quantization), otherwise 1D Tensor of size (input.dim_size(quantization_axis),) (per-axis quantization). + * @param outputZeroPoints The int32 value(s) to use as new zero_point(s) to quantize original data that {@code input} represents. + * Same shape condition as scales. + * @param Tout The type of output Tensor. A tf.DType from: tf.qint8, tf.qint32 + * @param inputQuantizationMinVal The quantization min value that was used when quantizing original data that {@code input} represents. + * The purpose of this attribute is typically (but not limited to) to indicate narrow range, where this is set to: + * {@code (Tin lowest) + 1} if narrow range, and {@code (Tin lowest)} otherwise. + * For example, if Tin is qint8, this is set to -127 if narrow range quantized or -128 if not. + * @param inputQuantizationMaxVal The quantization max value that was used when quantizing original data that {@code input} represents. + * The purpose of this attribute is typically (but not limited to) indicate narrow range, where this is set to: + * {@code (Tout max)} for both narrow range and not narrow range. + * For example, if Tin is qint8, this is set to 127. + * @param outputQuantizationMinVal The new quantization min value to quantize original data that {@code input} represents. + * @param outputQuantizationMaxVal The new quantization max value to quantize original data that {@code input} represents. + * @param options carries optional attribute values + * @param data type for {@code UniformRequantize} output and operands + * @return a new instance of UniformRequantize + */ + public UniformRequantize uniformRequantize( + Operand input, Operand inputScales, + Operand inputZeroPoints, Operand outputScales, + Operand outputZeroPoints, Class Tout, Long inputQuantizationMinVal, + Long inputQuantizationMaxVal, Long outputQuantizationMinVal, Long outputQuantizationMaxVal, + UniformRequantize.Options... options) { + return UniformRequantize.create(scope, input, inputScales, inputZeroPoints, outputScales, outputZeroPoints, Tout, inputQuantizationMinVal, inputQuantizationMaxVal, outputQuantizationMinVal, outputQuantizationMaxVal, options); + } + /** * Get the parent {@link Ops} object. */ diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/RaggedOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/RaggedOps.java index a1cd7bbdc4b..ddc1e0c4c0e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/RaggedOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/RaggedOps.java @@ -17,10 +17,21 @@ // package org.tensorflow.op; +import java.util.List; import org.tensorflow.Operand; import org.tensorflow.op.ragged.RaggedBincount; +import org.tensorflow.op.ragged.RaggedCountSparseOutput; +import org.tensorflow.op.ragged.RaggedCross; import org.tensorflow.op.ragged.RaggedFillEmptyRows; import org.tensorflow.op.ragged.RaggedFillEmptyRowsGrad; +import org.tensorflow.op.ragged.RaggedGather; +import org.tensorflow.op.ragged.RaggedRange; +import org.tensorflow.op.ragged.RaggedTensorFromVariant; +import org.tensorflow.op.ragged.RaggedTensorToSparse; +import org.tensorflow.op.ragged.RaggedTensorToTensor; +import org.tensorflow.op.ragged.RaggedTensorToVariant; +import org.tensorflow.op.ragged.RaggedTensorToVariantGradient; +import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -67,6 +78,60 @@ public RaggedBincount raggedBincount( return RaggedBincount.create(scope, splits, values, sizeOutput, weights, options); } + /** + * Performs sparse-output bin counting for a ragged tensor input. + * Counts the number of times each value occurs in the input. + * + * @param data type for {@code output_values} output + * @param splits Tensor containing the row splits of the ragged tensor to count. + * @param values Tensor containing values of the sparse tensor to count. + * @param weights A Tensor of the same shape as indices containing per-index weight values. + * May also be the empty tensor if no weights are used. + * @param binaryOutput Whether to output the number of occurrences of each value or 1. + * @param options carries optional attribute values + * @param data type for {@code RaggedCountSparseOutput} output and operands + * @return a new instance of RaggedCountSparseOutput + */ + public RaggedCountSparseOutput raggedCountSparseOutput( + Operand splits, Operand values, Operand weights, + Boolean binaryOutput, RaggedCountSparseOutput.Options... options) { + return RaggedCountSparseOutput.create(scope, splits, values, weights, binaryOutput, options); + } + + /** + * Generates a feature cross from a list of tensors, and returns it as a + * RaggedTensor. See {@code tf.ragged.cross} for more details. + * + * @param data type for {@code output_values} output + * @param data type for {@code output_row_splits} output + * @param raggedValues The values tensor for each RaggedTensor input. + * @param raggedRowSplits The row_splits tensor for each RaggedTensor input. + * @param sparseIndices The indices tensor for each SparseTensor input. + * @param sparseValues The values tensor for each SparseTensor input. + * @param sparseShape The dense_shape tensor for each SparseTensor input. + * @param denseInputs The tf.Tensor inputs. + * @param inputOrder String specifying the tensor type for each input. The {@code i}th character in + * this string specifies the type of the {@code i}th input, and is one of: 'R' (ragged), + * 'D' (dense), or 'S' (sparse). This attr is used to ensure that the crossed + * values are combined in the order of the inputs from the call to tf.ragged.cross. + * @param hashedOutput The value of the hashedOutput attribute + * @param numBuckets The value of the numBuckets attribute + * @param hashKey The value of the hashKey attribute + * @param outValuesType The value of the outValuesType attribute + * @param outRowSplitsType The value of the outRowSplitsType attribute + * @param data type for {@code RaggedCross} output and operands + * @param data type for {@code RaggedCross} output and operands + * @return a new instance of RaggedCross + */ + public RaggedCross raggedCross( + Iterable> raggedValues, Iterable> raggedRowSplits, + Iterable> sparseIndices, Iterable> sparseValues, + Iterable> sparseShape, Iterable> denseInputs, String inputOrder, + Boolean hashedOutput, Long numBuckets, Long hashKey, Class outValuesType, + Class outRowSplitsType) { + return RaggedCross.create(scope, raggedValues, raggedRowSplits, sparseIndices, sparseValues, sparseShape, denseInputs, inputOrder, hashedOutput, numBuckets, hashKey, outValuesType, outRowSplitsType); + } + /** * The RaggedFillEmptyRows operation * @@ -97,6 +162,297 @@ public RaggedFillEmptyRowsGrad raggedFillEmptyRowsGrad( return RaggedFillEmptyRowsGrad.create(scope, reverseIndexMap, gradValues); } + /** + * Gather ragged slices from {@code params} axis {@code 0} according to {@code indices}. + * Outputs a {@code RaggedTensor} output composed from {@code output_dense_values} and + * {@code output_nested_splits}, such that: + * 
+   *  output.shape = indices.shape + params.shape[1:]
+   *  output.ragged_rank = indices.shape.ndims + params.ragged_rank
+   *  output[i...j, d0...dn] = params[indices[i...j], d0...dn]
+   *
+ *

where + *

    + *
  • {@code params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits)} + * provides the values that should be gathered.
    • + *
  • {@code indices} ia a dense tensor with dtype {@code int32} or {@code int64}, indicating which + * values should be gathered.
    • + *
  • {@code output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits)} + * is the output tensor.
    • + *
+ *

(Note: This c++ op is used to implement the higher-level python + * {@code tf.ragged.gather} op, which also supports ragged indices.) + * + * @param data type for {@code output_nested_splits} output + * @param data type for {@code output_dense_values} output + * @param paramsNestedSplits The {@code nested_row_splits} tensors that define the row-partitioning for the + * {@code params} RaggedTensor input. + * @param paramsDenseValues The {@code flat_values} for the {@code params} RaggedTensor. There was a terminology change + * at the python level from dense_values to flat_values, so dense_values is the + * deprecated name. + * @param indices Indices in the outermost dimension of {@code params} of the values that should be + * gathered. + * @param OUTPUTRAGGEDRANK The ragged rank of the output RaggedTensor. {@code output_nested_splits} will contain + * this number of {@code row_splits} tensors. This value should equal + * {@code indices.shape.ndims + params.ragged_rank - 1}. + * @param data type for {@code RaggedGather} output and operands + * @param data type for {@code RaggedGather} output and operands + * @return a new instance of RaggedGather + */ + public RaggedGather raggedGather( + Iterable> paramsNestedSplits, Operand paramsDenseValues, + Operand indices, Long OUTPUTRAGGEDRANK) { + return RaggedGather.create(scope, paramsNestedSplits, paramsDenseValues, indices, OUTPUTRAGGEDRANK); + } + + /** + * Returns a {@code RaggedTensor} containing the specified sequences of numbers. + * Returns a {@code RaggedTensor} {@code result} composed from {@code rt_dense_values} and + * {@code rt_nested_splits}, such that + * {@code result[i] = range(starts[i], limits[i], deltas[i])}. + * 

+   *  (rt_nested_splits, rt_dense_values) = ragged_range(
+   *        starts=[2, 5, 8], limits=[3, 5, 12], deltas=1)
+   *  result = tf.ragged.from_row_splits(rt_dense_values, rt_nested_splits)
+   *  print(result)
+   *  <tf.RaggedTensor [[2], [], [8, 9, 10, 11]] >
+   *
+ *

The input tensors {@code starts}, {@code limits}, and {@code deltas} may be scalars or vectors. + * The vector inputs must all have the same size. Scalar inputs are broadcast + * to match the size of the vector inputs. + * + * @param data type for {@code rt_nested_splits} output + * @param data type for {@code rt_dense_values} output + * @param starts The starts of each range. + * @param limits The limits of each range. + * @param deltas The deltas of each range. + * @param data type for {@code RaggedRange} output and operands + * @return a new instance of RaggedRange, with default output types + */ + public RaggedRange raggedRange(Operand starts, + Operand limits, Operand deltas) { + return RaggedRange.create(scope, starts, limits, deltas); + } + + /** + * Returns a {@code RaggedTensor} containing the specified sequences of numbers. + * Returns a {@code RaggedTensor} {@code result} composed from {@code rt_dense_values} and + * {@code rt_nested_splits}, such that + * {@code result[i] = range(starts[i], limits[i], deltas[i])}. + * 

+   *  (rt_nested_splits, rt_dense_values) = ragged_range(
+   *        starts=[2, 5, 8], limits=[3, 5, 12], deltas=1)
+   *  result = tf.ragged.from_row_splits(rt_dense_values, rt_nested_splits)
+   *  print(result)
+   *  <tf.RaggedTensor [[2], [], [8, 9, 10, 11]] >
+   *
+ *

The input tensors {@code starts}, {@code limits}, and {@code deltas} may be scalars or vectors. + * The vector inputs must all have the same size. Scalar inputs are broadcast + * to match the size of the vector inputs. + * + * @param data type for {@code rt_nested_splits} output + * @param data type for {@code rt_dense_values} output + * @param starts The starts of each range. + * @param limits The limits of each range. + * @param deltas The deltas of each range. + * @param Tsplits The value of the Tsplits attribute + * @param data type for {@code RaggedRange} output and operands + * @param data type for {@code RaggedRange} output and operands + * @return a new instance of RaggedRange + */ + public RaggedRange raggedRange(Operand starts, + Operand limits, Operand deltas, Class Tsplits) { + return RaggedRange.create(scope, starts, limits, deltas, Tsplits); + } + + /** + * Decodes a {@code variant} Tensor into a {@code RaggedTensor}. + * Decodes the given {@code variant} Tensor and returns a {@code RaggedTensor}. The input + * could be a scalar, meaning it encodes a single {@code RaggedTensor} with ragged_rank + * {@code output_ragged_rank}. It could also have an arbitrary rank, in which case each + * element is decoded into a {@code RaggedTensor} with ragged_rank {@code input_ragged_rank} + * and these are then stacked according to the input shape to output a single + * {@code RaggedTensor} with ragged_rank {@code output_ragged_rank}. Each {@code variant} element in + * the input Tensor is decoded by retrieving from the element a 1-D {@code variant} + * Tensor with {@code input_ragged_rank + 1} Tensors, corresponding to the splits and + * values of the decoded {@code RaggedTensor}. If {@code input_ragged_rank} is -1, then it is + * inferred as {@code output_ragged_rank} - {@code rank(encoded_ragged)}. See + * {@code RaggedTensorToVariant} for the corresponding encoding logic. + * + * @param data type for {@code output_nested_splits} output + * @param data type for {@code output_dense_values} output + * @param encodedRagged A {@code variant} Tensor containing encoded {@code RaggedTensor}s. + * @param inputRaggedRank The ragged rank of each encoded {@code RaggedTensor} component in the input. If set to + * -1, this is inferred as {@code output_ragged_rank} - {@code rank(encoded_ragged)} + * @param outputRaggedRank The expected ragged rank of the output {@code RaggedTensor}. The following must hold: + * {@code output_ragged_rank = rank(encoded_ragged) + input_ragged_rank}. + * @param Tvalues The value of the Tvalues attribute + * @param data type for {@code RaggedTensorFromVariant} output and operands + * @return a new instance of RaggedTensorFromVariant, with default output types + */ + public RaggedTensorFromVariant raggedTensorFromVariant( + Operand encodedRagged, Long inputRaggedRank, Long outputRaggedRank, + Class Tvalues) { + return RaggedTensorFromVariant.create(scope, encodedRagged, inputRaggedRank, outputRaggedRank, Tvalues); + } + + /** + * Decodes a {@code variant} Tensor into a {@code RaggedTensor}. + * Decodes the given {@code variant} Tensor and returns a {@code RaggedTensor}. The input + * could be a scalar, meaning it encodes a single {@code RaggedTensor} with ragged_rank + * {@code output_ragged_rank}. It could also have an arbitrary rank, in which case each + * element is decoded into a {@code RaggedTensor} with ragged_rank {@code input_ragged_rank} + * and these are then stacked according to the input shape to output a single + * {@code RaggedTensor} with ragged_rank {@code output_ragged_rank}. Each {@code variant} element in + * the input Tensor is decoded by retrieving from the element a 1-D {@code variant} + * Tensor with {@code input_ragged_rank + 1} Tensors, corresponding to the splits and + * values of the decoded {@code RaggedTensor}. If {@code input_ragged_rank} is -1, then it is + * inferred as {@code output_ragged_rank} - {@code rank(encoded_ragged)}. See + * {@code RaggedTensorToVariant} for the corresponding encoding logic. + * + * @param data type for {@code output_nested_splits} output + * @param data type for {@code output_dense_values} output + * @param encodedRagged A {@code variant} Tensor containing encoded {@code RaggedTensor}s. + * @param inputRaggedRank The ragged rank of each encoded {@code RaggedTensor} component in the input. If set to + * -1, this is inferred as {@code output_ragged_rank} - {@code rank(encoded_ragged)} + * @param outputRaggedRank The expected ragged rank of the output {@code RaggedTensor}. The following must hold: + * {@code output_ragged_rank = rank(encoded_ragged) + input_ragged_rank}. + * @param Tvalues The value of the Tvalues attribute + * @param Tsplits The value of the Tsplits attribute + * @param data type for {@code RaggedTensorFromVariant} output and operands + * @param data type for {@code RaggedTensorFromVariant} output and operands + * @return a new instance of RaggedTensorFromVariant + */ + public RaggedTensorFromVariant raggedTensorFromVariant( + Operand encodedRagged, Long inputRaggedRank, Long outputRaggedRank, + Class Tvalues, Class Tsplits) { + return RaggedTensorFromVariant.create(scope, encodedRagged, inputRaggedRank, outputRaggedRank, Tvalues, Tsplits); + } + + /** + * Converts a {@code RaggedTensor} into a {@code SparseTensor} with the same values. + * input=ragged.from_nested_row_splits(rt_dense_values, rt_nested_splits) + * output=SparseTensor(indices=sparse_indices, values=sparse_values, + * dense_shape=sparse_dense_shape) + * + * @param data type for {@code sparse_values} output + * @param rtNestedSplits The {@code row_splits} for the {@code RaggedTensor}. + * @param rtDenseValues The {@code flat_values} for the {@code RaggedTensor}. + * @param data type for {@code RaggedTensorToSparse} output and operands + * @return a new instance of RaggedTensorToSparse + */ + public RaggedTensorToSparse raggedTensorToSparse( + Iterable> rtNestedSplits, Operand rtDenseValues) { + return RaggedTensorToSparse.create(scope, rtNestedSplits, rtDenseValues); + } + + /** + * Create a dense tensor from a ragged tensor, possibly altering its shape. + * The {@code ragged_to_dense} op creates a dense tensor from a list of row partition + * tensors, a value vector, and default values. If the shape is unspecified, the + * minimal shape required to contain all the elements in the ragged tensor (the + * natural shape) will be used. If some dimensions are left unspecified, then the + * size of the natural shape is used in that dimension. + *

The default_value will be broadcast to the output shape. After that, the values + * from the ragged tensor overwrite the default values. Note that the default_value + * must have less dimensions than the value. + *

The row partition tensors are in the order of the dimensions. + * At present, the types can be: + *

    + *
  • "ROW_SPLITS": the row_splits tensor from the ragged tensor.
    • + *
  • "VALUE_ROWIDS": the value_rowids tensor from the ragged tensor.
    • + *
  • "FIRST_DIM_SIZE": if value_rowids is used for the first dimension, then it + * is preceded by "FIRST_DIM_SIZE".
    • + *
+ * + * @param data type for {@code result} output + * @param shape The desired shape of the output tensor. If left unspecified (empty), + * the minimal shape required to contain all the elements in the ragged tensor + * (the natural shape) will be used. If some dimensions are left unspecified, then + * the size of the natural shape is used in that dimension. + *

Note that dense dimensions cannot be modified by the shape argument. Trying to + * change the size of a dense dimension will cause the op to fail. + * Examples: + * natural shape: [4, 5, 6] + * shape: -1 + * output shape: [4, 5, 6] + *

natural shape: [4, 5, 6] + * shape: [3, -1, 2] + * output shape: [3, 5, 2] + *

natural shape: [4, 5, 6] + * shape: [3, 7, 2] + * output shape: [3, 7, 2] + * @param values A 1D tensor representing the values of the ragged tensor. + * @param defaultValue The default_value when the shape is larger than the ragged tensor. The + * default_value is broadcast until it is the shape of the output tensor, and + * then overwritten by values in the ragged tensor. The default value must be + * compatible with this broadcast operation, and must have fewer dimensions than + * the value tensor. + * @param rowPartitionTensors The rowPartitionTensors value + * @param rowPartitionTypes The types of the row partition tensors. At present, these can be: + *

    + *
  • "ROW_SPLITS": the row_splits tensor from the ragged tensor.
    • + *
  • "VALUE_ROWIDS": the value_rowids tensor from the ragged tensor.
    • + *
  • "FIRST_DIM_SIZE": if value_rowids is used for the first dimension, then it + * is preceeded by "FIRST_DIM_SIZE". + * The tensors are in the order of the dimensions.
    • + *
+ * @param data type for {@code RaggedTensorToTensor} output and operands + * @return a new instance of RaggedTensorToTensor + */ + public RaggedTensorToTensor raggedTensorToTensor( + Operand shape, Operand values, Operand defaultValue, + Iterable> rowPartitionTensors, List rowPartitionTypes) { + return RaggedTensorToTensor.create(scope, shape, values, defaultValue, rowPartitionTensors, rowPartitionTypes); + } + + /** + * Encodes a {@code RaggedTensor} into a {@code variant} Tensor. + * Encodes the given {@code RaggedTensor} and returns a {@code variant} Tensor. If + * {@code batched_input} is True, then input {@code RaggedTensor} is unbatched along the + * zero-th dimension, each component {@code RaggedTensor} is encoded into a scalar + * {@code variant} Tensor, and these are stacked to return a 1-D {@code variant} Tensor. + * If {@code batched_input} is False, then the input {@code RaggedTensor} is encoded as is and + * a scalar {@code variant} Tensor is returned. A {@code RaggedTensor} is encoded by first + * creating a 1-D {@code variant} Tensor with {@code ragged_rank + 1} elements, containing the + * splits and values Tensors of the {@code RaggedTensor}. Then the 1-D {@code variant} Tensor + * is wrapped in a scalar {@code variant} Tensor. See {@code RaggedTensorFromVariant} for the + * corresponding decoding logic. + * + * @param rtNestedSplits A list of one or more Tensors representing the splits of the input + * {@code RaggedTensor}. + * @param rtDenseValues A Tensor representing the values of the input {@code RaggedTensor}. + * @param batchedInput A {@code bool} denoting whether the input is a batched {@code RaggedTensor}. + * @return a new instance of RaggedTensorToVariant + */ + public RaggedTensorToVariant raggedTensorToVariant( + Iterable> rtNestedSplits, Operand rtDenseValues, + Boolean batchedInput) { + return RaggedTensorToVariant.create(scope, rtNestedSplits, rtDenseValues, batchedInput); + } + + /** + * Helper used to compute the gradient for {@code RaggedTensorToVariant}. + * Computes the gradient for the dense_values input to the RaggedTensorToVariant + * op, given the variant-encoded ragged gradients of the outputs, along with + * the outer row-splits and the shape of the dense-values that were provided as + * inputs to the RaggedTensorToVariant op. + * + * @param data type for {@code dense_values_grad} output + * @param encodedRaggedGrad A {@code variant} Tensor containing encoded {@code RaggedTensor} gradients. + * @param rowSplits Outermost row-splits that were used as input to the RaggedTensorToVariant op. + * @param denseValuesShape Shape of the dense_values that was used as an input to the + * RaggedTensorToVariant op. + * @param Tvalues The value of the Tvalues attribute + * @param data type for {@code RaggedTensorToVariantGradient} output and operands + * @return a new instance of RaggedTensorToVariantGradient + */ + public RaggedTensorToVariantGradient raggedTensorToVariantGradient( + Operand encodedRaggedGrad, Operand rowSplits, + Operand denseValuesShape, Class Tvalues) { + return RaggedTensorToVariantGradient.create(scope, encodedRaggedGrad, rowSplits, denseValuesShape, Tvalues); + } + /** * Get the parent {@link Ops} object. */ diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/RandomExperimentalOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/RandomExperimentalOps.java new file mode 100644 index 00000000000..b1d86643cce --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/RandomExperimentalOps.java @@ -0,0 +1,71 @@ +// Copyright 2020-2022 The TensorFlow Authors. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// ============================================================================== +// +// This class has been generated, DO NOT EDIT! +// +package org.tensorflow.op; + +import org.tensorflow.Operand; +import org.tensorflow.op.random.experimental.StatelessShuffle; +import org.tensorflow.types.TInt32; +import org.tensorflow.types.family.TType; + +/** + * An API for building {@code random.experimental} operations as {@link Op Op}s + * + * @see {@link Ops} + */ +public final class RandomExperimentalOps { + private final Scope scope; + + private final Ops ops; + + RandomExperimentalOps(Ops ops) { + this.scope = ops.scope(); + this.ops = ops; + } + + /** + * Randomly and deterministically shuffles a tensor along its first dimension. + * The tensor is shuffled along dimension 0, such that each {@code value[j]} is mapped + * to one and only one {@code output[i]}. For example, a mapping that might occur for a + * 3x2 tensor is: + * 
+   *  [[1, 2],       [[5, 6],
+   *   [3, 4],  ==>   [1, 2],
+   *   [5, 6]]        [3, 4]]
+   *
+ *

The outputs are a deterministic function of {@code value}, {@code key}, {@code counter} and {@code alg}. + * + * @param data type for {@code output} output + * @param value The tensor to be shuffled. + * @param key Key for the counter-based RNG algorithm (shape uint64[1]). + * @param counter Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used. + * @param alg The RNG algorithm (shape int32[]). + * @param data type for {@code StatelessShuffle} output and operands + * @return a new instance of StatelessShuffle + */ + public StatelessShuffle statelessShuffle(Operand value, + Operand key, Operand counter, Operand alg) { + return StatelessShuffle.create(scope, value, key, counter, alg); + } + + /** + * Get the parent {@link Ops} object. + */ + public final Ops ops() { + return ops; + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/RandomOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/RandomOps.java index 9f2e4f7f288..8abc032de91 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/RandomOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/RandomOps.java @@ -19,24 +19,53 @@ import org.tensorflow.Operand; import org.tensorflow.op.random.AllCandidateSampler; +import org.tensorflow.op.random.AnonymousRandomSeedGenerator; +import org.tensorflow.op.random.AnonymousSeedGenerator; +import org.tensorflow.op.random.DeleteRandomSeedGenerator; +import org.tensorflow.op.random.DeleteSeedGenerator; +import org.tensorflow.op.random.DummySeedGenerator; import org.tensorflow.op.random.LogUniformCandidateSampler; import org.tensorflow.op.random.Multinomial; +import org.tensorflow.op.random.NonDeterministicInts; import org.tensorflow.op.random.ParameterizedTruncatedNormal; import org.tensorflow.op.random.RandomGamma; +import org.tensorflow.op.random.RandomGammaGrad; import org.tensorflow.op.random.RandomPoisson; import org.tensorflow.op.random.RandomShuffle; import org.tensorflow.op.random.RandomStandardNormal; import org.tensorflow.op.random.RandomUniform; import org.tensorflow.op.random.RandomUniformInt; import org.tensorflow.op.random.RecordInput; +import org.tensorflow.op.random.RngReadAndSkip; +import org.tensorflow.op.random.RngSkip; import org.tensorflow.op.random.StatefulRandomBinomial; import org.tensorflow.op.random.StatefulStandardNormal; +import org.tensorflow.op.random.StatefulTruncatedNormal; +import org.tensorflow.op.random.StatefulUniform; +import org.tensorflow.op.random.StatefulUniformFullInt; +import org.tensorflow.op.random.StatefulUniformInt; import org.tensorflow.op.random.StatelessMultinomial; +import org.tensorflow.op.random.StatelessParameterizedTruncatedNormal; +import org.tensorflow.op.random.StatelessRandomBinomial; +import org.tensorflow.op.random.StatelessRandomGamma; +import org.tensorflow.op.random.StatelessRandomGetAlg; +import org.tensorflow.op.random.StatelessRandomGetKeyCounter; +import org.tensorflow.op.random.StatelessRandomGetKeyCounterAlg; import org.tensorflow.op.random.StatelessRandomNormal; +import org.tensorflow.op.random.StatelessRandomNormalV2; +import org.tensorflow.op.random.StatelessRandomPoisson; import org.tensorflow.op.random.StatelessRandomUniform; +import org.tensorflow.op.random.StatelessRandomUniformFullInt; +import org.tensorflow.op.random.StatelessRandomUniformFullIntV2; +import org.tensorflow.op.random.StatelessRandomUniformInt; +import org.tensorflow.op.random.StatelessRandomUniformIntV2; +import org.tensorflow.op.random.StatelessRandomUniformV2; import org.tensorflow.op.random.StatelessTruncatedNormal; +import org.tensorflow.op.random.StatelessTruncatedNormalV2; +import org.tensorflow.op.random.ThreadUnsafeUnigramCandidateSampler; import org.tensorflow.op.random.TruncatedNormal; import org.tensorflow.op.random.UniformCandidateSampler; +import org.tensorflow.types.TBool; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -49,6 +78,8 @@ * @see {@link Ops} */ public final class RandomOps { + public final RandomExperimentalOps experimental; + private final Scope scope; private final Ops ops; @@ -56,6 +87,7 @@ public final class RandomOps { RandomOps(Ops ops) { this.scope = ops.scope(); this.ops = ops; + experimental = new RandomExperimentalOps(ops); } /** @@ -83,6 +115,64 @@ public AllCandidateSampler allCandidateSampler(Operand trueClasses, Long return AllCandidateSampler.create(scope, trueClasses, numTrue, numSampled, unique, options); } + /** + * The AnonymousRandomSeedGenerator operation + * + * @param seed The seed value + * @param seed2 The seed2 value + * @return a new instance of AnonymousRandomSeedGenerator + */ + public AnonymousRandomSeedGenerator anonymousRandomSeedGenerator(Operand seed, + Operand seed2) { + return AnonymousRandomSeedGenerator.create(scope, seed, seed2); + } + + /** + * The AnonymousSeedGenerator operation + * + * @param seed The seed value + * @param seed2 The seed2 value + * @param reshuffle The reshuffle value + * @return a new instance of AnonymousSeedGenerator + */ + public AnonymousSeedGenerator anonymousSeedGenerator(Operand seed, Operand seed2, + Operand reshuffle) { + return AnonymousSeedGenerator.create(scope, seed, seed2, reshuffle); + } + + /** + * The DeleteRandomSeedGenerator operation + * + * @param handle The handle value + * @param deleter The deleter value + * @return a new instance of DeleteRandomSeedGenerator + */ + public DeleteRandomSeedGenerator deleteRandomSeedGenerator(Operand handle, + Operand deleter) { + return DeleteRandomSeedGenerator.create(scope, handle, deleter); + } + + /** + * The DeleteSeedGenerator operation + * + * @param handle The handle value + * @param deleter The deleter value + * @return a new instance of DeleteSeedGenerator + */ + public DeleteSeedGenerator deleteSeedGenerator(Operand handle, + Operand deleter) { + return DeleteSeedGenerator.create(scope, handle, deleter); + } + + /** + * The DummySeedGenerator operation + * + * @return a new instance of DummySeedGenerator + */ + public DummySeedGenerator dummySeedGenerator() { + return DummySeedGenerator.create(scope); + } + /** * Generates labels for candidate sampling with a log-uniform distribution. * See explanations of candidate sampling and the data formats at @@ -142,6 +232,33 @@ public Multinomial multinomial(Operand return Multinomial.create(scope, logits, numSamples, outputDtype, options); } + /** + * Non-deterministically generates some integers. + * This op may use some OS-provided source of non-determinism (e.g. an RNG), so each execution will give different results. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @return a new instance of NonDeterministicInts, with default output types + */ + public NonDeterministicInts nonDeterministicInts(Operand shape) { + return NonDeterministicInts.create(scope, shape); + } + + /** + * Non-deterministically generates some integers. + * This op may use some OS-provided source of non-determinism (e.g. an RNG), so each execution will give different results. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param dtype The type of the output. + * @param data type for {@code NonDeterministicInts} output and operands + * @return a new instance of NonDeterministicInts + */ + public NonDeterministicInts nonDeterministicInts( + Operand shape, Class dtype) { + return NonDeterministicInts.create(scope, shape, dtype); + } + /** * Outputs random values from a normal distribution. The parameters may each be a * scalar which applies to the entire output, or a vector of length shape[0] which @@ -184,6 +301,20 @@ public RandomGamma randomGamma(Operand return RandomGamma.create(scope, shape, alpha, options); } + /** + * Computes the derivative of a Gamma random sample w.r.t. {@code alpha}. + * + * @param data type for {@code output} output + * @param alpha The alpha value + * @param sample The sample value + * @param data type for {@code RandomGammaGrad} output and operands + * @return a new instance of RandomGammaGrad + */ + public RandomGammaGrad randomGammaGrad(Operand alpha, + Operand sample) { + return RandomGammaGrad.create(scope, alpha, sample); + } + /** * Outputs random values from the Poisson distribution(s) described by rate. * This op uses two algorithms, depending on rate. If rate >= 10, then @@ -322,6 +453,41 @@ public RecordInput recordInput(String filePattern, RecordInput.Options... option return RecordInput.create(scope, filePattern, options); } + /** + * Advance the counter of a counter-based RNG. + * The state of the RNG after + * {@code rng_read_and_skip(n)} will be the same as that after {@code uniform([n])} + * (or any other distribution). The actual increment added to the + * counter is an unspecified implementation choice. + *

In the case that the input algorithm is RNG_ALG_AUTO_SELECT, the counter in the state needs to be of size int64[2], the current maximal counter size among algorithms. In this case, this op will manage the counter as if it is an 128-bit integer with layout [lower_64bits, higher_64bits]. If an algorithm needs less than 128 bits for the counter, it should use the left portion of the int64[2]. In this way, the int64[2] is compatible with all current RNG algorithms (Philox, ThreeFry and xla::RandomAlgorithm::RNG_DEFAULT). Downstream RNG ops can thus use this counter with any RNG algorithm. + * + * @param resource The handle of the resource variable that stores the state of the RNG. The state consists of the counter followed by the key. + * @param alg The RNG algorithm. + * @param delta The amount of advancement. + * @return a new instance of RngReadAndSkip + */ + public RngReadAndSkip rngReadAndSkip(Operand resource, Operand alg, + Operand delta) { + return RngReadAndSkip.create(scope, resource, alg, delta); + } + + /** + * Advance the counter of a counter-based RNG. + * The state of the RNG after + * {@code rng_skip(n)} will be the same as that after {@code stateful_uniform([n])} + * (or any other distribution). The actual increment added to the + * counter is an unspecified implementation detail. + * + * @param resource The handle of the resource variable that stores the state of the RNG. + * @param algorithm The RNG algorithm. + * @param delta The amount of advancement. + * @return a new instance of RngSkip + */ + public RngSkip rngSkip(Operand resource, Operand algorithm, + Operand delta) { + return RngSkip.create(scope, resource, algorithm, delta); + } + /** * The StatefulRandomBinomial operation * @@ -393,6 +559,120 @@ public StatefulStandardNormal statefulStandardNormal( return StatefulStandardNormal.create(scope, resource, algorithm, shape, dtype); } + /** + * Outputs random values from a truncated normal distribution. + * The generated values follow a normal distribution with mean 0 and standard + * deviation 1, except that values whose magnitude is more than 2 standard + * deviations from the mean are dropped and re-picked. + * + * @param data type for {@code output} output + * @param resource The handle of the resource variable that stores the state of the RNG. + * @param algorithm The RNG algorithm. + * @param shape The shape of the output tensor. + * @return a new instance of StatefulTruncatedNormal, with default output types + */ + public StatefulTruncatedNormal statefulTruncatedNormal( + Operand resource, Operand algorithm, + Operand shape) { + return StatefulTruncatedNormal.create(scope, resource, algorithm, shape); + } + + /** + * Outputs random values from a truncated normal distribution. + * The generated values follow a normal distribution with mean 0 and standard + * deviation 1, except that values whose magnitude is more than 2 standard + * deviations from the mean are dropped and re-picked. + * + * @param data type for {@code output} output + * @param resource The handle of the resource variable that stores the state of the RNG. + * @param algorithm The RNG algorithm. + * @param shape The shape of the output tensor. + * @param dtype The type of the output. + * @param data type for {@code StatefulTruncatedNormal} output and operands + * @return a new instance of StatefulTruncatedNormal + */ + public StatefulTruncatedNormal statefulTruncatedNormal( + Operand resource, Operand algorithm, Operand shape, + Class dtype) { + return StatefulTruncatedNormal.create(scope, resource, algorithm, shape, dtype); + } + + /** + * Outputs random values from a uniform distribution. + * The generated values follow a uniform distribution in the range {@code [0, 1)}. The + * lower bound 0 is included in the range, while the upper bound 1 is excluded. + * + * @param data type for {@code output} output + * @param resource The handle of the resource variable that stores the state of the RNG. + * @param algorithm The RNG algorithm. + * @param shape The shape of the output tensor. + * @return a new instance of StatefulUniform, with default output types + */ + public StatefulUniform statefulUniform(Operand resource, + Operand algorithm, Operand shape) { + return StatefulUniform.create(scope, resource, algorithm, shape); + } + + /** + * Outputs random values from a uniform distribution. + * The generated values follow a uniform distribution in the range {@code [0, 1)}. The + * lower bound 0 is included in the range, while the upper bound 1 is excluded. + * + * @param data type for {@code output} output + * @param resource The handle of the resource variable that stores the state of the RNG. + * @param algorithm The RNG algorithm. + * @param shape The shape of the output tensor. + * @param dtype The type of the output. + * @param data type for {@code StatefulUniform} output and operands + * @return a new instance of StatefulUniform + */ + public StatefulUniform statefulUniform(Operand resource, + Operand algorithm, Operand shape, Class dtype) { + return StatefulUniform.create(scope, resource, algorithm, shape, dtype); + } + + /** + * Outputs random integers from a uniform distribution. + * The generated values are uniform integers covering the whole range of {@code dtype}. + * + * @param data type for {@code output} output + * @param resource The handle of the resource variable that stores the state of the RNG. + * @param algorithm The RNG algorithm. + * @param shape The shape of the output tensor. + * @param dtype The type of the output. + * @param data type for {@code StatefulUniformFullInt} output and operands + * @return a new instance of StatefulUniformFullInt + */ + public StatefulUniformFullInt statefulUniformFullInt( + Operand resource, Operand algorithm, Operand shape, + Class dtype) { + return StatefulUniformFullInt.create(scope, resource, algorithm, shape, dtype); + } + + /** + * Outputs random integers from a uniform distribution. + * The generated values are uniform integers in the range {@code [minval, maxval)}. + * The lower bound {@code minval} is included in the range, while the upper bound + * {@code maxval} is excluded. + *

The random integers are slightly biased unless {@code maxval - minval} is an exact + * power of two. The bias is small for values of {@code maxval - minval} significantly + * smaller than the range of the output (either {@code 2^32} or {@code 2^64}). + * + * @param data type for {@code output} output + * @param resource The handle of the resource variable that stores the state of the RNG. + * @param algorithm The RNG algorithm. + * @param shape The shape of the output tensor. + * @param minval Minimum value (inclusive, scalar). + * @param maxval Maximum value (exclusive, scalar). + * @param data type for {@code StatefulUniformInt} output and operands + * @return a new instance of StatefulUniformInt + */ + public StatefulUniformInt statefulUniformInt( + Operand resource, Operand algorithm, Operand shape, + Operand minval, Operand maxval) { + return StatefulUniformInt.create(scope, resource, algorithm, shape, minval, maxval); + } + /** * Draws samples from a multinomial distribution. * @@ -426,6 +706,125 @@ public StatelessMultinomial statelessMultinomial( return StatelessMultinomial.create(scope, logits, numSamples, seed, outputDtype); } + /** + * The StatelessParameterizedTruncatedNormal operation + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param seed 2 seeds (shape [2]). + * @param means The mean parameter of each batch. + * @param stddevs The standard deviation parameter of each batch. Must be greater than 0. + * @param minvals The minimum cutoff. May be -infinity. + * @param maxvals The maximum cutoff. May be +infinity, and must be more than the minval + * for each batch. + * @param data type for {@code StatelessParameterizedTruncatedNormal} output and operands + * @return a new instance of StatelessParameterizedTruncatedNormal + */ + public StatelessParameterizedTruncatedNormal statelessParameterizedTruncatedNormal( + Operand shape, Operand seed, Operand means, + Operand stddevs, Operand minvals, Operand maxvals) { + return StatelessParameterizedTruncatedNormal.create(scope, shape, seed, means, stddevs, minvals, maxvals); + } + + /** + * Outputs deterministic pseudorandom random numbers from a binomial distribution. + * Outputs random values from a binomial distribution. + *

The outputs are a deterministic function of {@code shape}, {@code seed}, {@code counts}, and {@code probs}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param seed 2 seeds (shape [2]). + * @param counts The counts of the binomial distribution. Must be broadcastable with {@code probs}, + * and broadcastable with the rightmost dimensions of {@code shape}. + * @param probs The probability of success for the binomial distribution. Must be broadcastable + * with {@code counts} and broadcastable with the rightmost dimensions of {@code shape}. + * @param data type for {@code StatelessRandomBinomial} output and operands + * @return a new instance of StatelessRandomBinomial, with default output types + */ + public StatelessRandomBinomial statelessRandomBinomial( + Operand shape, Operand seed, Operand counts, + Operand probs) { + return StatelessRandomBinomial.create(scope, shape, seed, counts, probs); + } + + /** + * Outputs deterministic pseudorandom random numbers from a binomial distribution. + * Outputs random values from a binomial distribution. + *

The outputs are a deterministic function of {@code shape}, {@code seed}, {@code counts}, and {@code probs}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param seed 2 seeds (shape [2]). + * @param counts The counts of the binomial distribution. Must be broadcastable with {@code probs}, + * and broadcastable with the rightmost dimensions of {@code shape}. + * @param probs The probability of success for the binomial distribution. Must be broadcastable + * with {@code counts} and broadcastable with the rightmost dimensions of {@code shape}. + * @param dtype The type of the output. + * @param data type for {@code StatelessRandomBinomial} output and operands + * @param data type for {@code StatelessRandomBinomial} output and operands + * @return a new instance of StatelessRandomBinomial + */ + public StatelessRandomBinomial statelessRandomBinomial( + Operand shape, Operand seed, Operand counts, + Operand probs, Class dtype) { + return StatelessRandomBinomial.create(scope, shape, seed, counts, probs, dtype); + } + + /** + * Outputs deterministic pseudorandom random numbers from a gamma distribution. + * Outputs random values from a gamma distribution. + *

The outputs are a deterministic function of the inputs. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param key Key for the counter-based RNG algorithm (shape uint64[1]). + * @param counter Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used. + * @param alg The RNG algorithm (shape int32[]). + * @param alpha The concentration of the gamma distribution. Shape must match the rightmost + * dimensions of {@code shape}. + * @param data type for {@code StatelessRandomGammaV3} output and operands + * @return a new instance of StatelessRandomGamma + */ + public StatelessRandomGamma statelessRandomGamma( + Operand shape, Operand key, + Operand counter, Operand alg, Operand alpha) { + return StatelessRandomGamma.create(scope, shape, key, counter, alg, alpha); + } + + /** + * Picks the best counter-based RNG algorithm based on device. + * This op picks the best counter-based RNG algorithm based on device. + * + * @return a new instance of StatelessRandomGetAlg + */ + public StatelessRandomGetAlg statelessRandomGetAlg() { + return StatelessRandomGetAlg.create(scope); + } + + /** + * Scrambles seed into key and counter, using the best algorithm based on device. + * This op scrambles a shape-[2] seed into a key and a counter, both needed by counter-based RNG algorithms. The scrambing uses the best algorithm based on device. The scrambling is opaque but approximately satisfies the property that different seed results in different key/counter pair (which will in turn result in different random numbers). + * + * @param seed 2 seeds (shape [2]). + * @return a new instance of StatelessRandomGetKeyCounter + */ + public StatelessRandomGetKeyCounter statelessRandomGetKeyCounter( + Operand seed) { + return StatelessRandomGetKeyCounter.create(scope, seed); + } + + /** + * Picks the best algorithm based on device, and scrambles seed into key and counter. + * This op picks the best counter-based RNG algorithm based on device, and scrambles a shape-[2] seed into a key and a counter, both needed by the counter-based algorithm. The scrambling is opaque but approximately satisfies the property that different seed results in different key/counter pair (which will in turn result in different random numbers). + * + * @param seed 2 seeds (shape [2]). + * @return a new instance of StatelessRandomGetKeyCounterAlg + */ + public StatelessRandomGetKeyCounterAlg statelessRandomGetKeyCounterAlg( + Operand seed) { + return StatelessRandomGetKeyCounterAlg.create(scope, seed); + } + /** * Outputs deterministic pseudorandom values from a normal distribution. * The generated values will have mean 0 and standard deviation 1. @@ -458,6 +857,63 @@ public StatelessRandomNormal statelessRandomNormal( return StatelessRandomNormal.create(scope, shape, seed, dtype); } + /** + * Outputs deterministic pseudorandom values from a normal distribution. + * The generated values will have mean 0 and standard deviation 1. + *

The outputs are a deterministic function of {@code shape}, {@code key}, {@code counter} and {@code alg}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param key Key for the counter-based RNG algorithm (shape uint64[1]). + * @param counter Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used. + * @param alg The RNG algorithm (shape int32[]). + * @return a new instance of StatelessRandomNormalV2, with default output types + */ + public StatelessRandomNormalV2 statelessRandomNormalV2(Operand shape, + Operand key, Operand counter, Operand alg) { + return StatelessRandomNormalV2.create(scope, shape, key, counter, alg); + } + + /** + * Outputs deterministic pseudorandom values from a normal distribution. + * The generated values will have mean 0 and standard deviation 1. + *

The outputs are a deterministic function of {@code shape}, {@code key}, {@code counter} and {@code alg}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param key Key for the counter-based RNG algorithm (shape uint64[1]). + * @param counter Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used. + * @param alg The RNG algorithm (shape int32[]). + * @param dtype The type of the output. + * @param data type for {@code StatelessRandomNormalV2} output and operands + * @return a new instance of StatelessRandomNormalV2 + */ + public StatelessRandomNormalV2 statelessRandomNormalV2( + Operand shape, Operand key, + Operand counter, Operand alg, Class dtype) { + return StatelessRandomNormalV2.create(scope, shape, key, counter, alg, dtype); + } + + /** + * Outputs deterministic pseudorandom random numbers from a Poisson distribution. + * Outputs random values from a Poisson distribution. + *

The outputs are a deterministic function of {@code shape}, {@code seed}, and {@code lam}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param seed 2 seeds (shape [2]). + * @param lam The rate of the Poisson distribution. Shape must match the rightmost dimensions + * of {@code shape}. + * @param dtype The type of the output. + * @param data type for {@code StatelessRandomPoisson} output and operands + * @return a new instance of StatelessRandomPoisson + */ + public StatelessRandomPoisson statelessRandomPoisson( + Operand shape, Operand seed, + Operand lam, Class dtype) { + return StatelessRandomPoisson.create(scope, shape, seed, lam, dtype); + } + /** * Outputs deterministic pseudorandom random values from a uniform distribution. * The generated values follow a uniform distribution in the range {@code [0, 1)}. The @@ -492,6 +948,123 @@ public StatelessRandomUniform statelessRandomUniform( return StatelessRandomUniform.create(scope, shape, seed, dtype); } + /** + * Outputs deterministic pseudorandom random integers from a uniform distribution. + * The generated values are uniform integers covering the whole range of {@code dtype}. + *

The outputs are a deterministic function of {@code shape} and {@code seed}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param seed 2 seeds (shape [2]). + * @param dtype The type of the output. + * @param data type for {@code StatelessRandomUniformFullInt} output and operands + * @return a new instance of StatelessRandomUniformFullInt + */ + public StatelessRandomUniformFullInt statelessRandomUniformFullInt( + Operand shape, Operand seed, Class dtype) { + return StatelessRandomUniformFullInt.create(scope, shape, seed, dtype); + } + + /** + * Outputs deterministic pseudorandom random integers from a uniform distribution. + * The generated values are uniform integers covering the whole range of {@code dtype}. + *

The outputs are a deterministic function of {@code shape}, {@code key}, {@code counter} and {@code alg}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param key Key for the counter-based RNG algorithm (shape uint64[1]). + * @param counter Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used. + * @param alg The RNG algorithm (shape int32[]). + * @param dtype The type of the output. + * @param data type for {@code StatelessRandomUniformFullIntV2} output and operands + * @return a new instance of StatelessRandomUniformFullIntV2 + */ + public StatelessRandomUniformFullIntV2 statelessRandomUniformFullIntV2( + Operand shape, Operand key, + Operand counter, Operand alg, Class dtype) { + return StatelessRandomUniformFullIntV2.create(scope, shape, key, counter, alg, dtype); + } + + /** + * Outputs deterministic pseudorandom random integers from a uniform distribution. + * The generated values follow a uniform distribution in the range {@code [minval, maxval)}. + *

The outputs are a deterministic function of {@code shape}, {@code seed}, {@code minval}, and {@code maxval}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param seed 2 seeds (shape [2]). + * @param minval Minimum value (inclusive, scalar). + * @param maxval Maximum value (exclusive, scalar). + * @param data type for {@code StatelessRandomUniformInt} output and operands + * @return a new instance of StatelessRandomUniformInt + */ + public StatelessRandomUniformInt statelessRandomUniformInt( + Operand shape, Operand seed, Operand minval, + Operand maxval) { + return StatelessRandomUniformInt.create(scope, shape, seed, minval, maxval); + } + + /** + * Outputs deterministic pseudorandom random integers from a uniform distribution. + * The generated values follow a uniform distribution in the range {@code [minval, maxval)}. + *

The outputs are a deterministic function of {@code shape}, {@code key}, {@code counter}, {@code alg}, {@code minval} and {@code maxval}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param key Key for the counter-based RNG algorithm (shape uint64[1]). + * @param counter Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used. + * @param alg The RNG algorithm (shape int32[]). + * @param minval Minimum value (inclusive, scalar). + * @param maxval Maximum value (exclusive, scalar). + * @param data type for {@code StatelessRandomUniformIntV2} output and operands + * @return a new instance of StatelessRandomUniformIntV2 + */ + public StatelessRandomUniformIntV2 statelessRandomUniformIntV2( + Operand shape, Operand key, + Operand counter, Operand alg, Operand minval, Operand maxval) { + return StatelessRandomUniformIntV2.create(scope, shape, key, counter, alg, minval, maxval); + } + + /** + * Outputs deterministic pseudorandom random values from a uniform distribution. + * The generated values follow a uniform distribution in the range {@code [0, 1)}. The + * lower bound 0 is included in the range, while the upper bound 1 is excluded. + *

The outputs are a deterministic function of {@code shape}, {@code key}, {@code counter} and {@code alg}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param key Key for the counter-based RNG algorithm (shape uint64[1]). + * @param counter Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used. + * @param alg The RNG algorithm (shape int32[]). + * @return a new instance of StatelessRandomUniformV2, with default output types + */ + public StatelessRandomUniformV2 statelessRandomUniformV2( + Operand shape, Operand key, + Operand counter, Operand alg) { + return StatelessRandomUniformV2.create(scope, shape, key, counter, alg); + } + + /** + * Outputs deterministic pseudorandom random values from a uniform distribution. + * The generated values follow a uniform distribution in the range {@code [0, 1)}. The + * lower bound 0 is included in the range, while the upper bound 1 is excluded. + *

The outputs are a deterministic function of {@code shape}, {@code key}, {@code counter} and {@code alg}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param key Key for the counter-based RNG algorithm (shape uint64[1]). + * @param counter Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used. + * @param alg The RNG algorithm (shape int32[]). + * @param dtype The type of the output. + * @param data type for {@code StatelessRandomUniformV2} output and operands + * @return a new instance of StatelessRandomUniformV2 + */ + public StatelessRandomUniformV2 statelessRandomUniformV2( + Operand shape, Operand key, + Operand counter, Operand alg, Class dtype) { + return StatelessRandomUniformV2.create(scope, shape, key, counter, alg, dtype); + } + /** * Outputs deterministic pseudorandom values from a truncated normal distribution. * The generated values follow a normal distribution with mean 0 and standard @@ -528,6 +1101,75 @@ public StatelessTruncatedNormal statelessTruncatedNormal( return StatelessTruncatedNormal.create(scope, shape, seed, dtype); } + /** + * Outputs deterministic pseudorandom values from a truncated normal distribution. + * The generated values follow a normal distribution with mean 0 and standard + * deviation 1, except that values whose magnitude is more than 2 standard + * deviations from the mean are dropped and re-picked. + *

The outputs are a deterministic function of {@code shape}, {@code key}, {@code counter} and {@code alg}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param key Key for the counter-based RNG algorithm (shape uint64[1]). + * @param counter Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used. + * @param alg The RNG algorithm (shape int32[]). + * @return a new instance of StatelessTruncatedNormalV2, with default output types + */ + public StatelessTruncatedNormalV2 statelessTruncatedNormalV2( + Operand shape, Operand key, + Operand counter, Operand alg) { + return StatelessTruncatedNormalV2.create(scope, shape, key, counter, alg); + } + + /** + * Outputs deterministic pseudorandom values from a truncated normal distribution. + * The generated values follow a normal distribution with mean 0 and standard + * deviation 1, except that values whose magnitude is more than 2 standard + * deviations from the mean are dropped and re-picked. + *

The outputs are a deterministic function of {@code shape}, {@code key}, {@code counter} and {@code alg}. + * + * @param data type for {@code output} output + * @param shape The shape of the output tensor. + * @param key Key for the counter-based RNG algorithm (shape uint64[1]). + * @param counter Initial counter for the counter-based RNG algorithm (shape uint64[2] or uint64[1] depending on the algorithm). If a larger vector is given, only the needed portion on the left (i.e. [:N]) will be used. + * @param alg The RNG algorithm (shape int32[]). + * @param dtype The type of the output. + * @param data type for {@code StatelessTruncatedNormalV2} output and operands + * @return a new instance of StatelessTruncatedNormalV2 + */ + public StatelessTruncatedNormalV2 statelessTruncatedNormalV2( + Operand shape, Operand key, + Operand counter, Operand alg, Class dtype) { + return StatelessTruncatedNormalV2.create(scope, shape, key, counter, alg, dtype); + } + + /** + * Generates labels for candidate sampling with a learned unigram distribution. + * See explanations of candidate sampling and the data formats at + * go/candidate-sampling. + *

For each batch, this op picks a single set of sampled candidate labels. + *

The advantages of sampling candidates per-batch are simplicity and the + * possibility of efficient dense matrix multiplication. The disadvantage is that + * the sampled candidates must be chosen independently of the context and of the + * true labels. + * + * @param trueClasses A batch_size * num_true matrix, in which each row contains the + * IDs of the num_true target_classes in the corresponding original label. + * @param numTrue Number of true labels per context. + * @param numSampled Number of candidates to randomly sample. + * @param unique If unique is true, we sample with rejection, so that all sampled + * candidates in a batch are unique. This requires some approximation to + * estimate the post-rejection sampling probabilities. + * @param rangeMax The sampler will sample integers from the interval [0, range_max). + * @param options carries optional attribute values + * @return a new instance of ThreadUnsafeUnigramCandidateSampler + */ + public ThreadUnsafeUnigramCandidateSampler threadUnsafeUnigramCandidateSampler( + Operand trueClasses, Long numTrue, Long numSampled, Boolean unique, Long rangeMax, + ThreadUnsafeUnigramCandidateSampler.Options... options) { + return ThreadUnsafeUnigramCandidateSampler.create(scope, trueClasses, numTrue, numSampled, unique, rangeMax, options); + } + /** * Outputs random values from a truncated normal distribution. * The generated values follow a normal distribution with mean 0 and standard diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/SparseOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/SparseOps.java index c2b253fe29e..363421a5125 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/SparseOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/SparseOps.java @@ -21,6 +21,7 @@ import org.tensorflow.ndarray.Shape; import org.tensorflow.op.sparse.AddManySparseToTensorsMap; import org.tensorflow.op.sparse.AddSparseToTensorsMap; +import org.tensorflow.op.sparse.DenseCountSparseOutput; import org.tensorflow.op.sparse.DenseToDenseSetOperation; import org.tensorflow.op.sparse.DenseToSparseSetOperation; import org.tensorflow.op.sparse.DeserializeSparse; @@ -31,6 +32,7 @@ import org.tensorflow.op.sparse.SparseBincount; import org.tensorflow.op.sparse.SparseConcat; import org.tensorflow.op.sparse.SparseConditionalAccumulator; +import org.tensorflow.op.sparse.SparseCountSparseOutput; import org.tensorflow.op.sparse.SparseCross; import org.tensorflow.op.sparse.SparseCrossHashed; import org.tensorflow.op.sparse.SparseDenseCwiseAdd; @@ -149,6 +151,25 @@ public AddSparseToTensorsMap addSparseToTensorsMap(Operand sparseIndices return AddSparseToTensorsMap.create(scope, sparseIndices, sparseValues, sparseShape, options); } + /** + * Performs sparse-output bin counting for a tf.tensor input. + * Counts the number of times each value occurs in the input. + * + * @param data type for {@code output_values} output + * @param values Tensor containing data to count. + * @param weights A Tensor of the same shape as indices containing per-index weight values. May + * also be the empty tensor if no weights are used. + * @param binaryOutput Whether to output the number of occurrences of each value or 1. + * @param options carries optional attribute values + * @param data type for {@code DenseCountSparseOutput} output and operands + * @return a new instance of DenseCountSparseOutput + */ + public DenseCountSparseOutput denseCountSparseOutput( + Operand values, Operand weights, Boolean binaryOutput, + DenseCountSparseOutput.Options... options) { + return DenseCountSparseOutput.create(scope, values, weights, binaryOutput, options); + } + /** * Applies set operation along last dimension of 2 {@code Tensor} inputs. * See SetOperationOp::SetOperationFromContext for values of {@code set_operation}. @@ -465,6 +486,27 @@ public SparseConditionalAccumulator sparseConditionalAccumulat return SparseConditionalAccumulator.create(scope, dtype, shape, options); } + /** + * Performs sparse-output bin counting for a sparse tensor input. + * Counts the number of times each value occurs in the input. + * + * @param data type for {@code output_values} output + * @param indices Tensor containing the indices of the sparse tensor to count. + * @param values Tensor containing values of the sparse tensor to count. + * @param denseShape Tensor containing the dense shape of the sparse tensor to count. + * @param weights A Tensor of the same shape as indices containing per-index weight values. + * May also be the empty tensor if no weights are used. + * @param binaryOutput Whether to output the number of occurrences of each value or 1. + * @param options carries optional attribute values + * @param data type for {@code SparseCountSparseOutput} output and operands + * @return a new instance of SparseCountSparseOutput + */ + public SparseCountSparseOutput sparseCountSparseOutput( + Operand indices, Operand values, Operand denseShape, + Operand weights, Boolean binaryOutput, SparseCountSparseOutput.Options... options) { + return SparseCountSparseOutput.create(scope, indices, values, denseShape, weights, binaryOutput, options); + } + /** * Generates sparse cross from a list of sparse and dense tensors. * The op takes two lists, one of 2D {@code SparseTensor} and one of 2D {@code Tensor}, each diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/StringsOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/StringsOps.java index 99c49b96686..61b4655473e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/StringsOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/StringsOps.java @@ -24,6 +24,8 @@ import org.tensorflow.op.strings.ReduceJoin; import org.tensorflow.op.strings.RegexFullMatch; import org.tensorflow.op.strings.RegexReplace; +import org.tensorflow.op.strings.StaticRegexFullMatch; +import org.tensorflow.op.strings.StaticRegexReplace; import org.tensorflow.op.strings.StringFormat; import org.tensorflow.op.strings.StringLength; import org.tensorflow.op.strings.StringNGrams; @@ -34,11 +36,16 @@ import org.tensorflow.op.strings.ToHashBucketFast; import org.tensorflow.op.strings.ToHashBucketStrong; import org.tensorflow.op.strings.ToNumber; +import org.tensorflow.op.strings.UnicodeDecode; +import org.tensorflow.op.strings.UnicodeDecodeWithOffsets; +import org.tensorflow.op.strings.UnicodeEncode; import org.tensorflow.op.strings.UnicodeScript; import org.tensorflow.op.strings.UnicodeTranscode; +import org.tensorflow.op.strings.UnsortedSegmentJoin; import org.tensorflow.op.strings.Upper; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; +import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; import org.tensorflow.types.family.TNumber; @@ -181,6 +188,37 @@ public RegexReplace regexReplace(Operand input, Operand patter return RegexReplace.create(scope, input, pattern, rewrite, options); } + /** + * Check if the input matches the regex pattern. + * The input is a string tensor of any shape. The pattern is the + * regular expression to be matched with every element of the input tensor. + * The boolean values (True or False) of the output tensor indicate + * if the input matches the regex pattern provided. + *

The pattern follows the re2 syntax (https://github.com/google/re2/wiki/Syntax) + * + * @param input A string tensor of the text to be processed. + * @param pattern The regular expression to match the input. + * @return a new instance of StaticRegexFullMatch + */ + public StaticRegexFullMatch staticRegexFullMatch(Operand input, String pattern) { + return StaticRegexFullMatch.create(scope, input, pattern); + } + + /** + * Replaces the match of pattern in input with rewrite. + * It follows the re2 syntax (https://github.com/google/re2/wiki/Syntax) + * + * @param input The text to be processed. + * @param pattern The regular expression to match the input. + * @param rewrite The rewrite to be applied to the matched expression. + * @param options carries optional attribute values + * @return a new instance of StaticRegexReplace + */ + public StaticRegexReplace staticRegexReplace(Operand input, String pattern, + String rewrite, StaticRegexReplace.Options... options) { + return StaticRegexReplace.create(scope, input, pattern, rewrite, options); + } + /** * Formats a string template using a list of tensors. * Formats a string template using a list of tensors, pretty-printing tensor summaries. @@ -505,6 +543,164 @@ public ToNumber toNumber(Operand stringTensor, C return ToNumber.create(scope, stringTensor, outType); } + /** + * Decodes each string in {@code input} into a sequence of Unicode code points. + * The character codepoints for all strings are returned using a single vector + * {@code char_values}, with strings expanded to characters in row-major order. + *

The {@code row_splits} tensor indicates where the codepoints for + * each input string begin and end within the {@code char_values} tensor. + * In particular, the values for the {@code i}th + * string (in row-major order) are stored in the slice + * {@code [row_splits[i]:row_splits[i+1]]}. Thus: + *

    + *
  • {@code char_values[row_splits[i]+j]} is the Unicode codepoint for the {@code j}th + * character in the {@code i}th string (in row-major order).
    • + *
  • {@code row_splits[i+1] - row_splits[i]} is the number of characters in the {@code i}th + * string (in row-major order).
    • + *
+ * + * @param data type for {@code row_splits} output + * @param input The text to be decoded. Can have any shape. Note that the output is flattened + * to a vector of char values. + * @param inputEncoding Text encoding of the input strings. This is any of the encodings supported + * by ICU ucnv algorithmic converters. Examples: {@code "UTF-16", "US ASCII", "UTF-8"}. + * @param options carries optional attribute values + * @return a new instance of UnicodeDecode, with default output types + */ + public UnicodeDecode unicodeDecode(Operand input, String inputEncoding, + UnicodeDecode.Options[] options) { + return UnicodeDecode.create(scope, input, inputEncoding, options); + } + + /** + * Decodes each string in {@code input} into a sequence of Unicode code points. + * The character codepoints for all strings are returned using a single vector + * {@code char_values}, with strings expanded to characters in row-major order. + *

The {@code row_splits} tensor indicates where the codepoints for + * each input string begin and end within the {@code char_values} tensor. + * In particular, the values for the {@code i}th + * string (in row-major order) are stored in the slice + * {@code [row_splits[i]:row_splits[i+1]]}. Thus: + *

    + *
  • {@code char_values[row_splits[i]+j]} is the Unicode codepoint for the {@code j}th + * character in the {@code i}th string (in row-major order).
    • + *
  • {@code row_splits[i+1] - row_splits[i]} is the number of characters in the {@code i}th + * string (in row-major order).
    • + *
+ * + * @param data type for {@code row_splits} output + * @param input The text to be decoded. Can have any shape. Note that the output is flattened + * to a vector of char values. + * @param inputEncoding Text encoding of the input strings. This is any of the encodings supported + * by ICU ucnv algorithmic converters. Examples: {@code "UTF-16", "US ASCII", "UTF-8"}. + * @param Tsplits The value of the Tsplits attribute + * @param options carries optional attribute values + * @param data type for {@code UnicodeDecode} output and operands + * @return a new instance of UnicodeDecode + */ + public UnicodeDecode unicodeDecode(Operand input, + String inputEncoding, Class Tsplits, UnicodeDecode.Options... options) { + return UnicodeDecode.create(scope, input, inputEncoding, Tsplits, options); + } + + /** + * Decodes each string in {@code input} into a sequence of Unicode code points. + * The character codepoints for all strings are returned using a single vector + * {@code char_values}, with strings expanded to characters in row-major order. + * Similarly, the character start byte offsets are returned using a single vector + * {@code char_to_byte_starts}, with strings expanded in row-major order. + *

The {@code row_splits} tensor indicates where the codepoints and start offsets for + * each input string begin and end within the {@code char_values} and + * {@code char_to_byte_starts} tensors. In particular, the values for the {@code i}th + * string (in row-major order) are stored in the slice + * {@code [row_splits[i]:row_splits[i+1]]}. Thus: + *

    + *
  • {@code char_values[row_splits[i]+j]} is the Unicode codepoint for the {@code j}th + * character in the {@code i}th string (in row-major order).
    • + *
  • {@code char_to_bytes_starts[row_splits[i]+j]} is the start byte offset for the {@code j}th + * character in the {@code i}th string (in row-major order).
    • + *
  • {@code row_splits[i+1] - row_splits[i]} is the number of characters in the {@code i}th + * string (in row-major order).
    • + *
+ * + * @param data type for {@code row_splits} output + * @param input The text to be decoded. Can have any shape. Note that the output is flattened + * to a vector of char values. + * @param inputEncoding Text encoding of the input strings. This is any of the encodings supported + * by ICU ucnv algorithmic converters. Examples: {@code "UTF-16", "US ASCII", "UTF-8"}. + * @param options carries optional attribute values + * @return a new instance of UnicodeDecodeWithOffsets, with default output types + */ + public UnicodeDecodeWithOffsets unicodeDecodeWithOffsets(Operand input, + String inputEncoding, UnicodeDecodeWithOffsets.Options[] options) { + return UnicodeDecodeWithOffsets.create(scope, input, inputEncoding, options); + } + + /** + * Decodes each string in {@code input} into a sequence of Unicode code points. + * The character codepoints for all strings are returned using a single vector + * {@code char_values}, with strings expanded to characters in row-major order. + * Similarly, the character start byte offsets are returned using a single vector + * {@code char_to_byte_starts}, with strings expanded in row-major order. + *

The {@code row_splits} tensor indicates where the codepoints and start offsets for + * each input string begin and end within the {@code char_values} and + * {@code char_to_byte_starts} tensors. In particular, the values for the {@code i}th + * string (in row-major order) are stored in the slice + * {@code [row_splits[i]:row_splits[i+1]]}. Thus: + *

    + *
  • {@code char_values[row_splits[i]+j]} is the Unicode codepoint for the {@code j}th + * character in the {@code i}th string (in row-major order).
    • + *
  • {@code char_to_bytes_starts[row_splits[i]+j]} is the start byte offset for the {@code j}th + * character in the {@code i}th string (in row-major order).
    • + *
  • {@code row_splits[i+1] - row_splits[i]} is the number of characters in the {@code i}th + * string (in row-major order).
    • + *
+ * + * @param data type for {@code row_splits} output + * @param input The text to be decoded. Can have any shape. Note that the output is flattened + * to a vector of char values. + * @param inputEncoding Text encoding of the input strings. This is any of the encodings supported + * by ICU ucnv algorithmic converters. Examples: {@code "UTF-16", "US ASCII", "UTF-8"}. + * @param Tsplits The value of the Tsplits attribute + * @param options carries optional attribute values + * @param data type for {@code UnicodeDecodeWithOffsets} output and operands + * @return a new instance of UnicodeDecodeWithOffsets + */ + public UnicodeDecodeWithOffsets unicodeDecodeWithOffsets( + Operand input, String inputEncoding, Class Tsplits, + UnicodeDecodeWithOffsets.Options... options) { + return UnicodeDecodeWithOffsets.create(scope, input, inputEncoding, Tsplits, options); + } + + /** + * Encode a tensor of ints into unicode strings. + * Returns a vector of strings, where {@code output[i]} is constructed by encoding the + * Unicode codepoints in {@code input_values[input_splits[i]:input_splits[i+1]]} + * using {@code output_encoding}. + *
+ *

Example: + * 

+   *  input_values = [72, 101, 108, 108, 111, 87, 111, 114, 108, 100]
+   *  input_splits = [0, 5, 10]
+   *  output_encoding = 'UTF-8'
+   *
+   *  output = ['Hello', 'World']
+   *
+ * + * @param inputValues A 1D tensor containing the unicode codepoints that should be encoded. + * @param inputSplits A 1D tensor specifying how the unicode codepoints should be split into strings. + * In particular, {@code output[i]} is constructed by encoding the codepoints in the + * slice {@code input_values[input_splits[i]:input_splits[i+1]]}. + * @param outputEncoding Unicode encoding of the output strings. Valid encodings are: {@code "UTF-8", "UTF-16-BE", and "UTF-32-BE"}. + * @param options carries optional attribute values + * @return a new instance of UnicodeEncode + */ + public UnicodeEncode unicodeEncode(Operand inputValues, + Operand inputSplits, String outputEncoding, + UnicodeEncode.Options... options) { + return UnicodeEncode.create(scope, inputValues, inputSplits, outputEncoding, options); + } + /** * Determine the script codes of a given tensor of Unicode integer code points. * This operation converts Unicode code points to script codes corresponding to @@ -585,6 +781,21 @@ public UnicodeTranscode unicodeTranscode(Operand input, String inputEnc return UnicodeTranscode.create(scope, input, inputEncoding, outputEncoding, options); } + /** + * The UnsortedSegmentJoin operation + * + * @param inputs The inputs value + * @param segmentIds The segmentIds value + * @param numSegments The numSegments value + * @param options carries optional attribute values + * @return a new instance of UnsortedSegmentJoin + */ + public UnsortedSegmentJoin unsortedSegmentJoin(Operand inputs, + Operand segmentIds, Operand numSegments, + UnsortedSegmentJoin.Options... options) { + return UnsortedSegmentJoin.create(scope, inputs, segmentIds, numSegments, options); + } + /** * Converts all lowercase characters into their respective uppercase replacements. * Example: diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/SummaryOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/SummaryOps.java index eae152ec2a8..0cb2c188fe1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/SummaryOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/SummaryOps.java @@ -19,13 +19,30 @@ import org.tensorflow.Operand; import org.tensorflow.op.summary.AudioSummary; +import org.tensorflow.op.summary.CloseSummaryWriter; +import org.tensorflow.op.summary.CreateSummaryDbWriter; +import org.tensorflow.op.summary.CreateSummaryFileWriter; +import org.tensorflow.op.summary.FlushSummaryWriter; import org.tensorflow.op.summary.HistogramSummary; import org.tensorflow.op.summary.ImageSummary; +import org.tensorflow.op.summary.ImportEvent; import org.tensorflow.op.summary.MergeSummary; import org.tensorflow.op.summary.ScalarSummary; +import org.tensorflow.op.summary.StatsAggregatorSummary; +import org.tensorflow.op.summary.SummaryWriter; import org.tensorflow.op.summary.TensorSummary; +import org.tensorflow.op.summary.WriteAudioSummary; +import org.tensorflow.op.summary.WriteGraphSummary; +import org.tensorflow.op.summary.WriteHistogramSummary; +import org.tensorflow.op.summary.WriteImageSummary; +import org.tensorflow.op.summary.WriteRawProtoSummary; +import org.tensorflow.op.summary.WriteScalarSummary; +import org.tensorflow.op.summary.WriteSummary; import org.tensorflow.types.TFloat32; +import org.tensorflow.types.TInt32; +import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; +import org.tensorflow.types.TUint8; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -68,6 +85,58 @@ public AudioSummary audioSummary(Operand tag, Operand tensor, return AudioSummary.create(scope, tag, tensor, sampleRate, options); } + /** + * The CloseSummaryWriter operation + * + * @param writer The writer value + * @return a new instance of CloseSummaryWriter + */ + public CloseSummaryWriter closeSummaryWriter(Operand writer) { + return CloseSummaryWriter.create(scope, writer); + } + + /** + * The CreateSummaryDbWriter operation + * + * @param writer The writer value + * @param dbUri The dbUri value + * @param experimentName The experimentName value + * @param runName The runName value + * @param userName The userName value + * @return a new instance of CreateSummaryDbWriter + */ + public CreateSummaryDbWriter createSummaryDbWriter(Operand writer, + Operand dbUri, Operand experimentName, Operand runName, + Operand userName) { + return CreateSummaryDbWriter.create(scope, writer, dbUri, experimentName, runName, userName); + } + + /** + * The CreateSummaryFileWriter operation + * + * @param writer The writer value + * @param logdir The logdir value + * @param maxQueue The maxQueue value + * @param flushMillis The flushMillis value + * @param filenameSuffix The filenameSuffix value + * @return a new instance of CreateSummaryFileWriter + */ + public CreateSummaryFileWriter createSummaryFileWriter(Operand writer, + Operand logdir, Operand maxQueue, Operand flushMillis, + Operand filenameSuffix) { + return CreateSummaryFileWriter.create(scope, writer, logdir, maxQueue, flushMillis, filenameSuffix); + } + + /** + * The FlushSummaryWriter operation + * + * @param writer The writer value + * @return a new instance of FlushSummaryWriter + */ + public FlushSummaryWriter flushSummaryWriter(Operand writer) { + return FlushSummaryWriter.create(scope, writer); + } + /** * Outputs a {@code Summary} protocol buffer with a histogram. * The generated @@ -133,6 +202,17 @@ public ImageSummary imageSummary(Operand tag, Operand writer, Operand event) { + return ImportEvent.create(scope, writer, event); + } + /** * Merges summaries. * This op creates a @@ -163,6 +243,26 @@ public ScalarSummary scalarSummary(Operand tags, Operand iterator) { + return StatsAggregatorSummary.create(scope, iterator); + } + + /** + * The SummaryWriter operation + * + * @param options carries optional attribute values + * @return a new instance of SummaryWriter + */ + public SummaryWriter summaryWriter(SummaryWriter.Options... options) { + return SummaryWriter.create(scope, options); + } + /** * Outputs a {@code Summary} protocol buffer with a tensor and per-plugin data. * @@ -177,6 +277,118 @@ public TensorSummary tensorSummary(Operand tag, Operand writer, Operand step, + Operand tag, Operand tensor, Operand sampleRate, + WriteAudioSummary.Options... options) { + return WriteAudioSummary.create(scope, writer, step, tag, tensor, sampleRate, options); + } + + /** + * Writes a graph summary. + * Writes TensorFlow graph {@code tensor} at {@code step} using summary {@code writer}. + * + * @param writer The writer value + * @param step The step value + * @param tensor The tensor value + * @return a new instance of WriteGraphSummary + */ + public WriteGraphSummary writeGraphSummary(Operand writer, Operand step, + Operand tensor) { + return WriteGraphSummary.create(scope, writer, step, tensor); + } + + /** + * Writes a histogram summary. + * Writes histogram {@code values} at {@code step} with {@code tag} using summary {@code writer}. + * + * @param writer The writer value + * @param step The step value + * @param tag The tag value + * @param values The values value + * @return a new instance of WriteHistogramSummary + */ + public WriteHistogramSummary writeHistogramSummary(Operand writer, + Operand step, Operand tag, Operand values) { + return WriteHistogramSummary.create(scope, writer, step, tag, values); + } + + /** + * Writes an image summary. + * Writes image {@code tensor} at {@code step} with {@code tag} using summary {@code writer}. + * {@code tensor} is image with shape [height, width, channels]. + * + * @param writer The writer value + * @param step The step value + * @param tag The tag value + * @param tensor The tensor value + * @param badColor The badColor value + * @param options carries optional attribute values + * @return a new instance of WriteImageSummary + */ + public WriteImageSummary writeImageSummary(Operand writer, Operand step, + Operand tag, Operand tensor, Operand badColor, + WriteImageSummary.Options... options) { + return WriteImageSummary.create(scope, writer, step, tag, tensor, badColor, options); + } + + /** + * Writes a serialized proto summary. + * Writes {@code tensor}, a serialized proto at {@code step} using summary {@code writer}. + * + * @param writer The writer value + * @param step The step value + * @param tensor The tensor value + * @return a new instance of WriteRawProtoSummary + */ + public WriteRawProtoSummary writeRawProtoSummary(Operand writer, + Operand step, Operand tensor) { + return WriteRawProtoSummary.create(scope, writer, step, tensor); + } + + /** + * Writes a scalar summary. + * Writes scalar {@code value} at {@code step} with {@code tag} using summary {@code writer}. + * + * @param writer The writer value + * @param step The step value + * @param tag The tag value + * @param value The value value + * @return a new instance of WriteScalarSummary + */ + public WriteScalarSummary writeScalarSummary(Operand writer, + Operand step, Operand tag, Operand value) { + return WriteScalarSummary.create(scope, writer, step, tag, value); + } + + /** + * Writes a tensor summary. + * Writes {@code tensor} at {@code step} with {@code tag} using summary {@code writer}. + * + * @param writer The writer value + * @param step The step value + * @param tensor The tensor value + * @param tag The tag value + * @param summaryMetadata The summaryMetadata value + * @return a new instance of WriteSummary + */ + public WriteSummary writeSummary(Operand writer, Operand step, + Operand tensor, Operand tag, Operand summaryMetadata) { + return WriteSummary.create(scope, writer, step, tensor, tag, summaryMetadata); + } + /** * Get the parent {@link Ops} object. */ diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/TpuOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/TpuOps.java index 9e2411bc0a1..9fac44a94f3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/TpuOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/TpuOps.java @@ -21,16 +21,31 @@ import org.tensorflow.ConcreteFunction; import org.tensorflow.Operand; import org.tensorflow.ndarray.Shape; +import org.tensorflow.op.tpu.AllToAll; import org.tensorflow.op.tpu.CollateTPUEmbeddingMemory; +import org.tensorflow.op.tpu.CompilationResult; import org.tensorflow.op.tpu.Compile; import org.tensorflow.op.tpu.CompileSucceededAssert; import org.tensorflow.op.tpu.ComputeDedupDataSize; +import org.tensorflow.op.tpu.ComputeDedupDataTupleMask; import org.tensorflow.op.tpu.ConfigureAndInitializeGlobalTPU; +import org.tensorflow.op.tpu.ConfigureDistributedTPU; +import org.tensorflow.op.tpu.ConfigureTPUEmbedding; import org.tensorflow.op.tpu.ConfigureTPUEmbeddingHost; import org.tensorflow.op.tpu.ConfigureTPUEmbeddingMemory; import org.tensorflow.op.tpu.ConnectTPUEmbeddingHosts; import org.tensorflow.op.tpu.ConvertToCooTensor; +import org.tensorflow.op.tpu.CrossReplicaSum; import org.tensorflow.op.tpu.DTensorRestore; +import org.tensorflow.op.tpu.DynamicEnqueueTPUEmbeddingArbitraryTensorBatch; +import org.tensorflow.op.tpu.DynamicEnqueueTPUEmbeddingRaggedTensorBatch; +import org.tensorflow.op.tpu.EmbeddingActivations; +import org.tensorflow.op.tpu.EnqueueTPUEmbeddingArbitraryTensorBatch; +import org.tensorflow.op.tpu.EnqueueTPUEmbeddingBatch; +import org.tensorflow.op.tpu.EnqueueTPUEmbeddingIntegerBatch; +import org.tensorflow.op.tpu.EnqueueTPUEmbeddingRaggedTensorBatch; +import org.tensorflow.op.tpu.EnqueueTPUEmbeddingSparseBatch; +import org.tensorflow.op.tpu.EnqueueTPUEmbeddingSparseTensorBatch; import org.tensorflow.op.tpu.Execute; import org.tensorflow.op.tpu.ExecuteAndUpdateVariables; import org.tensorflow.op.tpu.ExecuteTPUEmbeddingPartitioner; @@ -38,17 +53,78 @@ import org.tensorflow.op.tpu.GetMinibatchSplitsWithPhysicalReplica; import org.tensorflow.op.tpu.GetMinibatchesInCsrWithPhysicalReplica; import org.tensorflow.op.tpu.GlobalIterId; +import org.tensorflow.op.tpu.InfeedDequeue; +import org.tensorflow.op.tpu.InfeedDequeueTuple; +import org.tensorflow.op.tpu.InfeedEnqueue; +import org.tensorflow.op.tpu.InfeedEnqueuePrelinearizedBuffer; +import org.tensorflow.op.tpu.InfeedEnqueueTuple; +import org.tensorflow.op.tpu.IsTPUEmbeddingInitialized; +import org.tensorflow.op.tpu.LoadAllTPUEmbeddingParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingADAMParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingAdadeltaParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingAdagradMomentumParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingAdagradParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingCenteredRMSPropParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingFTRLParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingFrequencyEstimatorParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingMDLAdagradLightParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingMomentumParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingProximalAdagradParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingProximalYogiParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingRMSPropParameters; +import org.tensorflow.op.tpu.LoadTPUEmbeddingStochasticGradientDescentParameters; +import org.tensorflow.op.tpu.MergeDedupData; +import org.tensorflow.op.tpu.OrdinalSelector; +import org.tensorflow.op.tpu.OutfeedDequeue; +import org.tensorflow.op.tpu.OutfeedDequeueTuple; +import org.tensorflow.op.tpu.OutfeedDequeueTupleV2; +import org.tensorflow.op.tpu.OutfeedDequeueV2; +import org.tensorflow.op.tpu.OutfeedEnqueue; +import org.tensorflow.op.tpu.OutfeedEnqueueTuple; +import org.tensorflow.op.tpu.PartitionedCall; +import org.tensorflow.op.tpu.PartitionedInput; import org.tensorflow.op.tpu.PartitionedOutput; +import org.tensorflow.op.tpu.Prelinearize; +import org.tensorflow.op.tpu.PrelinearizeTuple; +import org.tensorflow.op.tpu.RecvTPUEmbeddingActivations; +import org.tensorflow.op.tpu.ReplicateMetadata; +import org.tensorflow.op.tpu.ReplicatedInput; +import org.tensorflow.op.tpu.ReplicatedOutput; +import org.tensorflow.op.tpu.RetrieveAllTPUEmbeddingParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingADAMParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingAdadeltaParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingAdagradMomentumParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingAdagradParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingCenteredRMSPropParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingFTRLParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingFrequencyEstimatorParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingMDLAdagradLightParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingMomentumParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingProximalAdagradParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingProximalYogiParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingRMSPropParameters; +import org.tensorflow.op.tpu.RetrieveTPUEmbeddingStochasticGradientDescentParameters; +import org.tensorflow.op.tpu.SendTPUEmbeddingGradients; +import org.tensorflow.op.tpu.ShutdownDistributedTPU; import org.tensorflow.op.tpu.ShutdownTPUSystem; +import org.tensorflow.op.tpu.SplitDedupData; import org.tensorflow.op.tpu.StoreMinibatchStatisticsInFdo; import org.tensorflow.op.tpu.TPUAnnotateTensorsWithDynamicShape; +import org.tensorflow.op.tpu.TPUCompilationResult; import org.tensorflow.op.tpu.TPUCopyWithDynamicShape; +import org.tensorflow.op.tpu.TPUEmbeddingActivations; +import org.tensorflow.op.tpu.TPUReplicateMetadata; +import org.tensorflow.op.tpu.TPUReplicatedInput; +import org.tensorflow.op.tpu.TPUReplicatedOutput; +import org.tensorflow.op.tpu.TPUReshardVariables; import org.tensorflow.op.tpu.TPURoundRobin; import org.tensorflow.op.tpu.TpuHandleToProtoKey; +import org.tensorflow.op.tpu.WorkerHeartbeat; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; +import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; /** @@ -66,6 +142,39 @@ public final class TpuOps { this.ops = ops; } + /** + * An Op to exchange data across TPU replicas. + * On each replica, the input is split into {@code split_count} blocks along + * {@code split_dimension} and send to the other replicas given group_assignment. After + * receiving {@code split_count} - 1 blocks from other replicas, we concatenate the + * blocks along {@code concat_dimension} as the output. + *

For example, suppose there are 2 TPU replicas: + * replica 0 receives input: {@code [[A, B]]} + * replica 1 receives input: {@code [[C, D]]} + *

group_assignment={@code [[0, 1]]} + * concat_dimension=0 + * split_dimension=1 + * split_count=2 + *

replica 0's output: {@code [[A], [C]]} + * replica 1's output: {@code [[B], [D]]} + * + * @param data type for {@code output} output + * @param input The local input to the sum. + * @param groupAssignment An int32 tensor with shape + * [num_groups, num_replicas_per_group]. {@code group_assignment[i]} represents the + * replica ids in the ith subgroup. + * @param concatDimension The dimension number to concatenate. + * @param splitDimension The dimension number to split. + * @param splitCount The number of splits, this number must equal to the sub-group + * size(group_assignment.get_shape()[1]) + * @param data type for {@code AllToAll} output and operands + * @return a new instance of AllToAll + */ + public AllToAll allToAll(Operand input, Operand groupAssignment, + Long concatDimension, Long splitDimension, Long splitCount) { + return AllToAll.create(scope, input, groupAssignment, concatDimension, splitDimension, splitCount); + } + /** * An op that merges the string-encoded memory config protos from all hosts. * @@ -78,6 +187,18 @@ public CollateTPUEmbeddingMemory collateTPUEmbeddingMemory( return CollateTPUEmbeddingMemory.create(scope, memoryConfigs); } + /** + * Returns the result of a TPU compilation. + * This operation returns the result of a TPU compilation as a serialized + * CompilationResultProto, which holds a status and an error message if an error + * occurred during compilation. + * + * @return a new instance of CompilationResult + */ + public CompilationResult compilationResult() { + return CompilationResult.create(scope); + } + /** * Compiles a computations for execution on one or more TPU devices. * For the internal use of the distributed TPU compiler. @@ -134,6 +255,19 @@ public ComputeDedupDataSize computeDedupDataSize(String config) { return ComputeDedupDataSize.create(scope, config); } + /** + * An op computes tuple mask of deduplication data from embedding core. + * The deduplication data receiving from embedding core is a Tensor with + * type=DT_VARIANT. The tensor itself is an XLA nested tuple, whose elements are + * rank 1 tensors. This op is to represents types and length of these elements. + * + * @param config Serialized TPUEmbeddingConfiguration proto. + * @return a new instance of ComputeDedupDataTupleMask + */ + public ComputeDedupDataTupleMask computeDedupDataTupleMask(String config) { + return ComputeDedupDataTupleMask.create(scope, config); + } + /** * An op that sets up the centralized structures for a distributed TPU system. * @@ -145,6 +279,28 @@ public ConfigureAndInitializeGlobalTPU configureAndInitializeGlobalTPU( return ConfigureAndInitializeGlobalTPU.create(scope, options); } + /** + * Sets up the centralized structures for a distributed TPU system. + * + * @param options carries optional attribute values + * @return a new instance of ConfigureDistributedTPU + */ + public ConfigureDistributedTPU configureDistributedTPU( + ConfigureDistributedTPU.Options... options) { + return ConfigureDistributedTPU.create(scope, options); + } + + /** + * Sets up TPUEmbedding in a distributed TPU system. + * + * @param config Serialized tensorflow.tpu.TPUEmbeddingConfiguration that + * describes the embedding lookups of the program. + * @return a new instance of ConfigureTPUEmbedding + */ + public ConfigureTPUEmbedding configureTPUEmbedding(String config) { + return ConfigureTPUEmbedding.create(scope, config); + } + /** * An op that configures the TPUEmbedding software on a host. * @@ -201,6 +357,27 @@ public ConvertToCooTensor convertToCooTensor(Operand indicesOrRowSplits, return ConvertToCooTensor.create(scope, indicesOrRowSplits, values, weights, sampleCount, combiner); } + /** + * An Op to sum inputs across replicated TPU instances. + * Each instance supplies its own input. + *

For example, suppose there are 8 TPU instances: {@code [A, B, C, D, E, F, G, H]}. + * Passing group_assignment={@code [[0,2,4,6],[1,3,5,7]]} sets {@code A, C, E, G} as group 0, + * and {@code B, D, F, H} as group 1. Thus we get the outputs: + * {@code [A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H, A+C+E+G, B+D+F+H]}. + * + * @param data type for {@code output} output + * @param input The local input to the sum. + * @param groupAssignment An int32 tensor with shape + * [num_groups, num_replicas_per_group]. {@code group_assignment[i]} represents the + * replica ids in the ith subgroup. + * @param data type for {@code CrossReplicaSum} output and operands + * @return a new instance of CrossReplicaSum + */ + public CrossReplicaSum crossReplicaSum(Operand input, + Operand groupAssignment) { + return CrossReplicaSum.create(scope, input, groupAssignment); + } + /** * The DTensorRestoreV2 operation * @@ -218,6 +395,285 @@ public DTensorRestore dTensorRestore(Operand prefix, Operand t return DTensorRestore.create(scope, prefix, tensorNames, shapeAndSlices, inputShapes, inputLayouts, dtypes); } + /** + * Eases the porting of code that uses tf.nn.embedding_lookup_sparse(). + * embedding_indices[i] and aggregation_weights[i] correspond + * to the ith feature. + *

The tensors at corresponding positions in the three input lists (sample_indices, + * embedding_indices and aggregation_weights) must have the same shape, i.e. rank 1 + * with dim_size() equal to the total number of lookups into the table described by + * the corresponding feature. + * + * @param sampleIndicesOrRowSplits A list of rank 2 Tensors specifying the training example to which the + * corresponding embedding_indices and aggregation_weights values belong. + * If the size of its first dimension is 0, we assume each embedding_indices + * belongs to a different sample. Both int32 and int64 are allowed and will + * be converted to int32 internally. + *

Or a list of rank 1 Tensors specifying the row splits for splitting + * embedding_indices and aggregation_weights into rows. It corresponds to + * ids.row_splits in embedding_lookup(), when ids is a RaggedTensor. When + * enqueuing N-D ragged tensor, only the last dimension is allowed to be ragged. + * the row splits is 1-D dense tensor. When empty, we assume a dense tensor is + * passed to the op Both int32 and int64 are allowed and will be converted to + * int32 internally. + * @param embeddingIndices A list of rank 1 Tensors, indices into the embedding + * tables. Both int32 and int64 are allowed and will be converted to + * int32 internally. + * @param aggregationWeights A list of rank 1 Tensors containing per training + * example aggregation weights. Both float32 and float64 are allowed and will + * be converted to float32 internally. + * @param modeOverride A string input that overrides the mode specified in the + * TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference', + * 'training', 'backward_pass_only'}. When set to 'unspecified', the mode set + * in TPUEmbeddingConfiguration is used, otherwise mode_override is used. + * @param deviceOrdinal The TPU device to use. Should be >= 0 and less than the number + * of TPU cores in the task on which the node is placed. + * @param options carries optional attribute values + * @return a new instance of DynamicEnqueueTPUEmbeddingArbitraryTensorBatch + */ + public DynamicEnqueueTPUEmbeddingArbitraryTensorBatch dynamicEnqueueTPUEmbeddingArbitraryTensorBatch( + Iterable> sampleIndicesOrRowSplits, + Iterable> embeddingIndices, + Iterable> aggregationWeights, Operand modeOverride, + Operand deviceOrdinal, + DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.Options... options) { + return DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.create(scope, sampleIndicesOrRowSplits, embeddingIndices, aggregationWeights, modeOverride, deviceOrdinal, options); + } + + /** + * The DynamicEnqueueTPUEmbeddingRaggedTensorBatch operation + * + * @param sampleSplits The sampleSplits value + * @param embeddingIndices The embeddingIndices value + * @param aggregationWeights The aggregationWeights value + * @param modeOverride The modeOverride value + * @param deviceOrdinal The deviceOrdinal value + * @param tableIds The value of the tableIds attribute + * @param options carries optional attribute values + * @return a new instance of DynamicEnqueueTPUEmbeddingRaggedTensorBatch + */ + public DynamicEnqueueTPUEmbeddingRaggedTensorBatch dynamicEnqueueTPUEmbeddingRaggedTensorBatch( + Iterable> sampleSplits, + Iterable> embeddingIndices, + Iterable> aggregationWeights, Operand modeOverride, + Operand deviceOrdinal, List tableIds, + DynamicEnqueueTPUEmbeddingRaggedTensorBatch.Options... options) { + return DynamicEnqueueTPUEmbeddingRaggedTensorBatch.create(scope, sampleSplits, embeddingIndices, aggregationWeights, modeOverride, deviceOrdinal, tableIds, options); + } + + /** + * An op enabling differentiation of TPU Embeddings. + * This op simply returns its first input, which is assumed to have been sliced + * from the Tensors returned by TPUEmbeddingDequeueActivations. The presence of + * this op, and its first argument being a trainable Variable, enables automatic + * differentiation of graphs containing embeddings via the TPU Embedding Python + * libraries. + * + * @param embeddingVariable A trainable variable, enabling optimizers to find this op. + * @param slicedActivations The embedding activations Tensor to return. + * @param tableId The id of the table in the embedding layer configuration from which + * these activations were computed. + * @param lookupId Identifier of the set of embedding indices which produced these + * activations. + * @return a new instance of EmbeddingActivations + */ + public EmbeddingActivations embeddingActivations(Operand embeddingVariable, + Operand slicedActivations, Long tableId, Long lookupId) { + return EmbeddingActivations.create(scope, embeddingVariable, slicedActivations, tableId, lookupId); + } + + /** + * Eases the porting of code that uses tf.nn.embedding_lookup_sparse(). + * embedding_indices[i] and aggregation_weights[i] correspond + * to the ith feature. + *

The tensors at corresponding positions in the three input lists (sample_indices, + * embedding_indices and aggregation_weights) must have the same shape, i.e. rank 1 + * with dim_size() equal to the total number of lookups into the table described by + * the corresponding feature. + * + * @param sampleIndicesOrRowSplits A list of rank 2 Tensors specifying the training example to which the + * corresponding embedding_indices and aggregation_weights values belong. + * If the size of its first dimension is 0, we assume each embedding_indices + * belongs to a different sample. Both int32 and int64 are allowed and will + * be converted to int32 internally. + *

Or a list of rank 1 Tensors specifying the row splits for splitting + * embedding_indices and aggregation_weights into rows. It corresponds to + * ids.row_splits in embedding_lookup(), when ids is a RaggedTensor. When + * enqueuing N-D ragged tensor, only the last dimension is allowed to be ragged. + * the row splits is 1-D dense tensor. When empty, we assume a dense tensor is + * passed to the op Both int32 and int64 are allowed and will be converted to + * int32 internally. + * @param embeddingIndices A list of rank 1 Tensors, indices into the embedding + * tables. Both int32 and int64 are allowed and will be converted to + * int32 internally. + * @param aggregationWeights A list of rank 1 Tensors containing per training + * example aggregation weights. Both float32 and float64 are allowed and will + * be converted to float32 internally. + * @param modeOverride A string input that overrides the mode specified in the + * TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference', + * 'training', 'backward_pass_only'}. When set to 'unspecified', the mode set + * in TPUEmbeddingConfiguration is used, otherwise mode_override is used. + * @param options carries optional attribute values + * @return a new instance of EnqueueTPUEmbeddingArbitraryTensorBatch + */ + public EnqueueTPUEmbeddingArbitraryTensorBatch enqueueTPUEmbeddingArbitraryTensorBatch( + Iterable> sampleIndicesOrRowSplits, + Iterable> embeddingIndices, + Iterable> aggregationWeights, Operand modeOverride, + EnqueueTPUEmbeddingArbitraryTensorBatch.Options... options) { + return EnqueueTPUEmbeddingArbitraryTensorBatch.create(scope, sampleIndicesOrRowSplits, embeddingIndices, aggregationWeights, modeOverride, options); + } + + /** + * An op that enqueues a list of input batch tensors to TPUEmbedding. + * An op that enqueues a list of input batch tensors to TPUEmbedding. + * + * @param batch A list of 1D tensors, one for each embedding table, containing the + * batch inputs encoded as dist_belief.SparseFeatures protos. If the weight + * field in the SparseFeatures proto is not populated for an ID, a weight of + * 1.0 is assumed. + * @param modeOverride A string input that overrides the mode specified in the + * TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference', + * 'training', 'backward_pass_only'}. When set to 'unspecified', the mode set + * in TPUEmbeddingConfiguration is used, otherwise mode_override is used. + * @param options carries optional attribute values + * @return a new instance of EnqueueTPUEmbeddingBatch + */ + public EnqueueTPUEmbeddingBatch enqueueTPUEmbeddingBatch(Iterable> batch, + Operand modeOverride, EnqueueTPUEmbeddingBatch.Options... options) { + return EnqueueTPUEmbeddingBatch.create(scope, batch, modeOverride, options); + } + + /** + * An op that enqueues a list of input batch tensors to TPUEmbedding. + * + * @param batch A list of 1D tensors, one for each embedding table, containing the + * indices into the tables. + * @param modeOverride A string input that overrides the mode specified in the + * TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference', + * 'training', 'backward_pass_only'}. When set to 'unspecified', the mode set + * in TPUEmbeddingConfiguration is used, otherwise mode_override is used. + * @param options carries optional attribute values + * @return a new instance of EnqueueTPUEmbeddingIntegerBatch + */ + public EnqueueTPUEmbeddingIntegerBatch enqueueTPUEmbeddingIntegerBatch( + Iterable> batch, Operand modeOverride, + EnqueueTPUEmbeddingIntegerBatch.Options... options) { + return EnqueueTPUEmbeddingIntegerBatch.create(scope, batch, modeOverride, options); + } + + /** + * Eases the porting of code that uses tf.nn.embedding_lookup(). + * sample_splits[i], embedding_indices[i] and aggregation_weights[i] correspond + * to the ith feature. table_ids[i] indicates which embedding table to look up ith + * feature. + *

The tensors at corresponding positions in two of the input lists, + * embedding_indices and aggregation_weights, must have the same shape, i.e. rank 1 + * with dim_size() equal to the total number of lookups into the table described by + * the corresponding feature. + * + * @param sampleSplits A list of rank 1 Tensors specifying the break points for splitting + * embedding_indices and aggregation_weights into rows. + * It corresponds to ids.row_splits in embedding_lookup(), when ids is a + * RaggedTensor. + * @param embeddingIndices A list of rank 1 Tensors, indices into the embedding tables. + * It corresponds to ids.values in embedding_lookup(), when ids is a RaggedTensor. + * @param aggregationWeights A list of rank 1 Tensors containing per training example + * aggregation weights. It corresponds to the values field of a RaggedTensor + * with the same row_splits as ids in embedding_lookup(), when ids is a + * RaggedTensor. + * @param modeOverride A string input that overrides the mode specified in the + * TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference', + * 'training', 'backward_pass_only'}. When set to 'unspecified', the mode set + * in TPUEmbeddingConfiguration is used, otherwise mode_override is used. + * @param tableIds A list of integers specifying the identifier of the embedding table + * (offset of TableDescriptor in the TPUEmbeddingConfiguration) to lookup the + * corresponding input. The ith input is looked up using table_ids[i]. The size + * of the table_ids list must be equal to that of sample_indices, + * embedding_indices and aggregation_weights. + * @param options carries optional attribute values + * @return a new instance of EnqueueTPUEmbeddingRaggedTensorBatch + */ + public EnqueueTPUEmbeddingRaggedTensorBatch enqueueTPUEmbeddingRaggedTensorBatch( + Iterable> sampleSplits, + Iterable> embeddingIndices, + Iterable> aggregationWeights, Operand modeOverride, + List tableIds, EnqueueTPUEmbeddingRaggedTensorBatch.Options... options) { + return EnqueueTPUEmbeddingRaggedTensorBatch.create(scope, sampleSplits, embeddingIndices, aggregationWeights, modeOverride, tableIds, options); + } + + /** + * An op that enqueues TPUEmbedding input indices from a SparseTensor. + * This Op eases the porting of code that uses embedding_lookup_sparse(), + * although some Python preprocessing of the SparseTensor arguments to + * embedding_lookup_sparse() is required to produce the arguments to this Op, + * since only a single EnqueueTPUEmbeddingSparseBatch Op is allowed per training + * step. + *

The tensors at corresponding positions in the three input lists + * must have the same shape, i.e. rank 1 with dim_size() equal to the total + * number of lookups into the table described by the corresponding table_id. + * + * @param sampleIndices A list of rank 1 Tensors specifying the training example and + * feature to which the corresponding embedding_indices and aggregation_weights + * values belong. sample_indices[i] must equal b * nf + f, where nf is the + * number of features from the corresponding table, f is in [0, nf), and + * b is in [0, batch size). + * @param embeddingIndices A list of rank 1 Tensors, indices into the embedding tables. + * @param aggregationWeights A list of rank 1 Tensors containing per sample -- i.e. per + * (training example, feature) -- aggregation weights. + * @param modeOverride A string input that overrides the mode specified in the + * TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference', + * 'training', 'backward_pass_only'}. When set to 'unspecified', the mode set + * in TPUEmbeddingConfiguration is used, otherwise mode_override is used. + * @param options carries optional attribute values + * @return a new instance of EnqueueTPUEmbeddingSparseBatch + */ + public EnqueueTPUEmbeddingSparseBatch enqueueTPUEmbeddingSparseBatch( + Iterable> sampleIndices, + Iterable> embeddingIndices, + Iterable> aggregationWeights, Operand modeOverride, + EnqueueTPUEmbeddingSparseBatch.Options... options) { + return EnqueueTPUEmbeddingSparseBatch.create(scope, sampleIndices, embeddingIndices, aggregationWeights, modeOverride, options); + } + + /** + * Eases the porting of code that uses tf.nn.embedding_lookup_sparse(). + * sample_indices[i], embedding_indices[i] and aggregation_weights[i] correspond + * to the ith feature. table_ids[i] indicates which embedding table to look up ith + * feature. + *

The tensors at corresponding positions in the three input lists (sample_indices, + * embedding_indices and aggregation_weights) must have the same shape, i.e. rank 1 + * with dim_size() equal to the total number of lookups into the table described by + * the corresponding feature. + * + * @param sampleIndices A list of rank 1 Tensors specifying the training example to + * which the corresponding embedding_indices and aggregation_weights values + * belong. It corresponds to sp_ids.indices[:,0] in embedding_lookup_sparse(). + * @param embeddingIndices A list of rank 1 Tensors, indices into the embedding tables. + * It corresponds to sp_ids.values in embedding_lookup_sparse(). + * @param aggregationWeights A list of rank 1 Tensors containing per training example + * aggregation weights. It corresponds to sp_weights.values in + * embedding_lookup_sparse(). + * @param modeOverride A string input that overrides the mode specified in the + * TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference', + * 'training', 'backward_pass_only'}. When set to 'unspecified', the mode set + * in TPUEmbeddingConfiguration is used, otherwise mode_override is used. + * @param tableIds A list of integers specifying the identifier of the embedding table + * (offset of TableDescriptor in the TPUEmbeddingConfiguration) to lookup the + * corresponding input. The ith input is looked up using table_ids[i]. The size + * of the table_ids list must be equal to that of sample_indices, + * embedding_indices and aggregation_weights. + * @param options carries optional attribute values + * @return a new instance of EnqueueTPUEmbeddingSparseTensorBatch + */ + public EnqueueTPUEmbeddingSparseTensorBatch enqueueTPUEmbeddingSparseTensorBatch( + Iterable> sampleIndices, + Iterable> embeddingIndices, + Iterable> aggregationWeights, Operand modeOverride, + List tableIds, EnqueueTPUEmbeddingSparseTensorBatch.Options... options) { + return EnqueueTPUEmbeddingSparseTensorBatch.create(scope, sampleIndices, embeddingIndices, aggregationWeights, modeOverride, tableIds, options); + } + /** * Op that loads and executes a TPU program on a TPU device. * For the internal use of the distributed TPU compiler. @@ -342,6 +798,576 @@ public GlobalIterId globalIterId() { return GlobalIterId.create(scope); } + /** + * A placeholder op for a value that will be fed into the computation. + * + * @param data type for {@code output} output + * @param dtype The type of elements in the tensor. + * @param shape The shape of the tensor. + * @param data type for {@code InfeedDequeue} output and operands + * @return a new instance of InfeedDequeue + */ + public InfeedDequeue infeedDequeue(Class dtype, Shape shape) { + return InfeedDequeue.create(scope, dtype, shape); + } + + /** + * Fetches multiple values from infeed as an XLA tuple. + * + * @param dtypes The element types of each element in {@code outputs}. + * @param shapes The shapes of each tensor in {@code outputs}. + * @return a new instance of InfeedDequeueTuple + */ + public InfeedDequeueTuple infeedDequeueTuple(List> dtypes, + List shapes) { + return InfeedDequeueTuple.create(scope, dtypes, shapes); + } + + /** + * An op which feeds a single Tensor value into the computation. + * + * @param input A tensor that will be provided using the infeed mechanism. + * @param options carries optional attribute values + * @return a new instance of InfeedEnqueue + */ + public InfeedEnqueue infeedEnqueue(Operand input, + InfeedEnqueue.Options... options) { + return InfeedEnqueue.create(scope, input, options); + } + + /** + * An op which enqueues prelinearized buffer into TPU infeed. + * + * @param input A variant tensor representing linearized output. + * @param options carries optional attribute values + * @return a new instance of InfeedEnqueuePrelinearizedBuffer + */ + public InfeedEnqueuePrelinearizedBuffer infeedEnqueuePrelinearizedBuffer( + Operand input, InfeedEnqueuePrelinearizedBuffer.Options... options) { + return InfeedEnqueuePrelinearizedBuffer.create(scope, input, options); + } + + /** + * Feeds multiple Tensor values into the computation as an XLA tuple. + * + * @param inputs A list of tensors that will be provided using the infeed mechanism. + * @param shapes The shapes of each tensor in {@code inputs}. + * @param options carries optional attribute values + * @return a new instance of InfeedEnqueueTuple + */ + public InfeedEnqueueTuple infeedEnqueueTuple(Iterable> inputs, List shapes, + InfeedEnqueueTuple.Options... options) { + return InfeedEnqueueTuple.create(scope, inputs, shapes, options); + } + + /** + * Whether TPU Embedding is initialized in a distributed TPU system. + * + * @param options carries optional attribute values + * @return a new instance of IsTPUEmbeddingInitialized + */ + public IsTPUEmbeddingInitialized isTPUEmbeddingInitialized( + IsTPUEmbeddingInitialized.Options... options) { + return IsTPUEmbeddingInitialized.create(scope, options); + } + + /** + * An op that loads optimization parameters into embedding memory. + * An op that loads optimization parameters into embedding memory. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct embedding + * table configuration. For example, this op is used to install parameters that are + * loaded from a checkpoint before a training loop is executed. For Adagrad, + * auxiliary1 should be the accumulators. For SGD, all of the auxiliary* values + * should be empty. For FTRL, auxiliary1 should be the accumulators and auxiliary2 + * should be the linear terms. For ADAM, auxiliary1 should be the momenta and + * auxiliary2 should be the velocities. + * + * @param parameters A list of tensors, one for each embedding table, + * containing the initial embedding table parameters to use in embedding + * lookups. + * @param auxiliary1 A list of tensors, one for each embedding table, containing the + * initial values of the first auxiliary optimization parameter to use in embedding + * training loop updates. The shape of each entry is ignored (and thus can be + * empty) for those tables whose optimization algorithms do not have at least one + * auxiliary parameter. + * @param auxiliary2 A list of tensors, one for each embedding table, containing the + * initial values of the second auxiliary optimization parameter to use in + * embedding training loop updates. The shape of each entry is ignored (and thus + * can be empty) for those tables whose optimization algorithms do not have at + * least two auxiliary + * @param auxiliary3 A list of tensors, one for each embedding table, containing the + * initial values of the third auxiliary optimization parameter to use in embedding + * training loop updates. The shape of each entry is ignored (and thus can be + * empty) for those tables whose optimization algorithms do not have three + * auxiliary parameters. + * @param auxiliary4 A list of tensors, one for each embedding table, containing the + * initial values of the second auxiliary optimization parameter to use in + * embedding training loop updates. The shape of each entry is ignored (and thus + * can be empty) for those tables whose optimization algorithms do not have at + * least four auxiliary + * @param auxiliary5 A list of tensors, one for each embedding table, containing the + * initial values of the third auxiliary optimization parameter to use in embedding + * training loop updates. The shape of each entry is ignored (and thus can be + * empty) for those tables whose optimization algorithms do not have five + * auxiliary parameters. + * @param auxiliary6 A list of tensors, one for each embedding table, containing the + * initial values of the second auxiliary optimization parameter to use in + * embedding training loop updates. The shape of each entry is ignored (and thus + * can be empty) for those tables whose optimization algorithms do not have at + * least six auxiliary + * @param auxiliary7 A list of tensors, one for each embedding table, containing the + * initial values of the third auxiliary optimization parameter to use in embedding + * training loop updates. The shape of each entry is ignored (and thus can be + * empty) for those tables whose optimization algorithms do not have sevan + * auxiliary parameters. + * @param config An TPUEmbeddingConfiguration proto describing the + * table parameters being loaded, serialized to a string. + * @param numShards Number of shards into which the embedding tables are divided. + * @param shardId Identifier of shard for this operation. + * @return a new instance of LoadAllTPUEmbeddingParameters + */ + public LoadAllTPUEmbeddingParameters loadAllTPUEmbeddingParameters( + Iterable> parameters, Iterable> auxiliary1, + Iterable> auxiliary2, Iterable> auxiliary3, + Iterable> auxiliary4, Iterable> auxiliary5, + Iterable> auxiliary6, Iterable> auxiliary7, String config, + Long numShards, Long shardId) { + return LoadAllTPUEmbeddingParameters.create(scope, parameters, auxiliary1, auxiliary2, auxiliary3, auxiliary4, auxiliary5, auxiliary6, auxiliary7, config, numShards, shardId); + } + + /** + * Load ADAM embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the ADAM optimization algorithm. + * @param momenta Value of momenta used in the ADAM optimization algorithm. + * @param velocities Value of velocities used in the ADAM optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingADAMParameters + */ + public LoadTPUEmbeddingADAMParameters loadTPUEmbeddingADAMParameters(Operand parameters, + Operand momenta, Operand velocities, Long numShards, Long shardId, + LoadTPUEmbeddingADAMParameters.Options... options) { + return LoadTPUEmbeddingADAMParameters.create(scope, parameters, momenta, velocities, numShards, shardId, options); + } + + /** + * Load Adadelta embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the Adadelta optimization algorithm. + * @param accumulators Value of accumulators used in the Adadelta optimization algorithm. + * @param updates Value of updates used in the Adadelta optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingAdadeltaParameters + */ + public LoadTPUEmbeddingAdadeltaParameters loadTPUEmbeddingAdadeltaParameters( + Operand parameters, Operand accumulators, Operand updates, + Long numShards, Long shardId, LoadTPUEmbeddingAdadeltaParameters.Options... options) { + return LoadTPUEmbeddingAdadeltaParameters.create(scope, parameters, accumulators, updates, numShards, shardId, options); + } + + /** + * Load Adagrad Momentum embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the Adagrad Momentum optimization algorithm. + * @param accumulators Value of accumulators used in the Adagrad Momentum optimization algorithm. + * @param momenta Value of momenta used in the Adagrad Momentum optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingAdagradMomentumParameters + */ + public LoadTPUEmbeddingAdagradMomentumParameters loadTPUEmbeddingAdagradMomentumParameters( + Operand parameters, Operand accumulators, Operand momenta, + Long numShards, Long shardId, LoadTPUEmbeddingAdagradMomentumParameters.Options... options) { + return LoadTPUEmbeddingAdagradMomentumParameters.create(scope, parameters, accumulators, momenta, numShards, shardId, options); + } + + /** + * Load Adagrad embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the Adagrad optimization algorithm. + * @param accumulators Value of accumulators used in the Adagrad optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingAdagradParameters + */ + public LoadTPUEmbeddingAdagradParameters loadTPUEmbeddingAdagradParameters( + Operand parameters, Operand accumulators, Long numShards, Long shardId, + LoadTPUEmbeddingAdagradParameters.Options... options) { + return LoadTPUEmbeddingAdagradParameters.create(scope, parameters, accumulators, numShards, shardId, options); + } + + /** + * Load centered RMSProp embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the centered RMSProp optimization algorithm. + * @param ms Value of ms used in the centered RMSProp optimization algorithm. + * @param mom Value of mom used in the centered RMSProp optimization algorithm. + * @param mg Value of mg used in the centered RMSProp optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingCenteredRMSPropParameters + */ + public LoadTPUEmbeddingCenteredRMSPropParameters loadTPUEmbeddingCenteredRMSPropParameters( + Operand parameters, Operand ms, Operand mom, + Operand mg, Long numShards, Long shardId, + LoadTPUEmbeddingCenteredRMSPropParameters.Options... options) { + return LoadTPUEmbeddingCenteredRMSPropParameters.create(scope, parameters, ms, mom, mg, numShards, shardId, options); + } + + /** + * Load FTRL embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the FTRL optimization algorithm. + * @param accumulators Value of accumulators used in the FTRL optimization algorithm. + * @param linears Value of linears used in the FTRL optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingFTRLParameters + */ + public LoadTPUEmbeddingFTRLParameters loadTPUEmbeddingFTRLParameters(Operand parameters, + Operand accumulators, Operand linears, Long numShards, Long shardId, + LoadTPUEmbeddingFTRLParameters.Options... options) { + return LoadTPUEmbeddingFTRLParameters.create(scope, parameters, accumulators, linears, numShards, shardId, options); + } + + /** + * Load frequency estimator embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the frequency estimator optimization algorithm. + * @param lastHitStep Value of last_hit_step used in the frequency estimator optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingFrequencyEstimatorParameters + */ + public LoadTPUEmbeddingFrequencyEstimatorParameters loadTPUEmbeddingFrequencyEstimatorParameters( + Operand parameters, Operand lastHitStep, Long numShards, Long shardId, + LoadTPUEmbeddingFrequencyEstimatorParameters.Options... options) { + return LoadTPUEmbeddingFrequencyEstimatorParameters.create(scope, parameters, lastHitStep, numShards, shardId, options); + } + + /** + * Load MDL Adagrad Light embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the MDL Adagrad Light optimization algorithm. + * @param accumulators Value of accumulators used in the MDL Adagrad Light optimization algorithm. + * @param weights Value of weights used in the MDL Adagrad Light optimization algorithm. + * @param benefits Value of benefits used in the MDL Adagrad Light optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingMDLAdagradLightParameters + */ + public LoadTPUEmbeddingMDLAdagradLightParameters loadTPUEmbeddingMDLAdagradLightParameters( + Operand parameters, Operand accumulators, Operand weights, + Operand benefits, Long numShards, Long shardId, + LoadTPUEmbeddingMDLAdagradLightParameters.Options... options) { + return LoadTPUEmbeddingMDLAdagradLightParameters.create(scope, parameters, accumulators, weights, benefits, numShards, shardId, options); + } + + /** + * Load Momentum embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the Momentum optimization algorithm. + * @param momenta Value of momenta used in the Momentum optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingMomentumParameters + */ + public LoadTPUEmbeddingMomentumParameters loadTPUEmbeddingMomentumParameters( + Operand parameters, Operand momenta, Long numShards, Long shardId, + LoadTPUEmbeddingMomentumParameters.Options... options) { + return LoadTPUEmbeddingMomentumParameters.create(scope, parameters, momenta, numShards, shardId, options); + } + + /** + * Load proximal Adagrad embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the proximal Adagrad optimization algorithm. + * @param accumulators Value of accumulators used in the proximal Adagrad optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingProximalAdagradParameters + */ + public LoadTPUEmbeddingProximalAdagradParameters loadTPUEmbeddingProximalAdagradParameters( + Operand parameters, Operand accumulators, Long numShards, Long shardId, + LoadTPUEmbeddingProximalAdagradParameters.Options... options) { + return LoadTPUEmbeddingProximalAdagradParameters.create(scope, parameters, accumulators, numShards, shardId, options); + } + + /** + * The LoadTPUEmbeddingProximalYogiParameters operation + * + * @param parameters The parameters value + * @param v The v value + * @param m The m value + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingProximalYogiParameters + */ + public LoadTPUEmbeddingProximalYogiParameters loadTPUEmbeddingProximalYogiParameters( + Operand parameters, Operand v, Operand m, Long numShards, + Long shardId, LoadTPUEmbeddingProximalYogiParameters.Options... options) { + return LoadTPUEmbeddingProximalYogiParameters.create(scope, parameters, v, m, numShards, shardId, options); + } + + /** + * Load RMSProp embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the RMSProp optimization algorithm. + * @param ms Value of ms used in the RMSProp optimization algorithm. + * @param mom Value of mom used in the RMSProp optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingRMSPropParameters + */ + public LoadTPUEmbeddingRMSPropParameters loadTPUEmbeddingRMSPropParameters( + Operand parameters, Operand ms, Operand mom, Long numShards, + Long shardId, LoadTPUEmbeddingRMSPropParameters.Options... options) { + return LoadTPUEmbeddingRMSPropParameters.create(scope, parameters, ms, mom, numShards, shardId, options); + } + + /** + * Load SGD embedding parameters. + * An op that loads optimization parameters into HBM for embedding. Must be + * preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to install + * parameters that are loaded from a checkpoint before a training loop is + * executed. + * + * @param parameters Value of parameters used in the stochastic gradient descent optimization algorithm. + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of LoadTPUEmbeddingStochasticGradientDescentParameters + */ + public LoadTPUEmbeddingStochasticGradientDescentParameters loadTPUEmbeddingStochasticGradientDescentParameters( + Operand parameters, Long numShards, Long shardId, + LoadTPUEmbeddingStochasticGradientDescentParameters.Options... options) { + return LoadTPUEmbeddingStochasticGradientDescentParameters.create(scope, parameters, numShards, shardId, options); + } + + /** + * An op merges elements of integer and float tensors into deduplication data as + * XLA tuple. + * This op merges outputs of SplitDedupDataOp, which gives two 1-D tensors, integer + * and floating point. With respect to tuple_mask, this op merges values of these + * two tensors into an XLA tuple, which should be as same as input to + * SplitDedupDataOp. + * + * @param integerTensor A 1-D integer tensor, includes integer elements of deduplication data tuple. + * @param floatTensor A 1-D float tensor, includes float elements of deduplication data tuple. + * @param tupleMask A serialized TensorProto string of output tuple mask. This mask is a 2-D tensor, + * with first column as tuple element type, and second column as span of this type. + * For example, an output tuple of (1, 2, 0.1, 3), its mask is [[0, 2], [1, 1], [0, + * 1]]. We expect only two types of elements: integer(0) and float(1). + * @param options carries optional attribute values + * @return a new instance of MergeDedupData + */ + public MergeDedupData mergeDedupData(Operand integerTensor, + Operand floatTensor, String tupleMask, MergeDedupData.Options... options) { + return MergeDedupData.create(scope, integerTensor, floatTensor, tupleMask, options); + } + + /** + * A TPU core selector Op. + * This Op produces a set of TPU cores (for warm-up) or a single TPU core + * (for regular inference) to execute the TPU program on. The output is + * consumed by TPUPartitionedCall. + * + * @return a new instance of OrdinalSelector + */ + public OrdinalSelector ordinalSelector() { + return OrdinalSelector.create(scope); + } + + /** + * Retrieves a single tensor from the computation outfeed. + * This operation will block indefinitely until data is available. + * + * @param data type for {@code output} output + * @param dtype The type of elements in the tensor. + * @param shape The shape of the tensor. + * @param options carries optional attribute values + * @param data type for {@code OutfeedDequeue} output and operands + * @return a new instance of OutfeedDequeue + */ + public OutfeedDequeue outfeedDequeue(Class dtype, Shape shape, + OutfeedDequeue.Options... options) { + return OutfeedDequeue.create(scope, dtype, shape, options); + } + + /** + * Retrieve multiple values from the computation outfeed. + * This operation will block indefinitely until data is available. Output {@code i} + * corresponds to XLA tuple element {@code i}. + * + * @param dtypes The element types of each element in {@code outputs}. + * @param shapes The shapes of each tensor in {@code outputs}. + * @param options carries optional attribute values + * @return a new instance of OutfeedDequeueTuple + */ + public OutfeedDequeueTuple outfeedDequeueTuple(List> dtypes, + List shapes, OutfeedDequeueTuple.Options... options) { + return OutfeedDequeueTuple.create(scope, dtypes, shapes, options); + } + + /** + * Retrieve multiple values from the computation outfeed. Device ordinal is a + * tensor allowing dynamic outfeed. + * This operation will block indefinitely until data is available. Output {@code i} + * corresponds to XLA tuple element {@code i}. + * + * @param deviceOrdinal An int scalar tensor, representing the TPU device to use. This should be -1 when + * the Op is running on a TPU device, and >= 0 when the Op is running on the CPU + * device. + * @param dtypes The element types of each element in {@code outputs}. + * @param shapes The shapes of each tensor in {@code outputs}. + * @return a new instance of OutfeedDequeueTupleV2 + */ + public OutfeedDequeueTupleV2 outfeedDequeueTupleV2(Operand deviceOrdinal, + List> dtypes, List shapes) { + return OutfeedDequeueTupleV2.create(scope, deviceOrdinal, dtypes, shapes); + } + + /** + * Retrieves a single tensor from the computation outfeed. Device ordinal is a + * tensor allowing dynamic outfeed. + * This operation will block indefinitely until data is available. + * + * @param data type for {@code output} output + * @param deviceOrdinal An int scalar tensor, representing the TPU device to use. This should be -1 when + * the Op is running on a TPU device, and >= 0 when the Op is running on the CPU + * device. + * @param dtype The type of elements in the tensor. + * @param shape The shape of the tensor. + * @param data type for {@code OutfeedDequeueV2} output and operands + * @return a new instance of OutfeedDequeueV2 + */ + public OutfeedDequeueV2 outfeedDequeueV2(Operand deviceOrdinal, + Class dtype, Shape shape) { + return OutfeedDequeueV2.create(scope, deviceOrdinal, dtype, shape); + } + + /** + * Enqueue a Tensor on the computation outfeed. + * + * @param input A tensor that will be inserted into the outfeed queue. + * @return a new instance of OutfeedEnqueue + */ + public OutfeedEnqueue outfeedEnqueue(Operand input) { + return OutfeedEnqueue.create(scope, input); + } + + /** + * Enqueue multiple Tensor values on the computation outfeed. + * + * @param inputs A list of tensors that will be inserted into the outfeed queue as an + * XLA tuple. + * @return a new instance of OutfeedEnqueueTuple + */ + public OutfeedEnqueueTuple outfeedEnqueueTuple(Iterable> inputs) { + return OutfeedEnqueueTuple.create(scope, inputs); + } + + /** + * Calls a function placed on a specified TPU device. + * + * @param args The arguments to the function. + * @param deviceOrdinal The TPU device ordinal to run the function on. + * @param Tout The types of the outputs of the function. + * @param f The function to call. + * @param options carries optional attribute values + * @return a new instance of PartitionedCall + */ + public PartitionedCall partitionedCall(Iterable> args, Operand deviceOrdinal, + List> Tout, ConcreteFunction f, PartitionedCall.Options... options) { + return PartitionedCall.create(scope, args, deviceOrdinal, Tout, f, options); + } + + /** + * An op that groups a list of partitioned inputs together. Supports ND sharding. + * + * @param data type for {@code output} output + * @param inputs A list of partitioned inputs which must have the same shape. + * @param partitionDims A list of integers describing how each dimension is partitioned. Emptiness + * indicates the inputs are replicated. + * @param options carries optional attribute values + * @param data type for {@code TPUPartitionedInputV2} output and operands + * @return a new instance of PartitionedInput + */ + public PartitionedInput partitionedInput(Iterable> inputs, + List partitionDims, PartitionedInput.Options... options) { + return PartitionedInput.create(scope, inputs, partitionDims, options); + } + /** * An op that demultiplexes a tensor to be sharded by XLA to a list of partitioned * outputs outside the XLA computation. Supports ND sharding. @@ -359,6 +1385,389 @@ public PartitionedOutput partitionedOutput(Operand input return PartitionedOutput.create(scope, inputs, numSplits, partitionDims); } + /** + * An op which linearizes one Tensor value to an opaque variant tensor. + * + * @param input A tensor that will be linearized. + * @param options carries optional attribute values + * @return a new instance of Prelinearize + */ + public Prelinearize prelinearize(Operand input, + Prelinearize.Options... options) { + return Prelinearize.create(scope, input, options); + } + + /** + * An op which linearizes multiple Tensor values to an opaque variant tensor. + * + * @param inputs A list of tensors that will be provided using the infeed mechanism. + * @param shapes The shapes of each tensor in {@code inputs}. + * @param options carries optional attribute values + * @return a new instance of PrelinearizeTuple + */ + public PrelinearizeTuple prelinearizeTuple(Iterable> inputs, List shapes, + PrelinearizeTuple.Options... options) { + return PrelinearizeTuple.create(scope, inputs, shapes, options); + } + + /** + * An op that receives embedding activations on the TPU. + * The TPU system performs the embedding lookups and aggregations specified by + * the arguments to TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch. The + * results of these aggregations are visible to the Tensorflow Graph as the + * outputs of a RecvTPUEmbeddingActivations op. This op returns a list containing + * one Tensor of activations per table specified in the model. There can be at + * most one RecvTPUEmbeddingActivations op in the TPU graph. + * + * @param numOutputs The number of output activation tensors, equal to the number of + * embedding tables in the model. + * @param config Serialized TPUEmbeddingConfiguration proto. + * @return a new instance of RecvTPUEmbeddingActivations + */ + public RecvTPUEmbeddingActivations recvTPUEmbeddingActivations(Long numOutputs, String config) { + return RecvTPUEmbeddingActivations.create(scope, numOutputs, config); + } + + /** + * Metadata indicating how the TPU computation should be replicated. + * This operation holds the metadata common to operations of a {@code tpu.replicate()} computation subgraph. + * + * @param numReplicas Number of replicas of the computation + * @param options carries optional attribute values + * @return a new instance of ReplicateMetadata + */ + public ReplicateMetadata replicateMetadata(Long numReplicas, + ReplicateMetadata.Options... options) { + return ReplicateMetadata.create(scope, numReplicas, options); + } + + /** + * Connects N inputs to an N-way replicated TPU computation. + * This operation holds a replicated input to a {@code tpu.replicate()} computation subgraph. + * Each replicated input has the same shape and type alongside the output. + *

For example: + * 

+   *  %a = "tf.opA"()
+   *  %b = "tf.opB"()
+   *  %replicated_input = "tf.TPUReplicatedInput"(%a, %b)
+   *  %computation = "tf.Computation"(%replicated_input)
+   *
+ *

The above computation has a replicated input of two replicas. + * + * @param data type for {@code output} output + * @param inputs The inputs value + * @param options carries optional attribute values + * @param data type for {@code TPUReplicatedInput} output and operands + * @return a new instance of ReplicatedInput + */ + public ReplicatedInput replicatedInput(Iterable> inputs, + ReplicatedInput.Options... options) { + return ReplicatedInput.create(scope, inputs, options); + } + + /** + * Connects N outputs from an N-way replicated TPU computation. + * This operation holds a replicated output from a {@code tpu.replicate()} computation subgraph. + * Each replicated output has the same shape and type alongside the input. + *

For example: + * 

+   *  %computation = "tf.Computation"()
+   *  %replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
+   *
+ *

The above computation has a replicated output of two replicas. + * + * @param data type for {@code outputs} output + * @param input The input value + * @param numReplicas The value of the numReplicas attribute + * @param data type for {@code TPUReplicatedOutput} output and operands + * @return a new instance of ReplicatedOutput + */ + public ReplicatedOutput replicatedOutput(Operand input, + Long numReplicas) { + return ReplicatedOutput.create(scope, input, numReplicas); + } + + /** + * An op that retrieves optimization parameters from embedding to host memory. + * An op that retrieves optimization parameters from embedding to host memory. + * Must be preceded by a ConfigureTPUEmbeddingHost op that sets up the correct + * embedding table configuration. For example, this op is used to retrieve updated + * parameters before saving a checkpoint. For Adagrad, auxiliary1 will contain the + * accumulators after running this op. For SGD, all of the auxiliary* values will + * be empty (0x0 tensors for that table). For FTRL, auxiliary1 will contain the + * accumulators and auxiliary2 will contain the linear terms. For ADAM, auxiliary1 + * will contain the momenta and auxiliary2 will contain the velocities. + * + * @param NumTables The number of embedding tables. + * @param config An TPUEmbeddingConfiguration proto describing the + * table parameters being loaded, serialized to a string. + * @param numShards Number of shards into which the embedding tables are divided. + * @param shardId Identifier of shard for this operation. + * @return a new instance of RetrieveAllTPUEmbeddingParameters + */ + public RetrieveAllTPUEmbeddingParameters retrieveAllTPUEmbeddingParameters(Long NumTables, + String config, Long numShards, Long shardId) { + return RetrieveAllTPUEmbeddingParameters.create(scope, NumTables, config, numShards, shardId); + } + + /** + * Retrieve ADAM embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingADAMParameters + */ + public RetrieveTPUEmbeddingADAMParameters retrieveTPUEmbeddingADAMParameters(Long numShards, + Long shardId, RetrieveTPUEmbeddingADAMParameters.Options... options) { + return RetrieveTPUEmbeddingADAMParameters.create(scope, numShards, shardId, options); + } + + /** + * Retrieve Adadelta embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingAdadeltaParameters + */ + public RetrieveTPUEmbeddingAdadeltaParameters retrieveTPUEmbeddingAdadeltaParameters( + Long numShards, Long shardId, RetrieveTPUEmbeddingAdadeltaParameters.Options... options) { + return RetrieveTPUEmbeddingAdadeltaParameters.create(scope, numShards, shardId, options); + } + + /** + * Retrieve Adagrad Momentum embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingAdagradMomentumParameters + */ + public RetrieveTPUEmbeddingAdagradMomentumParameters retrieveTPUEmbeddingAdagradMomentumParameters( + Long numShards, Long shardId, + RetrieveTPUEmbeddingAdagradMomentumParameters.Options... options) { + return RetrieveTPUEmbeddingAdagradMomentumParameters.create(scope, numShards, shardId, options); + } + + /** + * Retrieve Adagrad embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingAdagradParameters + */ + public RetrieveTPUEmbeddingAdagradParameters retrieveTPUEmbeddingAdagradParameters(Long numShards, + Long shardId, RetrieveTPUEmbeddingAdagradParameters.Options... options) { + return RetrieveTPUEmbeddingAdagradParameters.create(scope, numShards, shardId, options); + } + + /** + * Retrieve centered RMSProp embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingCenteredRMSPropParameters + */ + public RetrieveTPUEmbeddingCenteredRMSPropParameters retrieveTPUEmbeddingCenteredRMSPropParameters( + Long numShards, Long shardId, + RetrieveTPUEmbeddingCenteredRMSPropParameters.Options... options) { + return RetrieveTPUEmbeddingCenteredRMSPropParameters.create(scope, numShards, shardId, options); + } + + /** + * Retrieve FTRL embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingFTRLParameters + */ + public RetrieveTPUEmbeddingFTRLParameters retrieveTPUEmbeddingFTRLParameters(Long numShards, + Long shardId, RetrieveTPUEmbeddingFTRLParameters.Options... options) { + return RetrieveTPUEmbeddingFTRLParameters.create(scope, numShards, shardId, options); + } + + /** + * Retrieve frequency estimator embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingFrequencyEstimatorParameters + */ + public RetrieveTPUEmbeddingFrequencyEstimatorParameters retrieveTPUEmbeddingFrequencyEstimatorParameters( + Long numShards, Long shardId, + RetrieveTPUEmbeddingFrequencyEstimatorParameters.Options... options) { + return RetrieveTPUEmbeddingFrequencyEstimatorParameters.create(scope, numShards, shardId, options); + } + + /** + * Retrieve MDL Adagrad Light embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingMDLAdagradLightParameters + */ + public RetrieveTPUEmbeddingMDLAdagradLightParameters retrieveTPUEmbeddingMDLAdagradLightParameters( + Long numShards, Long shardId, + RetrieveTPUEmbeddingMDLAdagradLightParameters.Options... options) { + return RetrieveTPUEmbeddingMDLAdagradLightParameters.create(scope, numShards, shardId, options); + } + + /** + * Retrieve Momentum embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingMomentumParameters + */ + public RetrieveTPUEmbeddingMomentumParameters retrieveTPUEmbeddingMomentumParameters( + Long numShards, Long shardId, RetrieveTPUEmbeddingMomentumParameters.Options... options) { + return RetrieveTPUEmbeddingMomentumParameters.create(scope, numShards, shardId, options); + } + + /** + * Retrieve proximal Adagrad embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingProximalAdagradParameters + */ + public RetrieveTPUEmbeddingProximalAdagradParameters retrieveTPUEmbeddingProximalAdagradParameters( + Long numShards, Long shardId, + RetrieveTPUEmbeddingProximalAdagradParameters.Options... options) { + return RetrieveTPUEmbeddingProximalAdagradParameters.create(scope, numShards, shardId, options); + } + + /** + * The RetrieveTPUEmbeddingProximalYogiParameters operation + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingProximalYogiParameters + */ + public RetrieveTPUEmbeddingProximalYogiParameters retrieveTPUEmbeddingProximalYogiParameters( + Long numShards, Long shardId, RetrieveTPUEmbeddingProximalYogiParameters.Options... options) { + return RetrieveTPUEmbeddingProximalYogiParameters.create(scope, numShards, shardId, options); + } + + /** + * Retrieve RMSProp embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingRMSPropParameters + */ + public RetrieveTPUEmbeddingRMSPropParameters retrieveTPUEmbeddingRMSPropParameters(Long numShards, + Long shardId, RetrieveTPUEmbeddingRMSPropParameters.Options... options) { + return RetrieveTPUEmbeddingRMSPropParameters.create(scope, numShards, shardId, options); + } + + /** + * Retrieve SGD embedding parameters. + * An op that retrieves optimization parameters from embedding to host + * memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up + * the correct embedding table configuration. For example, this op is + * used to retrieve updated parameters before saving a checkpoint. + * + * @param numShards The value of the numShards attribute + * @param shardId The value of the shardId attribute + * @param options carries optional attribute values + * @return a new instance of RetrieveTPUEmbeddingStochasticGradientDescentParameters + */ + public RetrieveTPUEmbeddingStochasticGradientDescentParameters retrieveTPUEmbeddingStochasticGradientDescentParameters( + Long numShards, Long shardId, + RetrieveTPUEmbeddingStochasticGradientDescentParameters.Options... options) { + return RetrieveTPUEmbeddingStochasticGradientDescentParameters.create(scope, numShards, shardId, options); + } + + /** + * Performs gradient updates of embedding tables. + * + * @param inputs A TensorList of gradients with which to update embedding tables. + * This argument has the same length and shapes as the return value of + * RecvTPUEmbeddingActivations, but contains gradients of the model's loss + * with respect to the embedding activations. The embedding tables are updated + * from these gradients via the optimizer specified in the TPU embedding + * configuration given to tpu.initialize_system. + * @param learningRates A TensorList of float32 scalars, one for each dynamic learning + * rate tag: see the comments in + * //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. + * Multiple tables can share the same dynamic learning rate tag as specified + * in the configuration. If the learning rates for all tables are constant, + * this list should be empty. + * @param config Serialized TPUEmbeddingConfiguration proto. + * @param options carries optional attribute values + * @return a new instance of SendTPUEmbeddingGradients + */ + public SendTPUEmbeddingGradients sendTPUEmbeddingGradients(Iterable> inputs, + Iterable> learningRates, String config, + SendTPUEmbeddingGradients.Options... options) { + return SendTPUEmbeddingGradients.create(scope, inputs, learningRates, config, options); + } + + /** + * Shuts down a running distributed TPU system. + * The op returns an error if no system is running. + * + * @return a new instance of ShutdownDistributedTPU + */ + public ShutdownDistributedTPU shutdownDistributedTPU() { + return ShutdownDistributedTPU.create(scope); + } + /** * An op that shuts down the TPU system. * @@ -368,6 +1777,33 @@ public ShutdownTPUSystem shutdownTPUSystem() { return ShutdownTPUSystem.create(scope); } + /** + * An op splits input deduplication data XLA tuple into integer and floating point + * tensors. + * Deduplication data is an XLA tuple, which consists of integer and floating point + * values. This op is to split these values into two groups for two types, and + * construct each group as one tensor to return. + * + * @param data type for {@code integer_tensor} output + * @param data type for {@code float_tensor} output + * @param input An XLA tuple including integer and float elements as deduplication data tuple. + * @param integerType integer_tensor type. Allowed types: int32, int64, uint32, uint64. + * @param floatType float_tensor type. Allowed types: half, bfloat16, float. + * @param tupleMask A serialized TensorProto string of output tuple mask. This mask is a 2-D tensor, + * with first column as tuple element type, and second column as span of this type. + * For example, an output tuple of (1, 2, 0.1, 3), its mask is [[0, 2], [1, 1], [0, + * 1]]. We expect only two types of elements: integer(0) and float(1). + * @param options carries optional attribute values + * @param data type for {@code SplitDedupData} output and operands + * @param data type for {@code SplitDedupData} output and operands + * @return a new instance of SplitDedupData + */ + public SplitDedupData splitDedupData( + Operand input, Class integerType, Class floatType, String tupleMask, + SplitDedupData.Options... options) { + return SplitDedupData.create(scope, input, integerType, floatType, tupleMask, options); + } + /** * The StoreMinibatchStatisticsInFdo operation * @@ -399,6 +1835,20 @@ public TPUAnnotateTensorsWithDynamicShape tPUAnnotateTensorsWithDynamicShape( return TPUAnnotateTensorsWithDynamicShape.create(scope, tensors); } + /** + * Returns the result of a TPU compilation. + * This operation returns the result of a TPU compilation as a serialized + * CompilationResultProto, which holds a status and an error message if an error + * occurred during compilation. + * + * @deprecated use {@link org.tensorflow.op.tpu.CompilationResult} instead + * @return a new instance of TPUCompilationResult + */ + @Deprecated + public TPUCompilationResult tPUCompilationResult() { + return TPUCompilationResult.create(scope); + } + /** * Op that copies host tensor to device with dynamic shape support. * For internal use only. @@ -412,6 +1862,112 @@ public TPUCopyWithDynamicShape tPUCopyWithDynamicShape(Iterable> tens return TPUCopyWithDynamicShape.create(scope, tensors, unpaddedSizes); } + /** + * An op enabling differentiation of TPU Embeddings. + * This op simply returns its first input, which is assumed to have been sliced + * from the Tensors returned by TPUEmbeddingDequeueActivations. The presence of + * this op, and its first argument being a trainable Variable, enables automatic + * differentiation of graphs containing embeddings via the TPU Embedding Python + * libraries. + * + * @deprecated use {@link org.tensorflow.op.tpu.EmbeddingActivations} instead + * @param embeddingVariable A trainable variable, enabling optimizers to find this op. + * @param slicedActivations The embedding activations Tensor to return. + * @param tableId The id of the table in the embedding layer configuration from which + * these activations were computed. + * @param lookupId Identifier of the set of embedding indices which produced these + * activations. + * @return a new instance of TPUEmbeddingActivations + */ + @Deprecated + public TPUEmbeddingActivations tPUEmbeddingActivations(Operand embeddingVariable, + Operand slicedActivations, Long tableId, Long lookupId) { + return TPUEmbeddingActivations.create(scope, embeddingVariable, slicedActivations, tableId, lookupId); + } + + /** + * Metadata indicating how the TPU computation should be replicated. + * This operation holds the metadata common to operations of a {@code tpu.replicate()} computation subgraph. + * + * @deprecated use {@link org.tensorflow.op.tpu.ReplicateMetadata} instead + * @param numReplicas Number of replicas of the computation + * @param options carries optional attribute values + * @return a new instance of TPUReplicateMetadata + */ + @Deprecated + public TPUReplicateMetadata tPUReplicateMetadata(Long numReplicas, + TPUReplicateMetadata.Options... options) { + return TPUReplicateMetadata.create(scope, numReplicas, options); + } + + /** + * Connects N inputs to an N-way replicated TPU computation. + * This operation holds a replicated input to a {@code tpu.replicate()} computation subgraph. + * Each replicated input has the same shape and type alongside the output. + *

For example: + * 

+   *  %a = "tf.opA"()
+   *  %b = "tf.opB"()
+   *  %replicated_input = "tf.TPUReplicatedInput"(%a, %b)
+   *  %computation = "tf.Computation"(%replicated_input)
+   *
+ *

The above computation has a replicated input of two replicas. + * + * @param data type for {@code output} output + * @deprecated use {@link org.tensorflow.op.tpu.ReplicatedInput} instead + * @param inputs The inputs value + * @param options carries optional attribute values + * @param data type for {@code TPUReplicatedInput} output and operands + * @return a new instance of TPUReplicatedInput + */ + @Deprecated + public TPUReplicatedInput tPUReplicatedInput(Iterable> inputs, + TPUReplicatedInput.Options... options) { + return TPUReplicatedInput.create(scope, inputs, options); + } + + /** + * Connects N outputs from an N-way replicated TPU computation. + * This operation holds a replicated output from a {@code tpu.replicate()} computation subgraph. + * Each replicated output has the same shape and type alongside the input. + *

For example: + * 

+   *  %computation = "tf.Computation"()
+   *  %replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
+   *
+ *

The above computation has a replicated output of two replicas. + * + * @param data type for {@code outputs} output + * @deprecated use {@link org.tensorflow.op.tpu.ReplicatedOutput} instead + * @param input The input value + * @param numReplicas The value of the numReplicas attribute + * @param data type for {@code TPUReplicatedOutput} output and operands + * @return a new instance of TPUReplicatedOutput + */ + @Deprecated + public TPUReplicatedOutput tPUReplicatedOutput(Operand input, + Long numReplicas) { + return TPUReplicatedOutput.create(scope, input, numReplicas); + } + + /** + * Op that reshards on-device TPU variables to specified state. + * Op that reshards on-device TPU variables to specified state. Internal use only. + *

The sharding state is represented as the key of the compilation that generated + * the sharding/unsharding programs along with the main program. new_format_key + * specifies the desired state, and format_state_var is the current state of the + * variables. + * + * @param vars The vars value + * @param newFormatKey The newFormatKey value + * @param formatStateVar The formatStateVar value + * @return a new instance of TPUReshardVariables + */ + public TPUReshardVariables tPUReshardVariables(Iterable> vars, + Operand newFormatKey, Operand formatStateVar) { + return TPUReshardVariables.create(scope, vars, newFormatKey, formatStateVar); + } + /** * Round-robin load balancing on TPU cores. * A load balancing op that round-robins among TPU cores. @@ -442,6 +1998,18 @@ public TpuHandleToProtoKey tpuHandleToProtoKey(Operand uid) { return TpuHandleToProtoKey.create(scope, uid); } + /** + * Worker heartbeat op. + * Heartbeats may be sent periodically to indicate the coordinator is still active, + * to retrieve the current worker status and to expedite shutdown when necessary. + * + * @param request A string tensor containing a serialized WorkerHeartbeatRequest + * @return a new instance of WorkerHeartbeat + */ + public WorkerHeartbeat workerHeartbeat(Operand request) { + return WorkerHeartbeat.create(scope, request); + } + /** * Get the parent {@link Ops} object. */ diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/TrainOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/TrainOps.java index a4333717ba5..fd4a9bbd02b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/TrainOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/TrainOps.java @@ -25,9 +25,11 @@ import org.tensorflow.op.train.AccumulatorNumAccumulated; import org.tensorflow.op.train.AccumulatorSetGlobalStep; import org.tensorflow.op.train.AccumulatorTakeGradient; +import org.tensorflow.op.train.ApplyAdaMax; import org.tensorflow.op.train.ApplyAdadelta; import org.tensorflow.op.train.ApplyAdagrad; import org.tensorflow.op.train.ApplyAdagradDa; +import org.tensorflow.op.train.ApplyAdagradV2; import org.tensorflow.op.train.ApplyAdam; import org.tensorflow.op.train.ApplyAddSign; import org.tensorflow.op.train.ApplyCenteredRmsProp; @@ -39,12 +41,20 @@ import org.tensorflow.op.train.ApplyProximalGradientDescent; import org.tensorflow.op.train.ApplyRmsProp; import org.tensorflow.op.train.BatchMatMul; +import org.tensorflow.op.train.ComputeBatchSize; import org.tensorflow.op.train.ConditionalAccumulator; +import org.tensorflow.op.train.DistributedSave; import org.tensorflow.op.train.GenerateVocabRemapping; import org.tensorflow.op.train.MergeV2Checkpoints; import org.tensorflow.op.train.NegTrain; import org.tensorflow.op.train.PreventGradient; +import org.tensorflow.op.train.ResourceAccumulatorApplyGradient; +import org.tensorflow.op.train.ResourceAccumulatorNumAccumulated; +import org.tensorflow.op.train.ResourceAccumulatorSetGlobalStep; +import org.tensorflow.op.train.ResourceAccumulatorTakeGradient; +import org.tensorflow.op.train.ResourceApplyAdaMax; import org.tensorflow.op.train.ResourceApplyAdadelta; +import org.tensorflow.op.train.ResourceApplyAdagrad; import org.tensorflow.op.train.ResourceApplyAdagradDa; import org.tensorflow.op.train.ResourceApplyAdam; import org.tensorflow.op.train.ResourceApplyAdamWithAmsgrad; @@ -58,9 +68,11 @@ import org.tensorflow.op.train.ResourceApplyProximalAdagrad; import org.tensorflow.op.train.ResourceApplyProximalGradientDescent; import org.tensorflow.op.train.ResourceApplyRmsProp; +import org.tensorflow.op.train.ResourceConditionalAccumulator; import org.tensorflow.op.train.ResourceSparseApplyAdadelta; import org.tensorflow.op.train.ResourceSparseApplyAdagrad; import org.tensorflow.op.train.ResourceSparseApplyAdagradDa; +import org.tensorflow.op.train.ResourceSparseApplyAdagradV2; import org.tensorflow.op.train.ResourceSparseApplyCenteredRmsProp; import org.tensorflow.op.train.ResourceSparseApplyFtrl; import org.tensorflow.op.train.ResourceSparseApplyKerasMomentum; @@ -73,8 +85,10 @@ import org.tensorflow.op.train.Save; import org.tensorflow.op.train.SaveSlices; import org.tensorflow.op.train.SdcaFprint; +import org.tensorflow.op.train.SdcaOptimizer; import org.tensorflow.op.train.SdcaShrinkL1; import org.tensorflow.op.train.SparseApplyAdadelta; +import org.tensorflow.op.train.SparseApplyAdagrad; import org.tensorflow.op.train.SparseApplyAdagradDa; import org.tensorflow.op.train.SparseApplyCenteredRmsProp; import org.tensorflow.op.train.SparseApplyFtrl; @@ -165,6 +179,32 @@ public AccumulatorTakeGradient accumulatorTakeGradient( return AccumulatorTakeGradient.create(scope, handle, numRequired, dtype); } + /** + * Update '*var' according to the AdaMax algorithm. + * m_t <- beta1 * m_{t-1} + (1 - beta1) * g + * v_t <- max(beta2 * v_{t-1}, abs(g)) + * variable <- variable - learning_rate / (1 - beta1^t) * m_t / (v_t + epsilon) + * + * @param data type for {@code out} output + * @param var Should be from a Variable(). + * @param m Should be from a Variable(). + * @param v Should be from a Variable(). + * @param beta1Power Must be a scalar. + * @param lr Scaling factor. Must be a scalar. + * @param beta1 Momentum factor. Must be a scalar. + * @param beta2 Momentum factor. Must be a scalar. + * @param epsilon Ridge term. Must be a scalar. + * @param grad The gradient. + * @param options carries optional attribute values + * @param data type for {@code ApplyAdaMax} output and operands + * @return a new instance of ApplyAdaMax + */ + public ApplyAdaMax applyAdaMax(Operand var, Operand m, Operand v, + Operand beta1Power, Operand lr, Operand beta1, Operand beta2, Operand epsilon, + Operand grad, ApplyAdaMax.Options... options) { + return ApplyAdaMax.create(scope, var, m, v, beta1Power, lr, beta1, beta2, epsilon, grad, options); + } + /** * Update '*var' according to the adadelta scheme. * accum = rho() * accum + (1 - rho()) * grad.square(); @@ -232,6 +272,26 @@ public ApplyAdagradDa applyAdagradDa(Operand var, return ApplyAdagradDa.create(scope, var, gradientAccumulator, gradientSquaredAccumulator, grad, lr, l1, l2, globalStep, options); } + /** + * Update '*var' according to the adagrad scheme. + * accum += grad * grad + * var -= lr * grad * (1 / sqrt(accum)) + * + * @param data type for {@code out} output + * @param var Should be from a Variable(). + * @param accum Should be from a Variable(). + * @param lr Scaling factor. Must be a scalar. + * @param epsilon Constant factor. Must be a scalar. + * @param grad The gradient. + * @param options carries optional attribute values + * @param data type for {@code ApplyAdagradV2} output and operands + * @return a new instance of ApplyAdagradV2 + */ + public ApplyAdagradV2 applyAdagradV2(Operand var, Operand accum, + Operand lr, Operand epsilon, Operand grad, ApplyAdagradV2.Options... options) { + return ApplyAdagradV2.create(scope, var, accum, lr, epsilon, grad, options); + } + /** * Update '*var' according to the Adam algorithm. * $$\text{lr}t := \mathrm{lr} \cdot \frac{\sqrt{1 - \beta_2^t}}{1 - \beta_1^t}$$ @@ -523,6 +583,16 @@ public BatchMatMul batchMatMul(Operand x, return BatchMatMul.create(scope, x, y, Tout, options); } + /** + * Computes the static batch size of a dataset sans partial batches. + * + * @param inputDataset The inputDataset value + * @return a new instance of ComputeBatchSize + */ + public ComputeBatchSize computeBatchSize(Operand inputDataset) { + return ComputeBatchSize.create(scope, inputDataset); + } + /** * A conditional accumulator for aggregating gradients. * The accumulator accepts gradients marked with local_step greater or @@ -543,6 +613,20 @@ public ConditionalAccumulator conditionalAccumulator(Class return ConditionalAccumulator.create(scope, dtype, shape, options); } + /** + * The DistributedSave operation + * + * @param dataset The dataset value + * @param directory The directory value + * @param address The address value + * @param options carries optional attribute values + * @return a new instance of DistributedSave + */ + public DistributedSave distributedSave(Operand dataset, + Operand directory, Operand address, DistributedSave.Options... options) { + return DistributedSave.create(scope, dataset, directory, address, options); + } + /** * Given a path to new and old vocabulary files, returns a remapping Tensor of * length {@code num_new_vocab}, where {@code remapping[i]} contains the row number in the old @@ -644,6 +728,93 @@ public PreventGradient preventGradient(Operand input, return PreventGradient.create(scope, input, options); } + /** + * Applies a gradient to a given accumulator. + * Does not add if local_step is lesser than the accumulator's global_step. + * + * @param handle The handle to a accumulator. + * @param localStep The local_step value at which the gradient was computed. + * @param gradient A tensor of the gradient to be accumulated. + * @return a new instance of ResourceAccumulatorApplyGradient + */ + public ResourceAccumulatorApplyGradient resourceAccumulatorApplyGradient( + Operand handle, Operand localStep, + Operand gradient) { + return ResourceAccumulatorApplyGradient.create(scope, handle, localStep, gradient); + } + + /** + * Returns the number of gradients aggregated in the given accumulators. + * + * @param handle The handle to an accumulator. + * @return a new instance of ResourceAccumulatorNumAccumulated + */ + public ResourceAccumulatorNumAccumulated resourceAccumulatorNumAccumulated( + Operand handle) { + return ResourceAccumulatorNumAccumulated.create(scope, handle); + } + + /** + * Updates the accumulator with a new value for global_step. + * Logs warning if the accumulator's value is already higher than + * new_global_step. + * + * @param handle The handle to an accumulator. + * @param newGlobalStep The new global_step value to set. + * @return a new instance of ResourceAccumulatorSetGlobalStep + */ + public ResourceAccumulatorSetGlobalStep resourceAccumulatorSetGlobalStep( + Operand handle, Operand newGlobalStep) { + return ResourceAccumulatorSetGlobalStep.create(scope, handle, newGlobalStep); + } + + /** + * Extracts the average gradient in the given ConditionalAccumulator. + * The op blocks until sufficient (i.e., more than num_required) + * gradients have been accumulated. If the accumulator has already + * aggregated more than num_required gradients, it returns the average of + * the accumulated gradients. Also automatically increments the recorded + * global_step in the accumulator by 1, and resets the aggregate to 0. + * + * @param data type for {@code average} output + * @param handle The handle to an accumulator. + * @param numRequired Number of gradients required before we return an aggregate. + * @param dtype The data type of accumulated gradients. Needs to correspond to the type + * of the accumulator. + * @param data type for {@code ResourceAccumulatorTakeGradient} output and operands + * @return a new instance of ResourceAccumulatorTakeGradient + */ + public ResourceAccumulatorTakeGradient resourceAccumulatorTakeGradient( + Operand handle, Operand numRequired, Class dtype) { + return ResourceAccumulatorTakeGradient.create(scope, handle, numRequired, dtype); + } + + /** + * Update '*var' according to the AdaMax algorithm. + * m_t <- beta1 * m_{t-1} + (1 - beta1) * g + * v_t <- max(beta2 * v_{t-1}, abs(g)) + * variable <- variable - learning_rate / (1 - beta1^t) * m_t / (v_t + epsilon) + * + * @param var Should be from a Variable(). + * @param m Should be from a Variable(). + * @param v Should be from a Variable(). + * @param beta1Power Must be a scalar. + * @param lr Scaling factor. Must be a scalar. + * @param beta1 Momentum factor. Must be a scalar. + * @param beta2 Momentum factor. Must be a scalar. + * @param epsilon Ridge term. Must be a scalar. + * @param grad The gradient. + * @param options carries optional attribute values + * @param data type for {@code ResourceApplyAdaMax} output and operands + * @return a new instance of ResourceApplyAdaMax + */ + public ResourceApplyAdaMax resourceApplyAdaMax(Operand var, + Operand m, Operand v, Operand beta1Power, Operand lr, + Operand beta1, Operand beta2, Operand epsilon, Operand grad, + ResourceApplyAdaMax.Options... options) { + return ResourceApplyAdaMax.create(scope, var, m, v, beta1Power, lr, beta1, beta2, epsilon, grad, options); + } + /** * Update '*var' according to the adadelta scheme. * accum = rho() * accum + (1 - rho()) * grad.square(); @@ -669,6 +840,26 @@ public ResourceApplyAdadelta resourceApplyAdadelta(Operand data type for {@code ResourceApplyAdagradV2} output and operands + * @return a new instance of ResourceApplyAdagrad + */ + public ResourceApplyAdagrad resourceApplyAdagrad(Operand var, + Operand accum, Operand lr, Operand epsilon, Operand grad, + ResourceApplyAdagrad.Options... options) { + return ResourceApplyAdagrad.create(scope, var, accum, lr, epsilon, grad, options); + } + /** * Update '*var' according to the proximal adagrad scheme. * @@ -995,6 +1186,28 @@ public ResourceApplyRmsProp resourceApplyRmsProp(Operand data type for {@code ResourceConditionalAccumulator} output and operands + * @return a new instance of ResourceConditionalAccumulator + */ + public ResourceConditionalAccumulator resourceConditionalAccumulator( + Class dtype, Shape shape, ResourceConditionalAccumulator.Options... options) { + return ResourceConditionalAccumulator.create(scope, dtype, shape, options); + } + /** * var: Should be from a Variable(). * @@ -1063,6 +1276,29 @@ public ResourceSparseApplyAdagradDa resourceSparseApplyAdagrad return ResourceSparseApplyAdagradDa.create(scope, var, gradientAccumulator, gradientSquaredAccumulator, grad, indices, lr, l1, l2, globalStep, options); } + /** + * Update relevant entries in '*var' and '*accum' according to the adagrad scheme. + * That is for rows we have grad for, we update var and accum as follows: + * accum += grad * grad + * var -= lr * grad * (1 / sqrt(accum)) + * + * @param var Should be from a Variable(). + * @param accum Should be from a Variable(). + * @param lr Learning rate. Must be a scalar. + * @param epsilon Constant factor. Must be a scalar. + * @param grad The gradient. + * @param indices A vector of indices into the first dimension of var and accum. + * @param options carries optional attribute values + * @param data type for {@code ResourceSparseApplyAdagradV2} output and operands + * @return a new instance of ResourceSparseApplyAdagradV2 + */ + public ResourceSparseApplyAdagradV2 resourceSparseApplyAdagradV2( + Operand var, Operand accum, Operand lr, + Operand epsilon, Operand grad, Operand indices, + ResourceSparseApplyAdagradV2.Options... options) { + return ResourceSparseApplyAdagradV2.create(scope, var, accum, lr, epsilon, grad, indices, options); + } + /** * Update '*var' according to the centered RMSProp algorithm. * The centered RMSProp algorithm uses an estimate of the centered second moment @@ -1382,6 +1618,59 @@ public SdcaFprint sdcaFprint(Operand input) { return SdcaFprint.create(scope, input); } + /** + * Distributed version of Stochastic Dual Coordinate Ascent (SDCA) optimizer for + * linear models with L1 + L2 regularization. As global optimization objective is + * strongly-convex, the optimizer optimizes the dual objective at each step. The + * optimizer applies each update one example at a time. Examples are sampled + * uniformly, and the optimizer is learning rate free and enjoys linear convergence + * rate. + *

Proximal Stochastic Dual Coordinate Ascent .
+ * Shai Shalev-Shwartz, Tong Zhang. 2012 + *

$$Loss Objective = \sum f_{i} (wx_{i}) + (l2 / 2) * |w|^2 + l1 * |w|$$ + *

Adding vs. Averaging in Distributed Primal-Dual Optimization .
+ * Chenxin Ma, Virginia Smith, Martin Jaggi, Michael I. Jordan, + * Peter Richtarik, Martin Takac. 2015 + *

Stochastic Dual Coordinate Ascent with Adaptive Probabilities .
+ * Dominik Csiba, Zheng Qu, Peter Richtarik. 2015 + * + * @param sparseExampleIndices a list of vectors which contain example indices. + * @param sparseFeatureIndices a list of vectors which contain feature indices. + * @param sparseFeatureValues a list of vectors which contains feature value + * associated with each feature group. + * @param denseFeatures a list of matrices which contains the dense feature values. + * @param exampleWeights a vector which contains the weight associated with each + * example. + * @param exampleLabels a vector which contains the label/target associated with each + * example. + * @param sparseIndices a list of vectors where each value is the indices which has + * corresponding weights in sparse_weights. This field maybe omitted for the + * dense approach. + * @param sparseWeights a list of vectors where each value is the weight associated with + * a sparse feature group. + * @param denseWeights a list of vectors where the values are the weights associated + * with a dense feature group. + * @param exampleStateData a list of vectors containing the example state data. + * @param lossType Type of the primal loss. Currently SdcaSolver supports logistic, + * squared and hinge losses. + * @param l1 Symmetric l1 regularization strength. + * @param l2 Symmetric l2 regularization strength. + * @param numLossPartitions Number of partitions of the global loss function. + * @param numInnerIterations Number of iterations per mini-batch. + * @param options carries optional attribute values + * @return a new instance of SdcaOptimizer + */ + public SdcaOptimizer sdcaOptimizer(Iterable> sparseExampleIndices, + Iterable> sparseFeatureIndices, + Iterable> sparseFeatureValues, Iterable> denseFeatures, + Operand exampleWeights, Operand exampleLabels, + Iterable> sparseIndices, Iterable> sparseWeights, + Iterable> denseWeights, Operand exampleStateData, String lossType, + Float l1, Float l2, Long numLossPartitions, Long numInnerIterations, + SdcaOptimizer.Options... options) { + return SdcaOptimizer.create(scope, sparseExampleIndices, sparseFeatureIndices, sparseFeatureValues, denseFeatures, exampleWeights, exampleLabels, sparseIndices, sparseWeights, denseWeights, exampleStateData, lossType, l1, l2, numLossPartitions, numInnerIterations, options); + } + /** * Applies L1 regularization shrink step on the parameters. * @@ -1417,6 +1706,29 @@ public SparseApplyAdadelta sparseApplyAdadelta(Operand v return SparseApplyAdadelta.create(scope, var, accum, accumUpdate, lr, rho, epsilon, grad, indices, options); } + /** + * Update relevant entries in '*var' and '*accum' according to the adagrad scheme. + * That is for rows we have grad for, we update var and accum as follows: + * $$accum += grad * grad$$ + * $$var -= lr * grad * (1 / sqrt(accum))$$ + * + * @param data type for {@code out} output + * @param var Should be from a Variable(). + * @param accum Should be from a Variable(). + * @param lr Learning rate. Must be a scalar. + * @param epsilon Constant factor. Must be a scalar. + * @param grad The gradient. + * @param indices A vector of indices into the first dimension of var and accum. + * @param options carries optional attribute values + * @param data type for {@code SparseApplyAdagradV2} output and operands + * @return a new instance of SparseApplyAdagrad + */ + public SparseApplyAdagrad sparseApplyAdagrad(Operand var, + Operand accum, Operand lr, Operand epsilon, Operand grad, + Operand indices, SparseApplyAdagrad.Options... options) { + return SparseApplyAdagrad.create(scope, var, accum, lr, epsilon, grad, indices, options); + } + /** * Update entries in '*var' and '*accum' according to the proximal adagrad scheme. * diff --git a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/XlaOps.java b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/XlaOps.java index 116388309ba..c1a5a97d146 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/XlaOps.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/annotations/org/tensorflow/op/XlaOps.java @@ -21,8 +21,15 @@ import org.tensorflow.ConcreteFunction; import org.tensorflow.Operand; import org.tensorflow.ndarray.Shape; +import org.tensorflow.op.xla.AssignVariableConcatND; +import org.tensorflow.op.xla.ConcatND; +import org.tensorflow.op.xla.ReadVariableSplitND; +import org.tensorflow.op.xla.SplitND; import org.tensorflow.op.xla.XlaHostCompute; import org.tensorflow.op.xla.XlaRecvFromHost; +import org.tensorflow.op.xla.XlaRecvTPUEmbeddingActivations; +import org.tensorflow.op.xla.XlaRecvTPUEmbeddingDeduplicationData; +import org.tensorflow.op.xla.XlaSendTPUEmbeddingGradients; import org.tensorflow.op.xla.XlaSendToHost; import org.tensorflow.op.xla.XlaSparseCoreAdagrad; import org.tensorflow.op.xla.XlaSparseCoreAdagradMomentum; @@ -55,6 +62,215 @@ public final class XlaOps { this.ops = ops; } + /** + * Concats input tensor across all dimensions. + * An op which merges slices the input tensor based on the given num_splits + * attribute, strips paddings optionally, and writes the merged tensor without + * paddings to the resource variable. + *

This op may be generated via the TPU bridge. + *

For example, with {@code input} tensor: + * 

+   *  [[0, 1],
+   *   [4, 5]]
+   *  [[2, 3],
+   *   [6, 7]]
+   *  [[8, 9],
+   *   [12, 13]]
+   *  [[10, 11],
+   *   [14, 15]]
+   *
+ *

{@code num_splits}: + * 

+   *  [2, 2]
+   *
+ *

and {@code paddings}: + * 

+   *  [1, 1]
+   *
+ *

the expected {@code outputs} is: + * 

+   *  [[0, 1, 2],
+   *   [4, 5, 6],
+   *   [8, 9, 10]]
+   *
+ * + * @param resource Resource variable for concatenated input tensors across all dimensions. + * } + * in_arg { + * name: "inputs" + * description: <<END + * Input tensor slices in row-major order to merge across all dimensions. All + * inputs must have the same shape. + * } + * out_arg { + * name: "output" + * description: <<END + * Output tensor formed from merging input slices based on num_concats defined. + * @param inputs The inputs value + * @param numConcats Number of ways to merge per dimension. + * @param options carries optional attribute values + * @return a new instance of AssignVariableConcatND + */ + public AssignVariableConcatND assignVariableConcatND(Operand resource, + Iterable> inputs, List numConcats, + AssignVariableConcatND.Options... options) { + return AssignVariableConcatND.create(scope, resource, inputs, numConcats, options); + } + + /** + * Concats input tensor across all dimensions. + * An op which merges slices the input tensor based on the given num_splits + * attribute, strips paddings optionally, and returns the merged tensor without + * paddings. + *

This op may be generated via the TPU bridge. + *

For example, with {@code input} tensor: + * 

+   *  [[0, 1],
+   *   [4, 5]]
+   *  [[2, 3],
+   *   [6, 7]]
+   *  [[8, 9],
+   *   [12, 13]]
+   *  [[10, 11],
+   *   [14, 15]]
+   *
+ *

{@code num_splits}: + * 

+   *  [2, 2]
+   *
+ *

and {@code paddings}: + * 

+   *  [1, 1]
+   *
+ *

the expected {@code outputs} is: + * 

+   *  [[0, 1, 2],
+   *   [4, 5, 6],
+   *   [8, 9, 10]]
+   *
+ * + * @param data type for {@code output} output + * @param inputs Input tensor slices in row-major order to merge across all dimensions. All + * inputs must have the same shape. + * } + * out_arg { + * name: "output" + * description: <<END + * Output tensor formed from merging input slices based on num_concats defined. + * @param numConcats Number of ways to merge per dimension. + * @param options carries optional attribute values + * @param data type for {@code XlaConcatND} output and operands + * @return a new instance of ConcatND + */ + public ConcatND concatND(Iterable> inputs, List numConcats, + ConcatND.Options... options) { + return ConcatND.create(scope, inputs, numConcats, options); + } + + /** + * Splits resource variable input tensor across all dimensions. + * An op which splits the resource variable input tensor based on the given + * num_splits attribute, pads slices optionally, and returned the slices. Slices + * are returned in row-major order. + *

This op may be generated via the TPU bridge. + *

For example, with {@code input} tensor: + * 

+   *  [[0, 1, 2],
+   *   [3, 4, 5],
+   *   [6, 7, 8]]
+   *
+ *

{@code num_splits}: + * 

+   *  [2, 2]
+   *
+ *

and {@code paddings}: + * 

+   *  [1, 1]
+   *
+ *

the expected {@code outputs} is: + * 

+   *  [[0, 1],
+   *   [3, 4]]
+   *  [[2, 0],
+   *   [5, 0]]
+   *  [[6, 7],
+   *   [0, 0]]
+   *  [[8, 0],
+   *   [0, 0]]
+   *
+ * + * @param data type for {@code outputs} output + * @param resource Resource variable of input tensor to split across all dimensions. + * } + * out_arg { + * name: "outputs" + * description: <<END + * Output slices based on input and num_splits defined, in row-major order. + * @param T The value of the T attribute + * @param N The value of the N attribute + * @param numSplits Number of ways to split per dimension. Shape dimensions must be evenly + * divisible. + * @param options carries optional attribute values + * @param data type for {@code ReadVariableXlaSplitND} output and operands + * @return a new instance of ReadVariableSplitND + */ + public ReadVariableSplitND readVariableSplitND( + Operand resource, Class T, Long N, List numSplits, + ReadVariableSplitND.Options... options) { + return ReadVariableSplitND.create(scope, resource, T, N, numSplits, options); + } + + /** + * Splits input tensor across all dimensions. + * An op which slices the input tensor based on the given num_splits attribute, + * pads slices optionally, and returned the slices. Slices are returned in + * row-major order. + *

This op may be generated via the TPU bridge. + *

For example, with {@code input} tensor: + * 

+   *  [[0, 1, 2],
+   *   [3, 4, 5],
+   *   [6, 7, 8]]
+   *
+ *

{@code num_splits}: + * 

+   *  [2, 2]
+   *
+ *

and {@code paddings}: + * 

+   *  [1, 1]
+   *
+ *

the expected {@code outputs} is: + * 

+   *  [[0, 1],
+   *   [3, 4]]
+   *  [[2, 0],
+   *   [5, 0]]
+   *  [[6, 7],
+   *   [0, 0]]
+   *  [[8, 0],
+   *   [0, 0]]
+   *
+ * + * @param data type for {@code outputs} output + * @param input Input tensor to split across all dimensions. + * } + * out_arg { + * name: "outputs" + * description: <<END + * Output slices based on input and num_splits defined, in row-major order. + * @param N The value of the N attribute + * @param numSplits Number of ways to split per dimension. Shape dimensions must be evenly + * divisible. + * @param options carries optional attribute values + * @param data type for {@code XlaSplitND} output and operands + * @return a new instance of SplitND + */ + public SplitND splitND(Operand input, Long N, List numSplits, + SplitND.Options... options) { + return SplitND.create(scope, input, N, numSplits, options); + } + /** * A pseudo-op to represent host-side computation in an XLA program. * @@ -94,6 +310,76 @@ public XlaRecvFromHost xlaRecvFromHost(Class Toutput, Sh return XlaRecvFromHost.create(scope, Toutput, shape, key); } + /** + * An op that receives embedding activations on the TPU. + * The TPU system performs the embedding lookups and aggregations. The results of + * these aggregations are visible to the Tensorflow Graph as the outputs of a + * XlaRecvTPUEmbeddingActivations Op. This op returns a list containing one + * Tensor of activations per table specified in the model. + * + * @param deduplicationData A Tensor with type=DT_VARIANT containing the deduplication + * data. The tensor is an XLA nested tuple containing N elements (where N is + * the ratio of the number of embedding to tensor cores per TPU chip). Each + * element of the nested tuple is a tuple of rank 1 tensors. Each tensor either + * contains indices (DT_UINT32) for embedding lookup on the TensorCore or + * weights (DT_FLOAT) to apply to the output of the embedding lookup operation. + * @param numTables The number of output activation tensors. If feature descriptor is + * present in the tpu embedding config, it is equal to the number of features + * otherwise equal to number of embedding tables in the model. + * @param config Serialized TPUEmbeddingConfiguration proto. + * @return a new instance of XlaRecvTPUEmbeddingActivations + */ + public XlaRecvTPUEmbeddingActivations xlaRecvTPUEmbeddingActivations( + Operand deduplicationData, Long numTables, String config) { + return XlaRecvTPUEmbeddingActivations.create(scope, deduplicationData, numTables, config); + } + + /** + * Receives deduplication data (indices and weights) from the embedding core. + * The deduplication data is a Tensor with type=DT_VARIANT. The tensor itself is an + * XLA nested tuple containing N elements (where N is the ratio of the number of + * embedding to tensor cores per TPU chip). Each element of the nested tuple is a + * tuple of rank 1 tensors. Each tensor either contains indices (DT_UINT32) for + * embedding lookup on the TensorCore or weights (DT_FLOAT) to apply to the output + * of the embedding lookup operation. + * + * @param config Serialized TPUEmbeddingConfiguration proto. + * @return a new instance of XlaRecvTPUEmbeddingDeduplicationData + */ + public XlaRecvTPUEmbeddingDeduplicationData xlaRecvTPUEmbeddingDeduplicationData(String config) { + return XlaRecvTPUEmbeddingDeduplicationData.create(scope, config); + } + + /** + * An op that performs gradient updates of embedding tables. + * The gradients argument is a TensorList having the same length and shapes as the + * return value of XlaRecvTPUEmbeddingActivations, but contains gradients of the + * model's loss with respect to the embedding activations. The embedding tables are + * updated from these gradients via the optimizer specified in the + * TPUEmbeddingConfiguration proto given to tpu.initialize_system. + * + * @param gradients A TensorList of gradients with which to update embedding tables. + * @param learningRates A TensorList of learning rates used for updating the embedding + * tables via the optimizer. The length of the TensorList must be equal to the + * number of dynamic learning rate tags specified in the + * TPUEmbeddingConfiguration proto. + * @param deduplicationData A Tensor with type=DT_VARIANT containing the deduplication + * data. The tensor is an XLA nested tuple containing N elements (where N is + * the ratio of the number of embedding to tensor cores per TPU chip). Each + * element of the nested tuple is a tuple of rank 1 tensors. Each tensor either + * contains indices (DT_UINT32) for embedding lookup on the TensorCore or + * weights (DT_FLOAT) to apply to the output of the embedding lookup operation. + * @param config Serialized TPUEmbeddingConfiguration proto. + * @param options carries optional attribute values + * @return a new instance of XlaSendTPUEmbeddingGradients + */ + public XlaSendTPUEmbeddingGradients xlaSendTPUEmbeddingGradients( + Iterable> gradients, Iterable> learningRates, + Operand deduplicationData, String config, + XlaSendTPUEmbeddingGradients.Options... options) { + return XlaSendTPUEmbeddingGradients.create(scope, gradients, learningRates, deduplicationData, config, options); + } + /** * An op to send a tensor to the host. * input: the tensor that will be sent to the host. diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KMC2ChainInitialization.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KMC2ChainInitialization.java index 9cef8e6f2fd..fb5a6c31581 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KMC2ChainInitialization.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KMC2ChainInitialization.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -43,6 +44,9 @@ opType = KMC2ChainInitialization.OP_NAME, inputsClass = KMC2ChainInitialization.Inputs.class ) +@Operator( + group = "cluster" +) public final class KMC2ChainInitialization extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KmeansPlusPlusInitialization.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KmeansPlusPlusInitialization.java index 0004fce5305..820fdabb7b9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KmeansPlusPlusInitialization.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/cluster/KmeansPlusPlusInitialization.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -43,6 +44,9 @@ opType = KmeansPlusPlusInitialization.OP_NAME, inputsClass = KmeansPlusPlusInitialization.Inputs.class ) +@Operator( + group = "cluster" +) public final class KmeansPlusPlusInitialization extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveAllToAll.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveAllToAll.java index ab3d80e1064..99ccff79289 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveAllToAll.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveAllToAll.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -43,6 +44,9 @@ opType = CollectiveAllToAll.OP_NAME, inputsClass = CollectiveAllToAll.Inputs.class ) +@Operator( + group = "collective" +) public final class CollectiveAllToAll extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveAssignGroup.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveAssignGroup.java index 3827bb1a158..598986e3da3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveAssignGroup.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveAssignGroup.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; /** @@ -38,6 +39,9 @@ opType = CollectiveAssignGroup.OP_NAME, inputsClass = CollectiveAssignGroup.Inputs.class ) +@Operator( + group = "collective" +) public final class CollectiveAssignGroup extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveBcastRecv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveBcastRecv.java index 566b48c0c27..332b5dcf9ab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveBcastRecv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveBcastRecv.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = CollectiveBcastRecv.OP_NAME, inputsClass = CollectiveBcastRecv.Inputs.class ) +@Operator( + group = "collective" +) public final class CollectiveBcastRecv extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveBcastSend.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveBcastSend.java index 4b093258948..ee495b56951 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveBcastSend.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveBcastSend.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = CollectiveBcastSend.OP_NAME, inputsClass = CollectiveBcastSend.Inputs.class ) +@Operator( + group = "collective" +) public final class CollectiveBcastSend extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveGather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveGather.java index 2b6dc692673..d3997e8743f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveGather.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveGather.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -47,6 +48,9 @@ opType = CollectiveGather.OP_NAME, inputsClass = CollectiveGather.Inputs.class ) +@Operator( + group = "collective" +) public final class CollectiveGather extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveInitializeCommunicator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveInitializeCommunicator.java index 696b2eb5e23..e6ea4b8b79c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveInitializeCommunicator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveInitializeCommunicator.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = CollectiveInitializeCommunicator.OP_NAME, inputsClass = CollectiveInitializeCommunicator.Inputs.class ) +@Operator( + group = "collective" +) public final class CollectiveInitializeCommunicator extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectivePermute.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectivePermute.java index 009f589f181..9fd029facf3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectivePermute.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectivePermute.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -46,6 +47,9 @@ opType = CollectivePermute.OP_NAME, inputsClass = CollectivePermute.Inputs.class ) +@Operator( + group = "collective" +) public final class CollectivePermute extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveReduce.java index 7f10a3ac4cd..7eab3bb0f17 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveReduce.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -43,6 +44,9 @@ opType = CollectiveReduce.OP_NAME, inputsClass = CollectiveReduce.Inputs.class ) +@Operator( + group = "collective" +) public final class CollectiveReduce extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveReduceScatter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveReduceScatter.java index eaae924f2df..5ab06edf273 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveReduceScatter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/collective/CollectiveReduceScatter.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -47,6 +48,9 @@ opType = CollectiveReduceScatter.OP_NAME, inputsClass = CollectiveReduceScatter.Inputs.class ) +@Operator( + group = "collective" +) public final class CollectiveReduceScatter extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousHashTable.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousHashTable.java index e33aaca7845..0dcfd8003dc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousHashTable.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousHashTable.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -48,6 +49,7 @@ opType = AnonymousHashTable.OP_NAME, inputsClass = AnonymousHashTable.Inputs.class ) +@Operator public final class AnonymousHashTable extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableDenseHashTable.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableDenseHashTable.java index d5902c86c9d..9e3baa34b1f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableDenseHashTable.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableDenseHashTable.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -52,6 +53,7 @@ opType = AnonymousMutableDenseHashTable.OP_NAME, inputsClass = AnonymousMutableDenseHashTable.Inputs.class ) +@Operator public final class AnonymousMutableDenseHashTable extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableHashTable.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableHashTable.java index a55af101afa..a010e6bc71c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableHashTable.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableHashTable.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -49,6 +50,7 @@ opType = AnonymousMutableHashTable.OP_NAME, inputsClass = AnonymousMutableHashTable.Inputs.class ) +@Operator public final class AnonymousMutableHashTable extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableHashTableOfTensors.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableHashTableOfTensors.java index 00c5eea12bf..263438f8099 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableHashTableOfTensors.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/AnonymousMutableHashTableOfTensors.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -50,6 +51,7 @@ opType = AnonymousMutableHashTableOfTensors.OP_NAME, inputsClass = AnonymousMutableHashTableOfTensors.Inputs.class ) +@Operator public final class AnonymousMutableHashTableOfTensors extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastGradientArgs.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastGradientArgs.java index 68283116a9e..fe9cf0e7039 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastGradientArgs.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/BroadcastGradientArgs.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +43,7 @@ opType = BroadcastGradientArgs.OP_NAME, inputsClass = BroadcastGradientArgs.Inputs.class ) +@Operator public final class BroadcastGradientArgs extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantFromComponents.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantFromComponents.java index 0c98059a436..bc5322574d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantFromComponents.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantFromComponents.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,7 @@ opType = CompositeTensorVariantFromComponents.OP_NAME, inputsClass = CompositeTensorVariantFromComponents.Inputs.class ) +@Operator public final class CompositeTensorVariantFromComponents extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantToComponents.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantToComponents.java index 11d268dfdd2..40f7209a04a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantToComponents.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/CompositeTensorVariantToComponents.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -45,6 +46,7 @@ opType = CompositeTensorVariantToComponents.OP_NAME, inputsClass = CompositeTensorVariantToComponents.Inputs.class ) +@Operator public final class CompositeTensorVariantToComponents extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ConcatOffset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ConcatOffset.java new file mode 100644 index 00000000000..df14b30a11b --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ConcatOffset.java @@ -0,0 +1,146 @@ +/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +=======================================================================*/ + +// This class has been generated, DO NOT EDIT! + +package org.tensorflow.op.core; + +import java.util.Arrays; +import java.util.Iterator; +import java.util.List; +import org.tensorflow.GraphOperation; +import org.tensorflow.Operand; +import org.tensorflow.Operation; +import org.tensorflow.OperationBuilder; +import org.tensorflow.Output; +import org.tensorflow.op.Operands; +import org.tensorflow.op.RawOp; +import org.tensorflow.op.RawOpInputs; +import org.tensorflow.op.Scope; +import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; +import org.tensorflow.proto.DataType; +import org.tensorflow.types.TInt32; +import org.tensorflow.types.family.TNumber; + +/** + * Computes offsets of concat inputs within its output. + * For example: + *
+ *
+ *
+ *

x = [2, 2, 7] + * y = [2, 3, 7] + * z = [2, 9, 7] + * offsets = concat_offset(1, [x, y, z]) + * [list(off.numpy()) for off in offsets] + * [[0, 0, 0], [0, 2, 0], [0, 5, 0]] + *

+ *
+ *
+ *

This is typically used by gradient computations for a concat operation. + * + * @param data type for {@code offset} output + */ +@OpMetadata( + opType = ConcatOffset.OP_NAME, + inputsClass = ConcatOffset.Inputs.class +) +@Operator +public final class ConcatOffset extends RawOp implements Iterable> { + /** + * The name of this op, as known by TensorFlow core engine + */ + public static final String OP_NAME = "ConcatOffset"; + + private List> offset; + + @SuppressWarnings("unchecked") + public ConcatOffset(Operation operation) { + super(operation, OP_NAME); + int outputIdx = 0; + int offsetLength = operation.outputListLength("offset"); + offset = Arrays.asList((Output[]) operation.outputList(outputIdx, offsetLength)); + outputIdx += offsetLength; + } + + /** + * Factory method to create a class wrapping a new ConcatOffset operation. + * + * @param scope current scope + * @param concatDim The dimension along which to concatenate. + * @param shape The {@code N} int32 or int64 vectors representing shape of tensors being concatenated. + * @param data type for {@code ConcatOffset} output and operands + * @return a new instance of ConcatOffset + */ + @Endpoint( + describeByClass = true + ) + public static ConcatOffset create(Scope scope, Operand concatDim, + Iterable> shape) { + OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "ConcatOffset"); + opBuilder.addInput(concatDim.asOutput()); + opBuilder.addInputList(Operands.asOutputs(shape)); + return new ConcatOffset<>(opBuilder.build()); + } + + /** + * Gets offset. + * The {@code N} vectors representing the starting offset + * of input tensors within the concatenated output with type matching {@code shape}. + * @return offset. + */ + public List> offset() { + return offset; + } + + @Override + @SuppressWarnings({"rawtypes", "unchecked"}) + public Iterator> iterator() { + return (Iterator) offset.iterator(); + } + + @OpInputsMetadata( + outputsClass = ConcatOffset.class + ) + public static class Inputs extends RawOpInputs> { + /** + * The dimension along which to concatenate. + */ + public final Operand concatDim; + + /** + * The {@code N} int32 or int64 vectors representing shape of tensors being concatenated. + */ + public final Iterable> shape; + + /** + * The shapeType attribute + */ + public final DataType shapeType; + + public Inputs(GraphOperation op) { + super(new ConcatOffset<>(op), op, Arrays.asList("shape_type")); + int inputIndex = 0; + concatDim = (Operand) op.input(inputIndex++); + int shapeLength = op.inputListLength("shape"); + shape = Arrays.asList((Operand[]) op.inputList(inputIndex, shapeLength)); + inputIndex += shapeLength; + shapeType = op.attributes().getAttrType("shape_type"); + } + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeviceIndex.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeviceIndex.java index 1e35dc9c5d7..fb05a1a7a09 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeviceIndex.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DeviceIndex.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; /** @@ -43,6 +44,7 @@ opType = DeviceIndex.OP_NAME, inputsClass = DeviceIndex.Inputs.class ) +@Operator public final class DeviceIndex extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DummyMemoryCache.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DummyMemoryCache.java index b7ac9a7d91d..d4dcdcb0735 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DummyMemoryCache.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/DummyMemoryCache.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -38,6 +39,7 @@ opType = DummyMemoryCache.OP_NAME, inputsClass = DummyMemoryCache.Inputs.class ) +@Operator public final class DummyMemoryCache extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Enter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Enter.java index e434208bade..baed3b18053 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Enter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Enter.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -46,6 +47,7 @@ opType = Enter.OP_NAME, inputsClass = Enter.Inputs.class ) +@Operator public final class Enter extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Exit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Exit.java index 9e73edcb1f4..c1535016b59 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Exit.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Exit.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,7 @@ opType = Exit.OP_NAME, inputsClass = Exit.Inputs.class ) +@Operator public final class Exit extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/FakeParam.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/FakeParam.java new file mode 100644 index 00000000000..1f6ceeecc16 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/FakeParam.java @@ -0,0 +1,125 @@ +/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +=======================================================================*/ + +// This class has been generated, DO NOT EDIT! + +package org.tensorflow.op.core; + +import java.util.Arrays; +import org.tensorflow.GraphOperation; +import org.tensorflow.Operand; +import org.tensorflow.Operation; +import org.tensorflow.OperationBuilder; +import org.tensorflow.Output; +import org.tensorflow.ndarray.Shape; +import org.tensorflow.op.Operands; +import org.tensorflow.op.RawOp; +import org.tensorflow.op.RawOpInputs; +import org.tensorflow.op.Scope; +import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; +import org.tensorflow.proto.DataType; +import org.tensorflow.types.family.TType; + +/** + * This op is used as a placeholder in If branch functions. It doesn't provide a + * valid output when run, so must either be removed (e.g. replaced with a + * function input) or guaranteed not to be used (e.g. if mirroring an + * intermediate output needed for the gradient computation of the other branch). + * + * @param data type for {@code output} output + */ +@OpMetadata( + opType = FakeParam.OP_NAME, + inputsClass = FakeParam.Inputs.class +) +@Operator +public final class FakeParam extends RawOp implements Operand { + /** + * The name of this op, as known by TensorFlow core engine + */ + public static final String OP_NAME = "FakeParam"; + + private Output output; + + public FakeParam(Operation operation) { + super(operation, OP_NAME); + int outputIdx = 0; + output = operation.output(outputIdx++); + } + + /** + * Factory method to create a class wrapping a new FakeParam operation. + * + * @param scope current scope + * @param dtype The type of the output. + * @param shape 

+   * The purported shape of the output. This is only used for shape inference;
+   * the output will not necessarily have this shape. Can be a partial shape.
+   *
+ * @param data type for {@code FakeParam} output and operands + * @return a new instance of FakeParam + */ + @Endpoint( + describeByClass = true + ) + public static FakeParam create(Scope scope, Class dtype, Shape shape) { + OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "FakeParam"); + opBuilder.setAttr("dtype", Operands.toDataType(dtype)); + opBuilder.setAttr("shape", shape); + return new FakeParam<>(opBuilder.build()); + } + + /** + * Gets output. + * 
+   * \"Fake\" output value. This should not be consumed by another op.
+   *
+ * @return output. + */ + public Output output() { + return output; + } + + @Override + public Output asOutput() { + return output; + } + + @OpInputsMetadata( + outputsClass = FakeParam.class + ) + public static class Inputs extends RawOpInputs> { + /** + * The type of the output. + */ + public final DataType dtype; + + /** + * The purported shape of the output. This is only used for shape inference; + * the output will not necessarily have this shape. Can be a partial shape. + */ + public final Shape shape; + + public Inputs(GraphOperation op) { + super(new FakeParam<>(op), op, Arrays.asList("dtype", "shape")); + int inputIndex = 0; + dtype = op.attributes().getAttrType("dtype"); + shape = op.attributes().getAttrShape("shape"); + } + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/FileSystemSetConfiguration.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/FileSystemSetConfiguration.java index b5ec9c52eb9..42984c322f9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/FileSystemSetConfiguration.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/FileSystemSetConfiguration.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** @@ -37,6 +38,7 @@ opType = FileSystemSetConfiguration.OP_NAME, inputsClass = FileSystemSetConfiguration.Inputs.class ) +@Operator public final class FileSystemSetConfiguration extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetElementAtIndex.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetElementAtIndex.java index 27851f15855..065404e1735 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetElementAtIndex.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetElementAtIndex.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -44,6 +45,7 @@ opType = GetElementAtIndex.OP_NAME, inputsClass = GetElementAtIndex.Inputs.class ) +@Operator public final class GetElementAtIndex extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetOptions.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetOptions.java index 43cf53336ce..f6c5340587d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetOptions.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/GetOptions.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -39,6 +40,7 @@ opType = GetOptions.OP_NAME, inputsClass = GetOptions.Inputs.class ) +@Operator public final class GetOptions extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCholesky.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/HostConst.java similarity index 55% rename from tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCholesky.java rename to tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/HostConst.java index 735ee23319f..b655a120ae6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCholesky.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/HostConst.java @@ -15,7 +15,7 @@ // This class has been generated, DO NOT EDIT! -package org.tensorflow.op.risc; +package org.tensorflow.op.core; import java.util.Arrays; import org.tensorflow.GraphOperation; @@ -23,53 +23,59 @@ import org.tensorflow.Operation; import org.tensorflow.OperationBuilder; import org.tensorflow.Output; +import org.tensorflow.Tensor; +import org.tensorflow.op.Operands; import org.tensorflow.op.RawOp; import org.tensorflow.op.RawOpInputs; import org.tensorflow.op.Scope; import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; +import org.tensorflow.types.family.TType; /** - * The RiscCholesky operation + * Returns a constant tensor on the host. Only for writing C++ tests. * * @param data type for {@code output} output */ @OpMetadata( - opType = RiscCholesky.OP_NAME, - inputsClass = RiscCholesky.Inputs.class + opType = HostConst.OP_NAME, + inputsClass = HostConst.Inputs.class ) -public final class RiscCholesky extends RawOp implements Operand { +@Operator +public final class HostConst extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine */ - public static final String OP_NAME = "RiscCholesky"; + public static final String OP_NAME = "HostConst"; private Output output; - public RiscCholesky(Operation operation) { + public HostConst(Operation operation) { super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } /** - * Factory method to create a class wrapping a new RiscCholesky operation. + * Factory method to create a class wrapping a new HostConst operation. * * @param scope current scope - * @param input The input value - * @param data type for {@code RiscCholesky} output and operands - * @return a new instance of RiscCholesky + * @param value Attr {@code value} is the tensor to return. + * @param dtype The value of the dtype attribute + * @param data type for {@code HostConst} output and operands + * @return a new instance of HostConst */ @Endpoint( describeByClass = true ) - public static RiscCholesky create(Scope scope, Operand input) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscCholesky"); - opBuilder.addInput(input.asOutput()); - return new RiscCholesky<>(opBuilder.build()); + public static HostConst create(Scope scope, Tensor value, Class dtype) { + OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "HostConst"); + opBuilder.setAttr("value", value); + opBuilder.setAttr("dtype", Operands.toDataType(dtype)); + return new HostConst<>(opBuilder.build()); } /** @@ -87,24 +93,24 @@ public Output asOutput() { } @OpInputsMetadata( - outputsClass = RiscCholesky.class + outputsClass = HostConst.class ) - public static class Inputs extends RawOpInputs> { + public static class Inputs extends RawOpInputs> { /** - * The input input + * Attr `value` is the tensor to return. */ - public final Operand input; + public final Tensor value; /** - * The T attribute + * The dtype attribute */ - public final DataType T; + public final DataType dtype; public Inputs(GraphOperation op) { - super(new RiscCholesky<>(op), op, Arrays.asList("T")); + super(new HostConst<>(op), op, Arrays.asList("value", "dtype")); int inputIndex = 0; - input = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); + value = op.attributes().getAttrTensor("value"); + dtype = op.attributes().getAttrType("dtype"); } } } diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LinSpace.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LinSpace.java index 3ae58b9158c..3473ddf487e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LinSpace.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LinSpace.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -48,6 +49,7 @@ opType = LinSpace.OP_NAME, inputsClass = LinSpace.Inputs.class ) +@Operator public final class LinSpace extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableRemove.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableRemove.java index e731a66f88d..9d52aae0ff8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableRemove.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LookupTableRemove.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +41,7 @@ opType = LookupTableRemove.OP_NAME, inputsClass = LookupTableRemove.Inputs.class ) +@Operator public final class LookupTableRemove extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LowerBound.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LowerBound.java index 97069fccf75..8cf633e2d7f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LowerBound.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/LowerBound.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -57,6 +58,7 @@ opType = LowerBound.OP_NAME, inputsClass = LowerBound.Inputs.class ) +@Operator public final class LowerBound extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapDefun.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapDefun.java index c8f4bbef925..5c91ffea662 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapDefun.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MapDefun.java @@ -34,6 +34,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -53,6 +54,7 @@ opType = MapDefun.OP_NAME, inputsClass = MapDefun.Inputs.class ) +@Operator public final class MapDefun extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPadGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPadGrad.java index a879cf4c9f0..6e8c0d0680f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPadGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/MirrorPadGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -56,6 +57,7 @@ opType = MirrorPadGrad.OP_NAME, inputsClass = MirrorPadGrad.Inputs.class ) +@Operator public final class MirrorPadGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclAllReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclAllReduce.java index b92c4b3e9e6..d49045a1bad 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclAllReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclAllReduce.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -54,6 +55,7 @@ inputsClass = NcclAllReduce.Inputs.class ) @Deprecated +@Operator public final class NcclAllReduce extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclBroadcast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclBroadcast.java index 4c5fad98b84..4d5c2d771de 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclBroadcast.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclBroadcast.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -51,6 +52,7 @@ inputsClass = NcclBroadcast.Inputs.class ) @Deprecated +@Operator public final class NcclBroadcast extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclReduce.java index 61ef2825e7b..8b050aba7e3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/NcclReduce.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -51,6 +52,7 @@ inputsClass = NcclReduce.Inputs.class ) @Deprecated +@Operator public final class NcclReduce extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Recv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Recv.java index 9ce12da0383..1853328543d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Recv.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Recv.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,7 @@ opType = Recv.OP_NAME, inputsClass = Recv.Inputs.class ) +@Operator public final class Recv extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefEnter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefEnter.java index 74108c84f6a..888c0ee977b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefEnter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefEnter.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -45,6 +46,7 @@ opType = RefEnter.OP_NAME, inputsClass = RefEnter.Inputs.class ) +@Operator public final class RefEnter extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefExit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefExit.java index 8af577f4f19..c23ff2d03d7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefExit.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefExit.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,7 @@ opType = RefExit.OP_NAME, inputsClass = RefExit.Inputs.class ) +@Operator public final class RefExit extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefIdentity.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefIdentity.java index 5e699612efb..53d515be8e1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefIdentity.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefIdentity.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +42,7 @@ opType = RefIdentity.OP_NAME, inputsClass = RefIdentity.Inputs.class ) +@Operator public final class RefIdentity extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefMerge.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefMerge.java index d7b4026f5c5..9354cb2847b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefMerge.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/RefMerge.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -47,6 +48,7 @@ opType = RefMerge.OP_NAME, inputsClass = RefMerge.Inputs.class ) +@Operator public final class RefMerge extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMax.java index df88fe448ac..0adccafee2a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMax.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -42,6 +43,7 @@ opType = ScatterNdMax.OP_NAME, inputsClass = ScatterNdMax.Inputs.class ) +@Operator public final class ScatterNdMax extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMin.java index 4bf938febaf..d2780381fcb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/ScatterNdMin.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -42,6 +43,7 @@ opType = ScatterNdMin.OP_NAME, inputsClass = ScatterNdMin.Inputs.class ) +@Operator public final class ScatterNdMin extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Send.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Send.java index cc4effd6fe1..e722ace450a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Send.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/Send.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +39,7 @@ opType = Send.OP_NAME, inputsClass = Send.Inputs.class ) +@Operator public final class Send extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackClose.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackClose.java new file mode 100644 index 00000000000..810fb716a34 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackClose.java @@ -0,0 +1,83 @@ +/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +=======================================================================*/ + +// This class has been generated, DO NOT EDIT! + +package org.tensorflow.op.core; + +import java.util.Arrays; +import org.tensorflow.GraphOperation; +import org.tensorflow.Operand; +import org.tensorflow.Operation; +import org.tensorflow.OperationBuilder; +import org.tensorflow.op.RawOp; +import org.tensorflow.op.RawOpInputs; +import org.tensorflow.op.Scope; +import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; +import org.tensorflow.types.family.TType; + +/** + * Delete the stack from its resource container. + */ +@OpMetadata( + opType = StackClose.OP_NAME, + inputsClass = StackClose.Inputs.class +) +@Operator +public final class StackClose extends RawOp { + /** + * The name of this op, as known by TensorFlow core engine + */ + public static final String OP_NAME = "StackCloseV2"; + + public StackClose(Operation operation) { + super(operation, OP_NAME); + } + + /** + * Factory method to create a class wrapping a new StackCloseV2 operation. + * + * @param scope current scope + * @param handle The handle to a stack. + * @return a new instance of StackClose + */ + @Endpoint( + describeByClass = true + ) + public static StackClose create(Scope scope, Operand handle) { + OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "StackClose"); + opBuilder.addInput(handle.asOutput()); + return new StackClose(opBuilder.build()); + } + + @OpInputsMetadata( + outputsClass = StackClose.class + ) + public static class Inputs extends RawOpInputs { + /** + * The handle to a stack. + */ + public final Operand handle; + + public Inputs(GraphOperation op) { + super(new StackClose(op), op, Arrays.asList()); + int inputIndex = 0; + handle = (Operand) op.input(inputIndex++); + } + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackCreate.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackCreate.java new file mode 100644 index 00000000000..173c63d4c1e --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackCreate.java @@ -0,0 +1,167 @@ +/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +=======================================================================*/ + +// This class has been generated, DO NOT EDIT! + +package org.tensorflow.op.core; + +import java.util.Arrays; +import org.tensorflow.GraphOperation; +import org.tensorflow.Operand; +import org.tensorflow.Operation; +import org.tensorflow.OperationBuilder; +import org.tensorflow.Output; +import org.tensorflow.op.Operands; +import org.tensorflow.op.RawOp; +import org.tensorflow.op.RawOpInputs; +import org.tensorflow.op.Scope; +import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; +import org.tensorflow.proto.DataType; +import org.tensorflow.types.TInt32; +import org.tensorflow.types.family.TType; + +/** + * A stack that produces elements in first-in last-out order. + */ +@OpMetadata( + opType = StackCreate.OP_NAME, + inputsClass = StackCreate.Inputs.class +) +@Operator +public final class StackCreate extends RawOp implements Operand { + /** + * The name of this op, as known by TensorFlow core engine + */ + public static final String OP_NAME = "StackV2"; + + private Output handle; + + @SuppressWarnings("unchecked") + public StackCreate(Operation operation) { + super(operation, OP_NAME); + int outputIdx = 0; + handle = operation.output(outputIdx++); + } + + /** + * Factory method to create a class wrapping a new StackV2 operation. + * + * @param scope current scope + * @param maxSize The maximum size of the stack if non-negative. If negative, the stack + * size is unlimited. + * @param elemType The type of the elements on the stack. + * @param options carries optional attribute values + * @param data type for {@code StackV2} output and operands + * @return a new instance of StackCreate + */ + @Endpoint( + describeByClass = true + ) + public static StackCreate create(Scope scope, Operand maxSize, + Class elemType, Options... options) { + OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "StackCreate"); + opBuilder.addInput(maxSize.asOutput()); + opBuilder.setAttr("elem_type", Operands.toDataType(elemType)); + if (options != null) { + for (Options opts : options) { + if (opts.stackName != null) { + opBuilder.setAttr("stack_name", opts.stackName); + } + } + } + return new StackCreate(opBuilder.build()); + } + + /** + * Sets the stackName option. + * + * @param stackName Overrides the name used for the temporary stack resource. Default + * value is the name of the 'Stack' op (which is guaranteed unique). + * @return this Options instance. + */ + public static Options stackName(String stackName) { + return new Options().stackName(stackName); + } + + /** + * Gets handle. + * The handle to the stack. + * @return handle. + */ + public Output handle() { + return handle; + } + + @Override + @SuppressWarnings("unchecked") + public Output asOutput() { + return (Output) handle; + } + + /** + * Optional attributes for {@link org.tensorflow.op.core.StackCreate} + */ + public static class Options { + private String stackName; + + private Options() { + } + + /** + * Sets the stackName option. + * + * @param stackName Overrides the name used for the temporary stack resource. Default + * value is the name of the 'Stack' op (which is guaranteed unique). + * @return this Options instance. + */ + public Options stackName(String stackName) { + this.stackName = stackName; + return this; + } + } + + @OpInputsMetadata( + outputsClass = StackCreate.class + ) + public static class Inputs extends RawOpInputs { + /** + * The maximum size of the stack if non-negative. If negative, the stack + * size is unlimited. + */ + public final Operand maxSize; + + /** + * The type of the elements on the stack. + */ + public final DataType elemType; + + /** + * Overrides the name used for the temporary stack resource. Default + * value is the name of the 'Stack' op (which is guaranteed unique). + */ + public final String stackName; + + public Inputs(GraphOperation op) { + super(new StackCreate(op), op, Arrays.asList("elem_type", "stack_name")); + int inputIndex = 0; + maxSize = (Operand) op.input(inputIndex++); + elemType = op.attributes().getAttrType("elem_type"); + stackName = op.attributes().getAttrString("stack_name"); + } + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackPop.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackPop.java new file mode 100644 index 00000000000..a6a3021ce14 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackPop.java @@ -0,0 +1,116 @@ +/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +=======================================================================*/ + +// This class has been generated, DO NOT EDIT! + +package org.tensorflow.op.core; + +import java.util.Arrays; +import org.tensorflow.GraphOperation; +import org.tensorflow.Operand; +import org.tensorflow.Operation; +import org.tensorflow.OperationBuilder; +import org.tensorflow.Output; +import org.tensorflow.op.Operands; +import org.tensorflow.op.RawOp; +import org.tensorflow.op.RawOpInputs; +import org.tensorflow.op.Scope; +import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; +import org.tensorflow.proto.DataType; +import org.tensorflow.types.family.TType; + +/** + * Pop the element at the top of the stack. + * + * @param data type for {@code elem} output + */ +@OpMetadata( + opType = StackPop.OP_NAME, + inputsClass = StackPop.Inputs.class +) +@Operator +public final class StackPop extends RawOp implements Operand { + /** + * The name of this op, as known by TensorFlow core engine + */ + public static final String OP_NAME = "StackPopV2"; + + private Output elem; + + public StackPop(Operation operation) { + super(operation, OP_NAME); + int outputIdx = 0; + elem = operation.output(outputIdx++); + } + + /** + * Factory method to create a class wrapping a new StackPopV2 operation. + * + * @param scope current scope + * @param handle The handle to a stack. + * @param elemType The type of the elem that is popped. + * @param data type for {@code StackPopV2} output and operands + * @return a new instance of StackPop + */ + @Endpoint( + describeByClass = true + ) + public static StackPop create(Scope scope, Operand handle, + Class elemType) { + OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "StackPop"); + opBuilder.addInput(handle.asOutput()); + opBuilder.setAttr("elem_type", Operands.toDataType(elemType)); + return new StackPop<>(opBuilder.build()); + } + + /** + * Gets elem. + * The tensor that is popped from the top of the stack. + * @return elem. + */ + public Output elem() { + return elem; + } + + @Override + public Output asOutput() { + return elem; + } + + @OpInputsMetadata( + outputsClass = StackPop.class + ) + public static class Inputs extends RawOpInputs> { + /** + * The handle to a stack. + */ + public final Operand handle; + + /** + * The type of the elem that is popped. + */ + public final DataType elemType; + + public Inputs(GraphOperation op) { + super(new StackPop<>(op), op, Arrays.asList("elem_type")); + int inputIndex = 0; + handle = (Operand) op.input(inputIndex++); + elemType = op.attributes().getAttrType("elem_type"); + } + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackPush.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackPush.java new file mode 100644 index 00000000000..33a9ecf85a1 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StackPush.java @@ -0,0 +1,166 @@ +/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +=======================================================================*/ + +// This class has been generated, DO NOT EDIT! + +package org.tensorflow.op.core; + +import java.util.Arrays; +import org.tensorflow.GraphOperation; +import org.tensorflow.Operand; +import org.tensorflow.Operation; +import org.tensorflow.OperationBuilder; +import org.tensorflow.Output; +import org.tensorflow.op.RawOp; +import org.tensorflow.op.RawOpInputs; +import org.tensorflow.op.Scope; +import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; +import org.tensorflow.proto.DataType; +import org.tensorflow.types.family.TType; + +/** + * Push an element onto the stack. + * + * @param data type for {@code output} output + */ +@OpMetadata( + opType = StackPush.OP_NAME, + inputsClass = StackPush.Inputs.class +) +@Operator +public final class StackPush extends RawOp implements Operand { + /** + * The name of this op, as known by TensorFlow core engine + */ + public static final String OP_NAME = "StackPushV2"; + + private Output output; + + public StackPush(Operation operation) { + super(operation, OP_NAME); + int outputIdx = 0; + output = operation.output(outputIdx++); + } + + /** + * Factory method to create a class wrapping a new StackPushV2 operation. + * + * @param scope current scope + * @param handle The handle to a stack. + * @param elem The tensor to be pushed onto the stack. + * @param options carries optional attribute values + * @param data type for {@code StackPushV2} output and operands + * @return a new instance of StackPush + */ + @Endpoint( + describeByClass = true + ) + public static StackPush create(Scope scope, Operand handle, + Operand elem, Options... options) { + OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "StackPush"); + opBuilder.addInput(handle.asOutput()); + opBuilder.addInput(elem.asOutput()); + if (options != null) { + for (Options opts : options) { + if (opts.swapMemory != null) { + opBuilder.setAttr("swap_memory", opts.swapMemory); + } + } + } + return new StackPush<>(opBuilder.build()); + } + + /** + * Sets the swapMemory option. + * + * @param swapMemory Swap {@code elem} to CPU. Default to false. + * @return this Options instance. + */ + public static Options swapMemory(Boolean swapMemory) { + return new Options().swapMemory(swapMemory); + } + + /** + * Gets output. + * The same tensor as the input 'elem'. + * @return output. + */ + public Output output() { + return output; + } + + @Override + public Output asOutput() { + return output; + } + + /** + * Optional attributes for {@link org.tensorflow.op.core.StackPush} + */ + public static class Options { + private Boolean swapMemory; + + private Options() { + } + + /** + * Sets the swapMemory option. + * + * @param swapMemory Swap {@code elem} to CPU. Default to false. + * @return this Options instance. + */ + public Options swapMemory(Boolean swapMemory) { + this.swapMemory = swapMemory; + return this; + } + } + + @OpInputsMetadata( + outputsClass = StackPush.class + ) + public static class Inputs extends RawOpInputs> { + /** + * The handle to a stack. + */ + public final Operand handle; + + /** + * The tensor to be pushed onto the stack. + */ + public final Operand elem; + + /** + * The T attribute + */ + public final DataType T; + + /** + * Swap `elem` to CPU. Default to false. + */ + public final boolean swapMemory; + + public Inputs(GraphOperation op) { + super(new StackPush<>(op), op, Arrays.asList("T", "swap_memory")); + int inputIndex = 0; + handle = (Operand) op.input(inputIndex++); + elem = (Operand) op.input(inputIndex++); + T = op.attributes().getAttrType("T"); + swapMemory = op.attributes().getAttrBool("swap_memory"); + } + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessCase.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessCase.java index d314d761266..915d6908936 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessCase.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StatelessCase.java @@ -34,6 +34,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -65,6 +66,7 @@ opType = StatelessCase.OP_NAME, inputsClass = StatelessCase.Inputs.class ) +@Operator public final class StatelessCase extends RawOp implements Case { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StochasticCastToInt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StochasticCastToInt.java index a4d7a203402..29da2cb9a53 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StochasticCastToInt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/StochasticCastToInt.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -46,6 +47,7 @@ opType = StochasticCastToInt.OP_NAME, inputsClass = StochasticCastToInt.Inputs.class ) +@Operator public final class StochasticCastToInt extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SyncDevice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SyncDevice.java index 357e65e6b2b..a85fd9312c5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SyncDevice.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/SyncDevice.java @@ -27,6 +27,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; /** * Synchronizes the device this op is run on. @@ -37,6 +38,7 @@ opType = SyncDevice.OP_NAME, inputsClass = SyncDevice.Inputs.class ) +@Operator public final class SyncDevice extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UniformQuantizedClipByValue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UniformQuantizedClipByValue.java index f96475d3bf8..ca3c5dfdd14 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UniformQuantizedClipByValue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UniformQuantizedClipByValue.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -46,6 +47,7 @@ opType = UniformQuantizedClipByValue.OP_NAME, inputsClass = UniformQuantizedClipByValue.Inputs.class ) +@Operator public final class UniformQuantizedClipByValue extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UpperBound.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UpperBound.java index b50ba0d9399..d5e939ffde6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UpperBound.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/core/UpperBound.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -57,6 +58,7 @@ opType = UpperBound.OP_NAME, inputsClass = UpperBound.Inputs.class ) +@Operator public final class UpperBound extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMemoryCache.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMemoryCache.java index ec5f54534a0..7554d2de110 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMemoryCache.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AnonymousMemoryCache.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -37,6 +38,9 @@ opType = AnonymousMemoryCache.OP_NAME, inputsClass = AnonymousMemoryCache.Inputs.class ) +@Operator( + group = "data" +) public final class AnonymousMemoryCache extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertPrevDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertPrevDataset.java index abd5286f98b..bab20bf1f31 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertPrevDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/AssertPrevDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -50,6 +51,9 @@ opType = AssertPrevDataset.OP_NAME, inputsClass = AssertPrevDataset.Inputs.class ) +@Operator( + group = "data" +) public final class AssertPrevDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CompressElement.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CompressElement.java index 996fd0328f1..b070f5995d4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CompressElement.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/CompressElement.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +41,9 @@ opType = CompressElement.OP_NAME, inputsClass = CompressElement.Inputs.class ) +@Operator( + group = "data" +) public final class CompressElement extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMemoryCache.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMemoryCache.java index 67486cf98bf..a701e8ab58b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMemoryCache.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMemoryCache.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -37,6 +38,9 @@ opType = DeleteMemoryCache.OP_NAME, inputsClass = DeleteMemoryCache.Inputs.class ) +@Operator( + group = "data" +) public final class DeleteMemoryCache extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMultiDeviceIterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMultiDeviceIterator.java index 47d761f78d3..8f8b0b35d33 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMultiDeviceIterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DeleteMultiDeviceIterator.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -38,6 +39,9 @@ opType = DeleteMultiDeviceIterator.OP_NAME, inputsClass = DeleteMultiDeviceIterator.Inputs.class ) +@Operator( + group = "data" +) public final class DeleteMultiDeviceIterator extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DummyIterationCounter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DummyIterationCounter.java index af321f086d8..be7fc1c6ee8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DummyIterationCounter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/DummyIterationCounter.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -38,6 +39,9 @@ opType = DummyIterationCounter.OP_NAME, inputsClass = DummyIterationCounter.Inputs.class ) +@Operator( + group = "data" +) public final class DummyIterationCounter extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorFromStringHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorFromStringHandle.java index 1d11ba1e78f..e8504283d32 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorFromStringHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorFromStringHandle.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = IteratorFromStringHandle.OP_NAME, inputsClass = IteratorFromStringHandle.Inputs.class ) +@Operator( + group = "data" +) public final class IteratorFromStringHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetDevice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetDevice.java index 8e9e0992a17..a30963f9e6d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetDevice.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/IteratorGetDevice.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = IteratorGetDevice.OP_NAME, inputsClass = IteratorGetDevice.Inputs.class ) +@Operator( + group = "data" +) public final class IteratorGetDevice extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LeakyReluGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LeakyReluGrad.java index e7a66350f0c..a42cc0f51d2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LeakyReluGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/LeakyReluGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = LeakyReluGrad.OP_NAME, inputsClass = LeakyReluGrad.Inputs.class ) +@Operator( + group = "data" +) public final class LeakyReluGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ListDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ListDataset.java index 45b9d31c0b1..76db7fe0eac 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ListDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ListDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = ListDataset.OP_NAME, inputsClass = ListDataset.Inputs.class ) +@Operator( + group = "data" +) public final class ListDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIterator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIterator.java index 7fe8aa712a0..467acc47e4d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIterator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIterator.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = MultiDeviceIterator.OP_NAME, inputsClass = MultiDeviceIterator.Inputs.class ) +@Operator( + group = "data" +) public final class MultiDeviceIterator extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorFromStringHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorFromStringHandle.java index b9963f2cbfe..20cfc4f1010 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorFromStringHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorFromStringHandle.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = MultiDeviceIteratorFromStringHandle.OP_NAME, inputsClass = MultiDeviceIteratorFromStringHandle.Inputs.class ) +@Operator( + group = "data" +) public final class MultiDeviceIteratorFromStringHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorGetNextFromShard.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorGetNextFromShard.java index 57410d3f0f1..642935599c7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorGetNextFromShard.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorGetNextFromShard.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -45,6 +46,9 @@ opType = MultiDeviceIteratorGetNextFromShard.OP_NAME, inputsClass = MultiDeviceIteratorGetNextFromShard.Inputs.class ) +@Operator( + group = "data" +) public final class MultiDeviceIteratorGetNextFromShard extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorInit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorInit.java index f1866ce63da..8cbc7b35693 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorInit.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorInit.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = MultiDeviceIteratorInit.OP_NAME, inputsClass = MultiDeviceIteratorInit.Inputs.class ) +@Operator( + group = "data" +) public final class MultiDeviceIteratorInit extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorToStringHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorToStringHandle.java index 5eebe6521f7..cc856b4ecc9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorToStringHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/MultiDeviceIteratorToStringHandle.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = MultiDeviceIteratorToStringHandle.OP_NAME, inputsClass = MultiDeviceIteratorToStringHandle.Inputs.class ) +@Operator( + group = "data" +) public final class MultiDeviceIteratorToStringHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelFilterDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelFilterDataset.java index 9d8b307f2ff..5dece763295 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelFilterDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ParallelFilterDataset.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -52,6 +53,9 @@ opType = ParallelFilterDataset.OP_NAME, inputsClass = ParallelFilterDataset.Inputs.class ) +@Operator( + group = "data" +) public final class ParallelFilterDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RewriteDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RewriteDataset.java index 9e0c1449324..d67c7d6e808 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RewriteDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/RewriteDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = RewriteDataset.OP_NAME, inputsClass = RewriteDataset.Inputs.class ) +@Operator( + group = "data" +) public final class RewriteDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotChunkDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotChunkDataset.java index 71c3718f840..df6172f9075 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotChunkDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotChunkDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = SnapshotChunkDataset.OP_NAME, inputsClass = SnapshotChunkDataset.Inputs.class ) +@Operator( + group = "data" +) public final class SnapshotChunkDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDatasetReader.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDatasetReader.java index 45b14fb5995..8a20a594bcf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDatasetReader.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotDatasetReader.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -44,6 +45,9 @@ opType = SnapshotDatasetReader.OP_NAME, inputsClass = SnapshotDatasetReader.Inputs.class ) +@Operator( + group = "data" +) public final class SnapshotDatasetReader extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotNestedDatasetReader.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotNestedDatasetReader.java index 7c288397398..3d60a2cc237 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotNestedDatasetReader.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/SnapshotNestedDatasetReader.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = SnapshotNestedDatasetReader.OP_NAME, inputsClass = SnapshotNestedDatasetReader.Inputs.class ) +@Operator( + group = "data" +) public final class SnapshotNestedDatasetReader extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorHandle.java index 33d218181e5..080585d34a6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorHandle.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -38,6 +39,9 @@ opType = StatsAggregatorHandle.OP_NAME, inputsClass = StatsAggregatorHandle.Inputs.class ) +@Operator( + group = "data" +) public final class StatsAggregatorHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorSetSummaryWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorSetSummaryWriter.java index c3310008cdd..c610409f62a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorSetSummaryWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/StatsAggregatorSetSummaryWriter.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -37,6 +38,9 @@ opType = StatsAggregatorSetSummaryWriter.OP_NAME, inputsClass = StatsAggregatorSetSummaryWriter.Inputs.class ) +@Operator( + group = "data" +) public final class StatsAggregatorSetSummaryWriter extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolHandle.java index 6ae9cfdc470..8e110f97a30 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/ThreadPoolHandle.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -38,6 +39,9 @@ opType = ThreadPoolHandle.OP_NAME, inputsClass = ThreadPoolHandle.Inputs.class ) +@Operator( + group = "data" +) public final class ThreadPoolHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UncompressElement.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UncompressElement.java index 22754dce45d..9c871ae7b08 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UncompressElement.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/UncompressElement.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = UncompressElement.OP_NAME, inputsClass = UncompressElement.Inputs.class ) +@Operator( + group = "data" +) public final class UncompressElement extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WindowOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WindowOp.java index a1cfcaf853e..74097c53e64 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WindowOp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/WindowOp.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = WindowOp.OP_NAME, inputsClass = WindowOp.Inputs.class ) +@Operator( + group = "data" +) public final class WindowOp extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AssertNextDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AssertNextDataset.java index f6f252f3ad5..92f21d06c07 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AssertNextDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AssertNextDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = AssertNextDataset.OP_NAME, inputsClass = AssertNextDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class AssertNextDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AutoShardDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AutoShardDataset.java index 9e8615898f6..8c22e6fa9e1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AutoShardDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/AutoShardDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -49,6 +50,9 @@ opType = AutoShardDataset.OP_NAME, inputsClass = AutoShardDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class AutoShardDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/BytesProducedStatsDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/BytesProducedStatsDataset.java index 92f1428e199..d98279464ed 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/BytesProducedStatsDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/BytesProducedStatsDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = BytesProducedStatsDataset.OP_NAME, inputsClass = BytesProducedStatsDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class BytesProducedStatsDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/CSVDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/CSVDataset.java index ad981a6f0c9..43e1d9d58a4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/CSVDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/CSVDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TBool; import org.tensorflow.types.TInt64; @@ -45,6 +46,9 @@ opType = CSVDataset.OP_NAME, inputsClass = CSVDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class CSVDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ChooseFastestDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ChooseFastestDataset.java index 640209ea50f..cab5575a928 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ChooseFastestDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ChooseFastestDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = ChooseFastestDataset.OP_NAME, inputsClass = ChooseFastestDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class ChooseFastestDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetCardinality.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetCardinality.java index 49f100d73a6..d73785b9a91 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetCardinality.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetCardinality.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -40,6 +41,9 @@ opType = DatasetCardinality.OP_NAME, inputsClass = DatasetCardinality.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class DatasetCardinality extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetToTFRecord.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetToTFRecord.java index 27e393a66a1..35889dec71c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetToTFRecord.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DatasetToTFRecord.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -38,6 +39,9 @@ opType = DatasetToTFRecord.OP_NAME, inputsClass = DatasetToTFRecord.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class DatasetToTFRecord extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DenseToSparseBatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DenseToSparseBatchDataset.java index 59992f2522c..414bb684323 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DenseToSparseBatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DenseToSparseBatchDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = DenseToSparseBatchDataset.OP_NAME, inputsClass = DenseToSparseBatchDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class DenseToSparseBatchDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DirectedInterleaveDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DirectedInterleaveDataset.java index 04a72437d9b..6d4464a7746 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DirectedInterleaveDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/DirectedInterleaveDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = DirectedInterleaveDataset.OP_NAME, inputsClass = DirectedInterleaveDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class DirectedInterleaveDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByReducerDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByReducerDataset.java index e56e2dbc49e..db37b2fc4e8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByReducerDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByReducerDataset.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +45,9 @@ opType = GroupByReducerDataset.OP_NAME, inputsClass = GroupByReducerDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class GroupByReducerDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByWindowDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByWindowDataset.java index 33db47df4f6..a24b41f050d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByWindowDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/GroupByWindowDataset.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +45,9 @@ opType = GroupByWindowDataset.OP_NAME, inputsClass = GroupByWindowDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class GroupByWindowDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IgnoreErrorsDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IgnoreErrorsDataset.java index def6476c057..26f1eabead2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IgnoreErrorsDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IgnoreErrorsDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = IgnoreErrorsDataset.OP_NAME, inputsClass = IgnoreErrorsDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class IgnoreErrorsDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IteratorGetDevice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IteratorGetDevice.java index 6f6d6f70c30..b7dde3d078e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IteratorGetDevice.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/IteratorGetDevice.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = IteratorGetDevice.OP_NAME, inputsClass = IteratorGetDevice.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class IteratorGetDevice extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LatencyStatsDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LatencyStatsDataset.java index bbb6ef677c9..582fde7e038 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LatencyStatsDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LatencyStatsDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = LatencyStatsDataset.OP_NAME, inputsClass = LatencyStatsDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class LatencyStatsDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LmdbDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LmdbDataset.java index 2d29444b0cc..5bbb82db008 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LmdbDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/LmdbDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = LmdbDataset.OP_NAME, inputsClass = LmdbDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class LmdbDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapAndBatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapAndBatchDataset.java index ced18243fe8..18ac4e051f9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapAndBatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapAndBatchDataset.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TBool; import org.tensorflow.types.TInt64; @@ -49,6 +50,9 @@ opType = MapAndBatchDataset.OP_NAME, inputsClass = MapAndBatchDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class MapAndBatchDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapDataset.java index 8ce3160e9b1..f02bba6e46a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MapDataset.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = MapDataset.OP_NAME, inputsClass = MapDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class MapDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MatchingFilesDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MatchingFilesDataset.java index ccdce76b4d2..37486a2a506 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MatchingFilesDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MatchingFilesDataset.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = MatchingFilesDataset.OP_NAME, inputsClass = MatchingFilesDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class MatchingFilesDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MaxIntraOpParallelismDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MaxIntraOpParallelismDataset.java index 87b17b10b50..e8d73813cbd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MaxIntraOpParallelismDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/MaxIntraOpParallelismDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = MaxIntraOpParallelismDataset.OP_NAME, inputsClass = MaxIntraOpParallelismDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class MaxIntraOpParallelismDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/NonSerializableDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/NonSerializableDataset.java index 37a41cce38f..fcfeda256ec 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/NonSerializableDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/NonSerializableDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = NonSerializableDataset.OP_NAME, inputsClass = NonSerializableDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class NonSerializableDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParallelInterleaveDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParallelInterleaveDataset.java index fc7905dd5e9..df03bf4d26a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParallelInterleaveDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParallelInterleaveDataset.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TBool; import org.tensorflow.types.TInt64; @@ -51,6 +52,9 @@ opType = ParallelInterleaveDataset.OP_NAME, inputsClass = ParallelInterleaveDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class ParallelInterleaveDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParseExampleDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParseExampleDataset.java index 5bbc30a8fe2..9ba31aa0995 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParseExampleDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ParseExampleDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = ParseExampleDataset.OP_NAME, inputsClass = ParseExampleDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class ParseExampleDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/PrivateThreadPoolDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/PrivateThreadPoolDataset.java index 3b6f70a5f6d..f555de177d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/PrivateThreadPoolDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/PrivateThreadPoolDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = PrivateThreadPoolDataset.OP_NAME, inputsClass = PrivateThreadPoolDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class PrivateThreadPoolDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RandomDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RandomDataset.java index 0e0b1581f7f..ca62de342d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RandomDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RandomDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = RandomDataset.OP_NAME, inputsClass = RandomDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class RandomDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RebatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RebatchDataset.java index d837c94a029..f83fa7d5bc6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RebatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/RebatchDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -45,6 +46,9 @@ opType = RebatchDataset.OP_NAME, inputsClass = RebatchDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class RebatchDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ScanDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ScanDataset.java index 77e67422992..782889a68ba 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ScanDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ScanDataset.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = ScanDataset.OP_NAME, inputsClass = ScanDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class ScanDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SetStatsAggregatorDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SetStatsAggregatorDataset.java index b661b04b49b..1d083fa51b2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SetStatsAggregatorDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SetStatsAggregatorDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = SetStatsAggregatorDataset.OP_NAME, inputsClass = SetStatsAggregatorDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class SetStatsAggregatorDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SleepDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SleepDataset.java index 4888318d074..8b494e8d2bd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SleepDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SleepDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = SleepDataset.OP_NAME, inputsClass = SleepDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class SleepDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SlidingWindowDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SlidingWindowDataset.java index 926a05dd1e6..71063d28d11 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SlidingWindowDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SlidingWindowDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = SlidingWindowDataset.OP_NAME, inputsClass = SlidingWindowDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class SlidingWindowDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SqlDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SqlDataset.java index 0fd9d24ac41..906ad5aeed0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SqlDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/SqlDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = SqlDataset.OP_NAME, inputsClass = SqlDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class SqlDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorHandle.java index 3e7e84f0ad3..23dea9a06da 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorHandle.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -38,6 +39,9 @@ opType = StatsAggregatorHandle.OP_NAME, inputsClass = StatsAggregatorHandle.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class StatsAggregatorHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorSummary.java index 66340c83c40..ee33f0944f5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/StatsAggregatorSummary.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = StatsAggregatorSummary.OP_NAME, inputsClass = StatsAggregatorSummary.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class StatsAggregatorSummary extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/TakeWhileDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/TakeWhileDataset.java index db4b8d19649..2d3097055fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/TakeWhileDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/TakeWhileDataset.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -49,6 +50,9 @@ opType = TakeWhileDataset.OP_NAME, inputsClass = TakeWhileDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class TakeWhileDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolDataset.java index 6e9999b4216..5ceeb9f1745 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = ThreadPoolDataset.OP_NAME, inputsClass = ThreadPoolDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class ThreadPoolDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolHandle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolHandle.java index 767c26aabfd..edcfe5cd3b8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolHandle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/ThreadPoolHandle.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -38,6 +39,9 @@ opType = ThreadPoolHandle.OP_NAME, inputsClass = ThreadPoolHandle.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class ThreadPoolHandle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UnbatchDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UnbatchDataset.java index e0d19f90c9d..6f207dabb59 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UnbatchDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UnbatchDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = UnbatchDataset.OP_NAME, inputsClass = UnbatchDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class UnbatchDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UniqueDataset.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UniqueDataset.java index ad60b1fade5..9ac956cc829 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UniqueDataset.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/data/experimental/UniqueDataset.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = UniqueDataset.OP_NAME, inputsClass = UniqueDataset.Inputs.class ) +@Operator( + group = "data.experimental" +) public final class UniqueDataset extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/CheckNumerics.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/CheckNumerics.java index 2067d99bf77..d1aae3e74ad 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/CheckNumerics.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/debugging/CheckNumerics.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -45,6 +46,9 @@ opType = CheckNumerics.OP_NAME, inputsClass = CheckNumerics.Inputs.class ) +@Operator( + group = "debugging" +) public final class CheckNumerics extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclAllReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclAllReduce.java index 6f804d8de1d..c5416746198 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclAllReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclAllReduce.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -51,6 +52,9 @@ opType = NcclAllReduce.OP_NAME, inputsClass = NcclAllReduce.Inputs.class ) +@Operator( + group = "distribute" +) public final class NcclAllReduce extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclBroadcast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclBroadcast.java index fe88425d478..3824d6a10dd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclBroadcast.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclBroadcast.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -48,6 +49,9 @@ opType = NcclBroadcast.OP_NAME, inputsClass = NcclBroadcast.Inputs.class ) +@Operator( + group = "distribute" +) public final class NcclBroadcast extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclReduce.java index 76d50c1ce3e..2a80593be6c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclReduce.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/distribute/NcclReduce.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -48,6 +49,9 @@ opType = NcclReduce.OP_NAME, inputsClass = NcclReduce.Inputs.class ) +@Operator( + group = "distribute" +) public final class NcclReduce extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/ToBool.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/ToBool.java index 48f9f4be62f..0db30e9ca62 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/ToBool.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/dtypes/ToBool.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TBool; import org.tensorflow.types.family.TType; @@ -55,6 +56,9 @@ opType = ToBool.OP_NAME, inputsClass = ToBool.Inputs.class ) +@Operator( + group = "dtypes" +) public final class ToBool extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesAggregateStats.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesAggregateStats.java deleted file mode 100644 index 4fa32ac6dd7..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesAggregateStats.java +++ /dev/null @@ -1,145 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; - -/** - * Aggregates the summary of accumulated stats for the batch. - * The summary stats contains gradients and hessians accumulated for each node, feature dimension id and bucket. - */ -@OpMetadata( - opType = BoostedTreesAggregateStats.OP_NAME, - inputsClass = BoostedTreesAggregateStats.Inputs.class -) -public final class BoostedTreesAggregateStats extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesAggregateStats"; - - private Output statsSummary; - - public BoostedTreesAggregateStats(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - statsSummary = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesAggregateStats operation. - * - * @param scope current scope - * @param nodeIds int32; Rank 1 Tensor containing node ids for each example, shape [batch_size]. - * @param gradients float32; Rank 2 Tensor (shape=[batch_size, logits_dimension]) with gradients for each example. - * @param hessians float32; Rank 2 Tensor (shape=[batch_size, hessian_dimension]) with hessians for each example. - * @param feature int32; Rank 2 feature Tensors (shape=[batch_size, feature_dimension]). - * @param maxSplits int; the maximum number of splits possible in the whole tree. - * @param numBuckets int; equals to the maximum possible value of bucketized feature. - * @return a new instance of BoostedTreesAggregateStats - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesAggregateStats create(Scope scope, Operand nodeIds, - Operand gradients, Operand hessians, Operand feature, - Long maxSplits, Long numBuckets) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesAggregateStats"); - opBuilder.addInput(nodeIds.asOutput()); - opBuilder.addInput(gradients.asOutput()); - opBuilder.addInput(hessians.asOutput()); - opBuilder.addInput(feature.asOutput()); - opBuilder.setAttr("max_splits", maxSplits); - opBuilder.setAttr("num_buckets", numBuckets); - return new BoostedTreesAggregateStats(opBuilder.build()); - } - - /** - * Gets statsSummary. - * output Rank 4 Tensor (shape=[splits, feature_dimension, buckets, logits_dimension + hessian_dimension]) - * containing accumulated stats for each node, feature dimension and bucket. - * @return statsSummary. - */ - public Output statsSummary() { - return statsSummary; - } - - @Override - public Output asOutput() { - return statsSummary; - } - - @OpInputsMetadata( - outputsClass = BoostedTreesAggregateStats.class - ) - public static class Inputs extends RawOpInputs { - /** - * int32; Rank 1 Tensor containing node ids for each example, shape [batch_size]. - */ - public final Operand nodeIds; - - /** - * float32; Rank 2 Tensor (shape=[batch_size, logits_dimension]) with gradients for each example. - */ - public final Operand gradients; - - /** - * float32; Rank 2 Tensor (shape=[batch_size, hessian_dimension]) with hessians for each example. - */ - public final Operand hessians; - - /** - * int32; Rank 2 feature Tensors (shape=[batch_size, feature_dimension]). - */ - public final Operand feature; - - /** - * int; the maximum number of splits possible in the whole tree. - */ - public final long maxSplits; - - /** - * int; equals to the maximum possible value of bucketized feature. - */ - public final long numBuckets; - - public Inputs(GraphOperation op) { - super(new BoostedTreesAggregateStats(op), op, Arrays.asList("max_splits", "num_buckets")); - int inputIndex = 0; - nodeIds = (Operand) op.input(inputIndex++); - gradients = (Operand) op.input(inputIndex++); - hessians = (Operand) op.input(inputIndex++); - feature = (Operand) op.input(inputIndex++); - maxSplits = op.attributes().getAttrInt("max_splits"); - numBuckets = op.attributes().getAttrInt("num_buckets"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesBucketize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesBucketize.java deleted file mode 100644 index ee2aecbace0..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesBucketize.java +++ /dev/null @@ -1,125 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import java.util.Iterator; -import java.util.List; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; - -/** - * Bucketize each feature based on bucket boundaries. - * An op that returns a list of float tensors, where each tensor represents the - * bucketized values for a single feature. - */ -@OpMetadata( - opType = BoostedTreesBucketize.OP_NAME, - inputsClass = BoostedTreesBucketize.Inputs.class -) -public final class BoostedTreesBucketize extends RawOp implements Iterable> { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesBucketize"; - - private List> buckets; - - @SuppressWarnings("unchecked") - public BoostedTreesBucketize(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - int bucketsLength = operation.outputListLength("buckets"); - buckets = Arrays.asList((Output[]) operation.outputList(outputIdx, bucketsLength)); - outputIdx += bucketsLength; - } - - /** - * Factory method to create a class wrapping a new BoostedTreesBucketize operation. - * - * @param scope current scope - * @param floatValues float; List of Rank 1 Tensor each containing float values for a single feature. - * @param bucketBoundaries float; List of Rank 1 Tensors each containing the bucket boundaries for a single - * feature. - * @return a new instance of BoostedTreesBucketize - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesBucketize create(Scope scope, Iterable> floatValues, - Iterable> bucketBoundaries) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesBucketize"); - opBuilder.addInputList(Operands.asOutputs(floatValues)); - opBuilder.addInputList(Operands.asOutputs(bucketBoundaries)); - return new BoostedTreesBucketize(opBuilder.build()); - } - - /** - * Gets buckets. - * int; List of Rank 1 Tensors each containing the bucketized values for a single feature. - * @return buckets. - */ - public List> buckets() { - return buckets; - } - - @Override - @SuppressWarnings({"rawtypes", "unchecked"}) - public Iterator> iterator() { - return (Iterator) buckets.iterator(); - } - - @OpInputsMetadata( - outputsClass = BoostedTreesBucketize.class - ) - public static class Inputs extends RawOpInputs { - /** - * float; List of Rank 1 Tensor each containing float values for a single feature. - */ - public final Iterable> floatValues; - - /** - * float; List of Rank 1 Tensors each containing the bucket boundaries for a single - * feature. - */ - public final Iterable> bucketBoundaries; - - public Inputs(GraphOperation op) { - super(new BoostedTreesBucketize(op), op, Arrays.asList()); - int inputIndex = 0; - int floatValuesLength = op.inputListLength("float_values"); - floatValues = Arrays.asList((Operand[]) op.inputList(inputIndex, floatValuesLength)); - inputIndex += floatValuesLength; - int bucketBoundariesLength = op.inputListLength("bucket_boundaries"); - bucketBoundaries = Arrays.asList((Operand[]) op.inputList(inputIndex, bucketBoundariesLength)); - inputIndex += bucketBoundariesLength; - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplit.java deleted file mode 100644 index 7d988e1935e..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplit.java +++ /dev/null @@ -1,272 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; -import org.tensorflow.types.TString; - -/** - * Calculates gains for each feature and returns the best possible split information for the feature. - * The split information is the best threshold (bucket id), gains and left/right node contributions per node for each feature. - *

It is possible that not all nodes can be split on each feature. Hence, the list of possible nodes can differ between the features. Therefore, we return {@code node_ids_list} for each feature, containing the list of nodes that this feature can be used to split. - *

In this manner, the output is the best split per features and per node, so that it needs to be combined later to produce the best split for each node (among all possible features). - *

The output shapes are compatible in a way that the first dimension of all tensors are the same and equal to the number of possible split nodes for each feature. - */ -@OpMetadata( - opType = BoostedTreesCalculateBestFeatureSplit.OP_NAME, - inputsClass = BoostedTreesCalculateBestFeatureSplit.Inputs.class -) -public final class BoostedTreesCalculateBestFeatureSplit extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesCalculateBestFeatureSplit"; - - private Output nodeIds; - - private Output gains; - - private Output featureDimensions; - - private Output thresholds; - - private Output leftNodeContribs; - - private Output rightNodeContribs; - - private Output splitWithDefaultDirections; - - public BoostedTreesCalculateBestFeatureSplit(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - nodeIds = operation.output(outputIdx++); - gains = operation.output(outputIdx++); - featureDimensions = operation.output(outputIdx++); - thresholds = operation.output(outputIdx++); - leftNodeContribs = operation.output(outputIdx++); - rightNodeContribs = operation.output(outputIdx++); - splitWithDefaultDirections = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesCalculateBestFeatureSplit operation. - * - * @param scope current scope - * @param nodeIdRange A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). - * @param statsSummary A Rank 4 tensor (#shape=[max_splits, feature_dims, bucket, stats_dims]) for accumulated stats summary (gradient/hessian) per node, per dimension, per buckets for each feature. - * The first dimension of the tensor is the maximum number of splits, and thus not all elements of it will be used, but only the indexes specified by node_ids will be used. - * @param l1 l1 regularization factor on leaf weights, per instance based. - * @param l2 l2 regularization factor on leaf weights, per instance based. - * @param treeComplexity adjustment to the gain, per leaf based. - * @param minNodeWeight minimum avg of hessians in a node before required for the node to be considered for splitting. - * @param logitsDimension The dimension of logit, i.e., number of classes. - * @param options carries optional attribute values - * @return a new instance of BoostedTreesCalculateBestFeatureSplit - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesCalculateBestFeatureSplit create(Scope scope, - Operand nodeIdRange, Operand statsSummary, Operand l1, - Operand l2, Operand treeComplexity, Operand minNodeWeight, - Long logitsDimension, Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesCalculateBestFeatureSplit"); - opBuilder.addInput(nodeIdRange.asOutput()); - opBuilder.addInput(statsSummary.asOutput()); - opBuilder.addInput(l1.asOutput()); - opBuilder.addInput(l2.asOutput()); - opBuilder.addInput(treeComplexity.asOutput()); - opBuilder.addInput(minNodeWeight.asOutput()); - opBuilder.setAttr("logits_dimension", logitsDimension); - if (options != null) { - for (Options opts : options) { - if (opts.splitType != null) { - opBuilder.setAttr("split_type", opts.splitType); - } - } - } - return new BoostedTreesCalculateBestFeatureSplit(opBuilder.build()); - } - - /** - * Sets the splitType option. - * - * @param splitType A string indicating if this Op should perform inequality split or equality split. - * @return this Options instance. - */ - public static Options splitType(String splitType) { - return new Options().splitType(splitType); - } - - /** - * Gets nodeIds. - * A Rank 1 tensors indicating possible split node ids for each feature. The length of the list is num_features, but each tensor has different size as each feature provides different possible nodes. See above for details like shapes and sizes. - * @return nodeIds. - */ - public Output nodeIds() { - return nodeIds; - } - - /** - * Gets gains. - * A Rank 1 tensors indicating the best gains for each feature to split for certain nodes. See above for details like shapes and sizes. - * @return gains. - */ - public Output gains() { - return gains; - } - - /** - * Gets featureDimensions. - * A Rank 1 tensors indicating the best feature dimension for each feature to split for certain nodes if the feature is multi-dimension. See above for details like shapes and sizes. - * @return featureDimensions. - */ - public Output featureDimensions() { - return featureDimensions; - } - - /** - * Gets thresholds. - * A Rank 1 tensors indicating the bucket id to compare with (as a threshold) for split in each node. See above for details like shapes and sizes. - * @return thresholds. - */ - public Output thresholds() { - return thresholds; - } - - /** - * Gets leftNodeContribs. - * A Rank 2 tensors indicating the contribution of the left nodes when branching from parent nodes (given by the tensor element in the output node_ids_list) to the left direction by the given threshold for each feature. This value will be used to make the left node value by adding to the parent node value. Second dimension size is 1 for 1-dimensional logits, but would be larger for multi-class problems. See above for details like shapes and sizes. - * @return leftNodeContribs. - */ - public Output leftNodeContribs() { - return leftNodeContribs; - } - - /** - * Gets rightNodeContribs. - * A Rank 2 tensors, with the same shape/conditions as left_node_contribs_list, but just that the value is for the right node. - * @return rightNodeContribs. - */ - public Output rightNodeContribs() { - return rightNodeContribs; - } - - /** - * Gets splitWithDefaultDirections. - * A Rank 1 tensors indicating the which direction to go if data is missing. See above for details like shapes and sizes. - * Inequality with default left returns 0, inequality with default right returns 1, equality with default right returns 2. - * @return splitWithDefaultDirections. - */ - public Output splitWithDefaultDirections() { - return splitWithDefaultDirections; - } - - /** - * Optional attributes for {@link org.tensorflow.op.estimator.BoostedTreesCalculateBestFeatureSplit} - */ - public static class Options { - private String splitType; - - private Options() { - } - - /** - * Sets the splitType option. - * - * @param splitType A string indicating if this Op should perform inequality split or equality split. - * @return this Options instance. - */ - public Options splitType(String splitType) { - this.splitType = splitType; - return this; - } - } - - @OpInputsMetadata( - outputsClass = BoostedTreesCalculateBestFeatureSplit.class - ) - public static class Inputs extends RawOpInputs { - /** - * A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). - */ - public final Operand nodeIdRange; - - /** - * A Rank 4 tensor (#shape=[max_splits, feature_dims, bucket, stats_dims]) for accumulated stats summary (gradient/hessian) per node, per dimension, per buckets for each feature. - * The first dimension of the tensor is the maximum number of splits, and thus not all elements of it will be used, but only the indexes specified by node_ids will be used. - */ - public final Operand statsSummary; - - /** - * l1 regularization factor on leaf weights, per instance based. - */ - public final Operand l1; - - /** - * l2 regularization factor on leaf weights, per instance based. - */ - public final Operand l2; - - /** - * adjustment to the gain, per leaf based. - */ - public final Operand treeComplexity; - - /** - * minimum avg of hessians in a node before required for the node to be considered for splitting. - */ - public final Operand minNodeWeight; - - /** - * The dimension of logit, i.e., number of classes. - */ - public final long logitsDimension; - - /** - * A string indicating if this Op should perform inequality split or equality split. - */ - public final String splitType; - - public Inputs(GraphOperation op) { - super(new BoostedTreesCalculateBestFeatureSplit(op), op, Arrays.asList("logits_dimension", "split_type")); - int inputIndex = 0; - nodeIdRange = (Operand) op.input(inputIndex++); - statsSummary = (Operand) op.input(inputIndex++); - l1 = (Operand) op.input(inputIndex++); - l2 = (Operand) op.input(inputIndex++); - treeComplexity = (Operand) op.input(inputIndex++); - minNodeWeight = (Operand) op.input(inputIndex++); - logitsDimension = op.attributes().getAttrInt("logits_dimension"); - splitType = op.attributes().getAttrString("split_type"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplitV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplitV2.java deleted file mode 100644 index 52a1c17c65a..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestFeatureSplitV2.java +++ /dev/null @@ -1,259 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; -import org.tensorflow.types.TString; - -/** - * Calculates gains for each feature and returns the best possible split information for each node. However, if no split is found, then no split information is returned for that node. - * The split information is the best threshold (bucket id), gains and left/right node contributions per node for each feature. - *

It is possible that not all nodes can be split on each feature. Hence, the list of possible nodes can differ between the features. Therefore, we return {@code node_ids_list} for each feature, containing the list of nodes that this feature can be used to split. - *

In this manner, the output is the best split per features and per node, so that it needs to be combined later to produce the best split for each node (among all possible features). - *

The output shapes are compatible in a way that the first dimension of all tensors are the same and equal to the number of possible split nodes for each feature. - */ -@OpMetadata( - opType = BoostedTreesCalculateBestFeatureSplitV2.OP_NAME, - inputsClass = BoostedTreesCalculateBestFeatureSplitV2.Inputs.class -) -public final class BoostedTreesCalculateBestFeatureSplitV2 extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesCalculateBestFeatureSplitV2"; - - private Output nodeIds; - - private Output gains; - - private Output featureIds; - - private Output featureDimensions; - - private Output thresholds; - - private Output leftNodeContribs; - - private Output rightNodeContribs; - - private Output splitWithDefaultDirections; - - public BoostedTreesCalculateBestFeatureSplitV2(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - nodeIds = operation.output(outputIdx++); - gains = operation.output(outputIdx++); - featureIds = operation.output(outputIdx++); - featureDimensions = operation.output(outputIdx++); - thresholds = operation.output(outputIdx++); - leftNodeContribs = operation.output(outputIdx++); - rightNodeContribs = operation.output(outputIdx++); - splitWithDefaultDirections = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesCalculateBestFeatureSplitV2 operation. - * - * @param scope current scope - * @param nodeIdRange A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). - * @param statsSummariesList A list of Rank 4 tensor (#shape=[max_splits, feature_dims, bucket, stats_dims]) for accumulated stats summary (gradient/hessian) per node, per dimension, per buckets for each feature. - * The first dimension of the tensor is the maximum number of splits, and thus not all elements of it will be used, but only the indexes specified by node_ids will be used. - * @param splitTypes A Rank 1 tensor indicating if this Op should perform inequality split or equality split per feature. - * @param candidateFeatureIds Rank 1 tensor with ids for each feature. This is the real id of the feature. - * @param l1 l1 regularization factor on leaf weights, per instance based. - * @param l2 l2 regularization factor on leaf weights, per instance based. - * @param treeComplexity adjustment to the gain, per leaf based. - * @param minNodeWeight minimum avg of hessians in a node before required for the node to be considered for splitting. - * @param logitsDimension The dimension of logit, i.e., number of classes. - * @return a new instance of BoostedTreesCalculateBestFeatureSplitV2 - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesCalculateBestFeatureSplitV2 create(Scope scope, - Operand nodeIdRange, Iterable> statsSummariesList, - Operand splitTypes, Operand candidateFeatureIds, Operand l1, - Operand l2, Operand treeComplexity, Operand minNodeWeight, - Long logitsDimension) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesCalculateBestFeatureSplitV2"); - opBuilder.addInput(nodeIdRange.asOutput()); - opBuilder.addInputList(Operands.asOutputs(statsSummariesList)); - opBuilder.addInput(splitTypes.asOutput()); - opBuilder.addInput(candidateFeatureIds.asOutput()); - opBuilder.addInput(l1.asOutput()); - opBuilder.addInput(l2.asOutput()); - opBuilder.addInput(treeComplexity.asOutput()); - opBuilder.addInput(minNodeWeight.asOutput()); - opBuilder.setAttr("logits_dimension", logitsDimension); - return new BoostedTreesCalculateBestFeatureSplitV2(opBuilder.build()); - } - - /** - * Gets nodeIds. - * A Rank 1 tensors indicating possible split node ids for each feature. The length of the list is num_features, but each tensor has different size as each feature provides different possible nodes. See above for details like shapes and sizes. - * @return nodeIds. - */ - public Output nodeIds() { - return nodeIds; - } - - /** - * Gets gains. - * A Rank 1 tensor indicating the best gains for each feature to split for certain nodes. See above for details like shapes and sizes. - * @return gains. - */ - public Output gains() { - return gains; - } - - /** - * Gets featureIds. - * A Rank 1 tensors indicating the best feature id for each node. See above for details like shapes and sizes. - * @return featureIds. - */ - public Output featureIds() { - return featureIds; - } - - /** - * Gets featureDimensions. - * A Rank 1 tensors indicating the best feature dimension for each feature to split for certain nodes if the feature is multi-dimension. See above for details like shapes and sizes. - * @return featureDimensions. - */ - public Output featureDimensions() { - return featureDimensions; - } - - /** - * Gets thresholds. - * A Rank 1 tensors indicating the bucket id to compare with (as a threshold) for split in each node. See above for details like shapes and sizes. - * @return thresholds. - */ - public Output thresholds() { - return thresholds; - } - - /** - * Gets leftNodeContribs. - * A Rank 2 tensors indicating the contribution of the left nodes when branching from parent nodes (given by the tensor element in the output node_ids_list) to the left direction by the given threshold for each feature. This value will be used to make the left node value by adding to the parent node value. Second dimension size is 1 for 1-dimensional logits, but would be larger for multi-class problems. See above for details like shapes and sizes. - * @return leftNodeContribs. - */ - public Output leftNodeContribs() { - return leftNodeContribs; - } - - /** - * Gets rightNodeContribs. - * A Rank 2 tensors, with the same shape/conditions as left_node_contribs_list, but just that the value is for the right node. - * @return rightNodeContribs. - */ - public Output rightNodeContribs() { - return rightNodeContribs; - } - - /** - * Gets splitWithDefaultDirections. - * A Rank 1 tensors indicating the which direction to go if data is missing. See above for details like shapes and sizes. - * Inequality with default left returns 0, inequality with default right returns 1, equality with default right returns 2. - * @return splitWithDefaultDirections. - */ - public Output splitWithDefaultDirections() { - return splitWithDefaultDirections; - } - - @OpInputsMetadata( - outputsClass = BoostedTreesCalculateBestFeatureSplitV2.class - ) - public static class Inputs extends RawOpInputs { - /** - * A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). - */ - public final Operand nodeIdRange; - - /** - * A list of Rank 4 tensor (#shape=[max_splits, feature_dims, bucket, stats_dims]) for accumulated stats summary (gradient/hessian) per node, per dimension, per buckets for each feature. - * The first dimension of the tensor is the maximum number of splits, and thus not all elements of it will be used, but only the indexes specified by node_ids will be used. - */ - public final Iterable> statsSummariesList; - - /** - * A Rank 1 tensor indicating if this Op should perform inequality split or equality split per feature. - */ - public final Operand splitTypes; - - /** - * Rank 1 tensor with ids for each feature. This is the real id of the feature. - */ - public final Operand candidateFeatureIds; - - /** - * l1 regularization factor on leaf weights, per instance based. - */ - public final Operand l1; - - /** - * l2 regularization factor on leaf weights, per instance based. - */ - public final Operand l2; - - /** - * adjustment to the gain, per leaf based. - */ - public final Operand treeComplexity; - - /** - * minimum avg of hessians in a node before required for the node to be considered for splitting. - */ - public final Operand minNodeWeight; - - /** - * The dimension of logit, i.e., number of classes. - */ - public final long logitsDimension; - - public Inputs(GraphOperation op) { - super(new BoostedTreesCalculateBestFeatureSplitV2(op), op, Arrays.asList("logits_dimension")); - int inputIndex = 0; - nodeIdRange = (Operand) op.input(inputIndex++); - int statsSummariesListLength = op.inputListLength("stats_summaries_list"); - statsSummariesList = Arrays.asList((Operand[]) op.inputList(inputIndex, statsSummariesListLength)); - inputIndex += statsSummariesListLength; - splitTypes = (Operand) op.input(inputIndex++); - candidateFeatureIds = (Operand) op.input(inputIndex++); - l1 = (Operand) op.input(inputIndex++); - l2 = (Operand) op.input(inputIndex++); - treeComplexity = (Operand) op.input(inputIndex++); - minNodeWeight = (Operand) op.input(inputIndex++); - logitsDimension = op.attributes().getAttrInt("logits_dimension"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestGainsPerFeature.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestGainsPerFeature.java deleted file mode 100644 index 8b91e200217..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCalculateBestGainsPerFeature.java +++ /dev/null @@ -1,215 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import java.util.List; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; - -/** - * Calculates gains for each feature and returns the best possible split information for the feature. - * The split information is the best threshold (bucket id), gains and left/right node contributions per node for each feature. - *

It is possible that not all nodes can be split on each feature. Hence, the list of possible nodes can differ between the features. Therefore, we return {@code node_ids_list} for each feature, containing the list of nodes that this feature can be used to split. - *

In this manner, the output is the best split per features and per node, so that it needs to be combined later to produce the best split for each node (among all possible features). - *

The length of output lists are all of the same length, {@code num_features}. - * The output shapes are compatible in a way that the first dimension of all tensors of all lists are the same and equal to the number of possible split nodes for each feature. - */ -@OpMetadata( - opType = BoostedTreesCalculateBestGainsPerFeature.OP_NAME, - inputsClass = BoostedTreesCalculateBestGainsPerFeature.Inputs.class -) -public final class BoostedTreesCalculateBestGainsPerFeature extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesCalculateBestGainsPerFeature"; - - private List> nodeIdsList; - - private List> gainsList; - - private List> thresholdsList; - - private List> leftNodeContribsList; - - private List> rightNodeContribsList; - - @SuppressWarnings("unchecked") - public BoostedTreesCalculateBestGainsPerFeature(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - int nodeIdsListLength = operation.outputListLength("node_ids_list"); - nodeIdsList = Arrays.asList((Output[]) operation.outputList(outputIdx, nodeIdsListLength)); - outputIdx += nodeIdsListLength; - int gainsListLength = operation.outputListLength("gains_list"); - gainsList = Arrays.asList((Output[]) operation.outputList(outputIdx, gainsListLength)); - outputIdx += gainsListLength; - int thresholdsListLength = operation.outputListLength("thresholds_list"); - thresholdsList = Arrays.asList((Output[]) operation.outputList(outputIdx, thresholdsListLength)); - outputIdx += thresholdsListLength; - int leftNodeContribsListLength = operation.outputListLength("left_node_contribs_list"); - leftNodeContribsList = Arrays.asList((Output[]) operation.outputList(outputIdx, leftNodeContribsListLength)); - outputIdx += leftNodeContribsListLength; - int rightNodeContribsListLength = operation.outputListLength("right_node_contribs_list"); - rightNodeContribsList = Arrays.asList((Output[]) operation.outputList(outputIdx, rightNodeContribsListLength)); - outputIdx += rightNodeContribsListLength; - } - - /** - * Factory method to create a class wrapping a new BoostedTreesCalculateBestGainsPerFeature operation. - * - * @param scope current scope - * @param nodeIdRange A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). - * @param statsSummaryList A list of Rank 3 tensor (#shape=[max_splits, bucket, 2]) for accumulated stats summary (gradient/hessian) per node per buckets for each feature. The first dimension of the tensor is the maximum number of splits, and thus not all elements of it will be used, but only the indexes specified by node_ids will be used. - * @param l1 l1 regularization factor on leaf weights, per instance based. - * @param l2 l2 regularization factor on leaf weights, per instance based. - * @param treeComplexity adjustment to the gain, per leaf based. - * @param minNodeWeight minimum avg of hessians in a node before required for the node to be considered for splitting. - * @param maxSplits the number of nodes that can be split in the whole tree. Used as a dimension of output tensors. - * @return a new instance of BoostedTreesCalculateBestGainsPerFeature - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesCalculateBestGainsPerFeature create(Scope scope, - Operand nodeIdRange, Iterable> statsSummaryList, - Operand l1, Operand l2, Operand treeComplexity, - Operand minNodeWeight, Long maxSplits) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesCalculateBestGainsPerFeature"); - opBuilder.addInput(nodeIdRange.asOutput()); - opBuilder.addInputList(Operands.asOutputs(statsSummaryList)); - opBuilder.addInput(l1.asOutput()); - opBuilder.addInput(l2.asOutput()); - opBuilder.addInput(treeComplexity.asOutput()); - opBuilder.addInput(minNodeWeight.asOutput()); - opBuilder.setAttr("max_splits", maxSplits); - return new BoostedTreesCalculateBestGainsPerFeature(opBuilder.build()); - } - - /** - * Gets nodeIdsList. - * An output list of Rank 1 tensors indicating possible split node ids for each feature. The length of the list is num_features, but each tensor has different size as each feature provides different possible nodes. See above for details like shapes and sizes. - * @return nodeIdsList. - */ - public List> nodeIdsList() { - return nodeIdsList; - } - - /** - * Gets gainsList. - * An output list of Rank 1 tensors indicating the best gains for each feature to split for certain nodes. See above for details like shapes and sizes. - * @return gainsList. - */ - public List> gainsList() { - return gainsList; - } - - /** - * Gets thresholdsList. - * An output list of Rank 1 tensors indicating the bucket id to compare with (as a threshold) for split in each node. See above for details like shapes and sizes. - * @return thresholdsList. - */ - public List> thresholdsList() { - return thresholdsList; - } - - /** - * Gets leftNodeContribsList. - * A list of Rank 2 tensors indicating the contribution of the left nodes when branching from parent nodes (given by the tensor element in the output node_ids_list) to the left direction by the given threshold for each feature. This value will be used to make the left node value by adding to the parent node value. Second dimension size is 1 for 1-dimensional logits, but would be larger for multi-class problems. See above for details like shapes and sizes. - * @return leftNodeContribsList. - */ - public List> leftNodeContribsList() { - return leftNodeContribsList; - } - - /** - * Gets rightNodeContribsList. - * A list of Rank 2 tensors, with the same shape/conditions as left_node_contribs_list, but just that the value is for the right node. - * @return rightNodeContribsList. - */ - public List> rightNodeContribsList() { - return rightNodeContribsList; - } - - @OpInputsMetadata( - outputsClass = BoostedTreesCalculateBestGainsPerFeature.class - ) - public static class Inputs extends RawOpInputs { - /** - * A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). - */ - public final Operand nodeIdRange; - - /** - * A list of Rank 3 tensor (#shape=[max_splits, bucket, 2]) for accumulated stats summary (gradient/hessian) per node per buckets for each feature. The first dimension of the tensor is the maximum number of splits, and thus not all elements of it will be used, but only the indexes specified by node_ids will be used. - */ - public final Iterable> statsSummaryList; - - /** - * l1 regularization factor on leaf weights, per instance based. - */ - public final Operand l1; - - /** - * l2 regularization factor on leaf weights, per instance based. - */ - public final Operand l2; - - /** - * adjustment to the gain, per leaf based. - */ - public final Operand treeComplexity; - - /** - * minimum avg of hessians in a node before required for the node to be considered for splitting. - */ - public final Operand minNodeWeight; - - /** - * the number of nodes that can be split in the whole tree. Used as a dimension of output tensors. - */ - public final long maxSplits; - - public Inputs(GraphOperation op) { - super(new BoostedTreesCalculateBestGainsPerFeature(op), op, Arrays.asList("max_splits")); - int inputIndex = 0; - nodeIdRange = (Operand) op.input(inputIndex++); - int statsSummaryListLength = op.inputListLength("stats_summary_list"); - statsSummaryList = Arrays.asList((Operand[]) op.inputList(inputIndex, statsSummaryListLength)); - inputIndex += statsSummaryListLength; - l1 = (Operand) op.input(inputIndex++); - l2 = (Operand) op.input(inputIndex++); - treeComplexity = (Operand) op.input(inputIndex++); - minNodeWeight = (Operand) op.input(inputIndex++); - maxSplits = op.attributes().getAttrInt("max_splits"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCenterBias.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCenterBias.java deleted file mode 100644 index 9c4073ab5ee..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCenterBias.java +++ /dev/null @@ -1,136 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TBool; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.family.TType; - -/** - * Calculates the prior from the training data (the bias) and fills in the first node with the logits' prior. Returns a boolean indicating whether to continue centering. - */ -@OpMetadata( - opType = BoostedTreesCenterBias.OP_NAME, - inputsClass = BoostedTreesCenterBias.Inputs.class -) -public final class BoostedTreesCenterBias extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesCenterBias"; - - private Output continueCentering; - - public BoostedTreesCenterBias(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - continueCentering = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesCenterBias operation. - * - * @param scope current scope - * @param treeEnsembleHandle Handle to the tree ensemble. - * @param meanGradients A tensor with shape=[logits_dimension] with mean of gradients for a first node. - * @param meanHessians A tensor with shape=[logits_dimension] mean of hessians for a first node. - * @param l1 l1 regularization factor on leaf weights, per instance based. - * @param l2 l2 regularization factor on leaf weights, per instance based. - * @return a new instance of BoostedTreesCenterBias - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesCenterBias create(Scope scope, - Operand treeEnsembleHandle, Operand meanGradients, - Operand meanHessians, Operand l1, Operand l2) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesCenterBias"); - opBuilder.addInput(treeEnsembleHandle.asOutput()); - opBuilder.addInput(meanGradients.asOutput()); - opBuilder.addInput(meanHessians.asOutput()); - opBuilder.addInput(l1.asOutput()); - opBuilder.addInput(l2.asOutput()); - return new BoostedTreesCenterBias(opBuilder.build()); - } - - /** - * Gets continueCentering. - * Bool, whether to continue bias centering. - * @return continueCentering. - */ - public Output continueCentering() { - return continueCentering; - } - - @Override - public Output asOutput() { - return continueCentering; - } - - @OpInputsMetadata( - outputsClass = BoostedTreesCenterBias.class - ) - public static class Inputs extends RawOpInputs { - /** - * Handle to the tree ensemble. - */ - public final Operand treeEnsembleHandle; - - /** - * A tensor with shape=[logits_dimension] with mean of gradients for a first node. - */ - public final Operand meanGradients; - - /** - * A tensor with shape=[logits_dimension] mean of hessians for a first node. - */ - public final Operand meanHessians; - - /** - * l1 regularization factor on leaf weights, per instance based. - */ - public final Operand l1; - - /** - * l2 regularization factor on leaf weights, per instance based. - */ - public final Operand l2; - - public Inputs(GraphOperation op) { - super(new BoostedTreesCenterBias(op), op, Arrays.asList()); - int inputIndex = 0; - treeEnsembleHandle = (Operand) op.input(inputIndex++); - meanGradients = (Operand) op.input(inputIndex++); - meanHessians = (Operand) op.input(inputIndex++); - l1 = (Operand) op.input(inputIndex++); - l2 = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateEnsemble.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateEnsemble.java deleted file mode 100644 index 63bfe3ad920..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateEnsemble.java +++ /dev/null @@ -1,101 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TInt64; -import org.tensorflow.types.TString; -import org.tensorflow.types.family.TType; - -/** - * Creates a tree ensemble model and returns a handle to it. - */ -@OpMetadata( - opType = BoostedTreesCreateEnsemble.OP_NAME, - inputsClass = BoostedTreesCreateEnsemble.Inputs.class -) -public final class BoostedTreesCreateEnsemble extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesCreateEnsemble"; - - public BoostedTreesCreateEnsemble(Operation operation) { - super(operation, OP_NAME); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesCreateEnsemble operation. - * - * @param scope current scope - * @param treeEnsembleHandle Handle to the tree ensemble resource to be created. - * @param stampToken Token to use as the initial value of the resource stamp. - * @param treeEnsembleSerialized Serialized proto of the tree ensemble. - * @return a new instance of BoostedTreesCreateEnsemble - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesCreateEnsemble create(Scope scope, - Operand treeEnsembleHandle, Operand stampToken, - Operand treeEnsembleSerialized) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesCreateEnsemble"); - opBuilder.addInput(treeEnsembleHandle.asOutput()); - opBuilder.addInput(stampToken.asOutput()); - opBuilder.addInput(treeEnsembleSerialized.asOutput()); - return new BoostedTreesCreateEnsemble(opBuilder.build()); - } - - @OpInputsMetadata( - outputsClass = BoostedTreesCreateEnsemble.class - ) - public static class Inputs extends RawOpInputs { - /** - * Handle to the tree ensemble resource to be created. - */ - public final Operand treeEnsembleHandle; - - /** - * Token to use as the initial value of the resource stamp. - */ - public final Operand stampToken; - - /** - * Serialized proto of the tree ensemble. - */ - public final Operand treeEnsembleSerialized; - - public Inputs(GraphOperation op) { - super(new BoostedTreesCreateEnsemble(op), op, Arrays.asList()); - int inputIndex = 0; - treeEnsembleHandle = (Operand) op.input(inputIndex++); - stampToken = (Operand) op.input(inputIndex++); - treeEnsembleSerialized = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateQuantileStreamResource.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateQuantileStreamResource.java deleted file mode 100644 index 06f3c56e7cb..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesCreateQuantileStreamResource.java +++ /dev/null @@ -1,146 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt64; -import org.tensorflow.types.family.TType; - -/** - * Create the Resource for Quantile Streams. - */ -@OpMetadata( - opType = BoostedTreesCreateQuantileStreamResource.OP_NAME, - inputsClass = BoostedTreesCreateQuantileStreamResource.Inputs.class -) -public final class BoostedTreesCreateQuantileStreamResource extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesCreateQuantileStreamResource"; - - public BoostedTreesCreateQuantileStreamResource(Operation operation) { - super(operation, OP_NAME); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesCreateQuantileStreamResource operation. - * - * @param scope current scope - * @param quantileStreamResourceHandle resource; Handle to quantile stream resource. - * @param epsilon float; The required approximation error of the stream resource. - * @param numStreams int; The number of streams managed by the resource that shares the same epsilon. - * @param options carries optional attribute values - * @return a new instance of BoostedTreesCreateQuantileStreamResource - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesCreateQuantileStreamResource create(Scope scope, - Operand quantileStreamResourceHandle, Operand epsilon, - Operand numStreams, Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesCreateQuantileStreamResource"); - opBuilder.addInput(quantileStreamResourceHandle.asOutput()); - opBuilder.addInput(epsilon.asOutput()); - opBuilder.addInput(numStreams.asOutput()); - if (options != null) { - for (Options opts : options) { - if (opts.maxElements != null) { - opBuilder.setAttr("max_elements", opts.maxElements); - } - } - } - return new BoostedTreesCreateQuantileStreamResource(opBuilder.build()); - } - - /** - * Sets the maxElements option. - * - * @param maxElements int; The maximum number of data points that can be fed to the stream. - * @return this Options instance. - */ - public static Options maxElements(Long maxElements) { - return new Options().maxElements(maxElements); - } - - /** - * Optional attributes for {@link org.tensorflow.op.estimator.BoostedTreesCreateQuantileStreamResource} - */ - public static class Options { - private Long maxElements; - - private Options() { - } - - /** - * Sets the maxElements option. - * - * @param maxElements int; The maximum number of data points that can be fed to the stream. - * @return this Options instance. - */ - public Options maxElements(Long maxElements) { - this.maxElements = maxElements; - return this; - } - } - - @OpInputsMetadata( - outputsClass = BoostedTreesCreateQuantileStreamResource.class - ) - public static class Inputs extends RawOpInputs { - /** - * resource; Handle to quantile stream resource. - */ - public final Operand quantileStreamResourceHandle; - - /** - * float; The required approximation error of the stream resource. - */ - public final Operand epsilon; - - /** - * int; The number of streams managed by the resource that shares the same epsilon. - */ - public final Operand numStreams; - - /** - * int; The maximum number of data points that can be fed to the stream. - */ - public final long maxElements; - - public Inputs(GraphOperation op) { - super(new BoostedTreesCreateQuantileStreamResource(op), op, Arrays.asList("max_elements")); - int inputIndex = 0; - quantileStreamResourceHandle = (Operand) op.input(inputIndex++); - epsilon = (Operand) op.input(inputIndex++); - numStreams = (Operand) op.input(inputIndex++); - maxElements = op.attributes().getAttrInt("max_elements"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesDeserializeEnsemble.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesDeserializeEnsemble.java deleted file mode 100644 index 371500c3064..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesDeserializeEnsemble.java +++ /dev/null @@ -1,102 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TInt64; -import org.tensorflow.types.TString; -import org.tensorflow.types.family.TType; - -/** - * Deserializes a serialized tree ensemble config and replaces current tree - * ensemble. - */ -@OpMetadata( - opType = BoostedTreesDeserializeEnsemble.OP_NAME, - inputsClass = BoostedTreesDeserializeEnsemble.Inputs.class -) -public final class BoostedTreesDeserializeEnsemble extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesDeserializeEnsemble"; - - public BoostedTreesDeserializeEnsemble(Operation operation) { - super(operation, OP_NAME); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesDeserializeEnsemble operation. - * - * @param scope current scope - * @param treeEnsembleHandle Handle to the tree ensemble. - * @param stampToken Token to use as the new value of the resource stamp. - * @param treeEnsembleSerialized Serialized proto of the ensemble. - * @return a new instance of BoostedTreesDeserializeEnsemble - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesDeserializeEnsemble create(Scope scope, - Operand treeEnsembleHandle, Operand stampToken, - Operand treeEnsembleSerialized) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesDeserializeEnsemble"); - opBuilder.addInput(treeEnsembleHandle.asOutput()); - opBuilder.addInput(stampToken.asOutput()); - opBuilder.addInput(treeEnsembleSerialized.asOutput()); - return new BoostedTreesDeserializeEnsemble(opBuilder.build()); - } - - @OpInputsMetadata( - outputsClass = BoostedTreesDeserializeEnsemble.class - ) - public static class Inputs extends RawOpInputs { - /** - * Handle to the tree ensemble. - */ - public final Operand treeEnsembleHandle; - - /** - * Token to use as the new value of the resource stamp. - */ - public final Operand stampToken; - - /** - * Serialized proto of the ensemble. - */ - public final Operand treeEnsembleSerialized; - - public Inputs(GraphOperation op) { - super(new BoostedTreesDeserializeEnsemble(op), op, Arrays.asList()); - int inputIndex = 0; - treeEnsembleHandle = (Operand) op.input(inputIndex++); - stampToken = (Operand) op.input(inputIndex++); - treeEnsembleSerialized = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesEnsembleResourceHandleOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesEnsembleResourceHandleOp.java deleted file mode 100644 index 6afffd3a1ec..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesEnsembleResourceHandleOp.java +++ /dev/null @@ -1,171 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.family.TType; - -/** - * Creates a handle to a BoostedTreesEnsembleResource - */ -@OpMetadata( - opType = BoostedTreesEnsembleResourceHandleOp.OP_NAME, - inputsClass = BoostedTreesEnsembleResourceHandleOp.Inputs.class -) -public final class BoostedTreesEnsembleResourceHandleOp extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesEnsembleResourceHandleOp"; - - private Output resource; - - @SuppressWarnings("unchecked") - public BoostedTreesEnsembleResourceHandleOp(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - resource = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesEnsembleResourceHandleOp operation. - * - * @param scope current scope - * @param options carries optional attribute values - * @return a new instance of BoostedTreesEnsembleResourceHandleOp - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesEnsembleResourceHandleOp create(Scope scope, Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesEnsembleResourceHandleOp"); - if (options != null) { - for (Options opts : options) { - if (opts.container != null) { - opBuilder.setAttr("container", opts.container); - } - if (opts.sharedName != null) { - opBuilder.setAttr("shared_name", opts.sharedName); - } - } - } - return new BoostedTreesEnsembleResourceHandleOp(opBuilder.build()); - } - - /** - * Sets the container option. - * - * @param container the container option - * @return this Options instance. - */ - public static Options container(String container) { - return new Options().container(container); - } - - /** - * Sets the sharedName option. - * - * @param sharedName the sharedName option - * @return this Options instance. - */ - public static Options sharedName(String sharedName) { - return new Options().sharedName(sharedName); - } - - /** - * Gets resource. - * - * @return resource. - */ - public Output resource() { - return resource; - } - - @Override - @SuppressWarnings("unchecked") - public Output asOutput() { - return (Output) resource; - } - - /** - * Optional attributes for {@link org.tensorflow.op.estimator.BoostedTreesEnsembleResourceHandleOp} - */ - public static class Options { - private String container; - - private String sharedName; - - private Options() { - } - - /** - * Sets the container option. - * - * @param container the container option - * @return this Options instance. - */ - public Options container(String container) { - this.container = container; - return this; - } - - /** - * Sets the sharedName option. - * - * @param sharedName the sharedName option - * @return this Options instance. - */ - public Options sharedName(String sharedName) { - this.sharedName = sharedName; - return this; - } - } - - @OpInputsMetadata( - outputsClass = BoostedTreesEnsembleResourceHandleOp.class - ) - public static class Inputs extends RawOpInputs { - /** - * The container attribute - */ - public final String container; - - /** - * The sharedName attribute - */ - public final String sharedName; - - public Inputs(GraphOperation op) { - super(new BoostedTreesEnsembleResourceHandleOp(op), op, Arrays.asList("container", "shared_name")); - int inputIndex = 0; - container = op.attributes().getAttrString("container"); - sharedName = op.attributes().getAttrString("shared_name"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesExampleDebugOutputs.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesExampleDebugOutputs.java deleted file mode 100644 index 06ee048d9f5..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesExampleDebugOutputs.java +++ /dev/null @@ -1,130 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TInt32; -import org.tensorflow.types.TString; -import org.tensorflow.types.family.TType; - -/** - * Debugging/model interpretability outputs for each example. - * It traverses all the trees and computes debug metrics for individual examples, - * such as getting split feature ids and logits after each split along the decision - * path used to compute directional feature contributions. - */ -@OpMetadata( - opType = BoostedTreesExampleDebugOutputs.OP_NAME, - inputsClass = BoostedTreesExampleDebugOutputs.Inputs.class -) -public final class BoostedTreesExampleDebugOutputs extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesExampleDebugOutputs"; - - private Output examplesDebugOutputsSerialized; - - public BoostedTreesExampleDebugOutputs(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - examplesDebugOutputsSerialized = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesExampleDebugOutputs operation. - * - * @param scope current scope - * @param treeEnsembleHandle The treeEnsembleHandle value - * @param bucketizedFeatures A list of rank 1 Tensors containing bucket id for each - * feature. - * @param logitsDimension scalar, dimension of the logits, to be used for constructing the protos in - * examples_debug_outputs_serialized. - * @return a new instance of BoostedTreesExampleDebugOutputs - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesExampleDebugOutputs create(Scope scope, - Operand treeEnsembleHandle, Iterable> bucketizedFeatures, - Long logitsDimension) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesExampleDebugOutputs"); - opBuilder.addInput(treeEnsembleHandle.asOutput()); - opBuilder.addInputList(Operands.asOutputs(bucketizedFeatures)); - opBuilder.setAttr("logits_dimension", logitsDimension); - return new BoostedTreesExampleDebugOutputs(opBuilder.build()); - } - - /** - * Gets examplesDebugOutputsSerialized. - * Output rank 1 Tensor containing a proto serialized as a string for each example. - * @return examplesDebugOutputsSerialized. - */ - public Output examplesDebugOutputsSerialized() { - return examplesDebugOutputsSerialized; - } - - @Override - public Output asOutput() { - return examplesDebugOutputsSerialized; - } - - @OpInputsMetadata( - outputsClass = BoostedTreesExampleDebugOutputs.class - ) - public static class Inputs extends RawOpInputs { - /** - * The treeEnsembleHandle input - */ - public final Operand treeEnsembleHandle; - - /** - * A list of rank 1 Tensors containing bucket id for each - * feature. - */ - public final Iterable> bucketizedFeatures; - - /** - * scalar, dimension of the logits, to be used for constructing the protos in - * examples_debug_outputs_serialized. - */ - public final long logitsDimension; - - public Inputs(GraphOperation op) { - super(new BoostedTreesExampleDebugOutputs(op), op, Arrays.asList("logits_dimension")); - int inputIndex = 0; - treeEnsembleHandle = (Operand) op.input(inputIndex++); - int bucketizedFeaturesLength = op.inputListLength("bucketized_features"); - bucketizedFeatures = Arrays.asList((Operand[]) op.inputList(inputIndex, bucketizedFeaturesLength)); - inputIndex += bucketizedFeaturesLength; - logitsDimension = op.attributes().getAttrInt("logits_dimension"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesFlushQuantileSummaries.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesFlushQuantileSummaries.java deleted file mode 100644 index 3a523c29430..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesFlushQuantileSummaries.java +++ /dev/null @@ -1,113 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import java.util.Iterator; -import java.util.List; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.family.TType; - -/** - * Flush the quantile summaries from each quantile stream resource. - * An op that outputs a list of quantile summaries of a quantile stream resource. - * Each summary Tensor is rank 2, containing summaries (value, weight, min_rank, - * max_rank) for a single feature. - */ -@OpMetadata( - opType = BoostedTreesFlushQuantileSummaries.OP_NAME, - inputsClass = BoostedTreesFlushQuantileSummaries.Inputs.class -) -public final class BoostedTreesFlushQuantileSummaries extends RawOp implements Iterable> { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesFlushQuantileSummaries"; - - private List> summaries; - - @SuppressWarnings("unchecked") - public BoostedTreesFlushQuantileSummaries(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - int summariesLength = operation.outputListLength("summaries"); - summaries = Arrays.asList((Output[]) operation.outputList(outputIdx, summariesLength)); - outputIdx += summariesLength; - } - - /** - * Factory method to create a class wrapping a new BoostedTreesFlushQuantileSummaries operation. - * - * @param scope current scope - * @param quantileStreamResourceHandle resource handle referring to a QuantileStreamResource. - * @param numFeatures The value of the numFeatures attribute - * @return a new instance of BoostedTreesFlushQuantileSummaries - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesFlushQuantileSummaries create(Scope scope, - Operand quantileStreamResourceHandle, Long numFeatures) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesFlushQuantileSummaries"); - opBuilder.addInput(quantileStreamResourceHandle.asOutput()); - opBuilder.setAttr("num_features", numFeatures); - return new BoostedTreesFlushQuantileSummaries(opBuilder.build()); - } - - /** - * Gets summaries. - * - * @return summaries. - */ - public List> summaries() { - return summaries; - } - - @Override - @SuppressWarnings({"rawtypes", "unchecked"}) - public Iterator> iterator() { - return (Iterator) summaries.iterator(); - } - - @OpInputsMetadata( - outputsClass = BoostedTreesFlushQuantileSummaries.class - ) - public static class Inputs extends RawOpInputs { - /** - * resource handle referring to a QuantileStreamResource. - */ - public final Operand quantileStreamResourceHandle; - - public Inputs(GraphOperation op) { - super(new BoostedTreesFlushQuantileSummaries(op), op, Arrays.asList()); - int inputIndex = 0; - quantileStreamResourceHandle = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesGetEnsembleStates.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesGetEnsembleStates.java deleted file mode 100644 index 09d0bdeedd1..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesGetEnsembleStates.java +++ /dev/null @@ -1,147 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TInt32; -import org.tensorflow.types.TInt64; -import org.tensorflow.types.family.TType; - -/** - * Retrieves the tree ensemble resource stamp token, number of trees and growing statistics. - */ -@OpMetadata( - opType = BoostedTreesGetEnsembleStates.OP_NAME, - inputsClass = BoostedTreesGetEnsembleStates.Inputs.class -) -public final class BoostedTreesGetEnsembleStates extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesGetEnsembleStates"; - - private Output stampToken; - - private Output numTrees; - - private Output numFinalizedTrees; - - private Output numAttemptedLayers; - - private Output lastLayerNodesRange; - - public BoostedTreesGetEnsembleStates(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - stampToken = operation.output(outputIdx++); - numTrees = operation.output(outputIdx++); - numFinalizedTrees = operation.output(outputIdx++); - numAttemptedLayers = operation.output(outputIdx++); - lastLayerNodesRange = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesGetEnsembleStates operation. - * - * @param scope current scope - * @param treeEnsembleHandle Handle to the tree ensemble. - * @return a new instance of BoostedTreesGetEnsembleStates - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesGetEnsembleStates create(Scope scope, - Operand treeEnsembleHandle) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesGetEnsembleStates"); - opBuilder.addInput(treeEnsembleHandle.asOutput()); - return new BoostedTreesGetEnsembleStates(opBuilder.build()); - } - - /** - * Gets stampToken. - * Stamp token of the tree ensemble resource. - * @return stampToken. - */ - public Output stampToken() { - return stampToken; - } - - /** - * Gets numTrees. - * The number of trees in the tree ensemble resource. - * @return numTrees. - */ - public Output numTrees() { - return numTrees; - } - - /** - * Gets numFinalizedTrees. - * The number of trees that were finished successfully. - * @return numFinalizedTrees. - */ - public Output numFinalizedTrees() { - return numFinalizedTrees; - } - - /** - * Gets numAttemptedLayers. - * The number of layers we attempted to build (but not necessarily succeeded). - * @return numAttemptedLayers. - */ - public Output numAttemptedLayers() { - return numAttemptedLayers; - } - - /** - * Gets lastLayerNodesRange. - * Rank size 2 tensor that contains start and end ids of the nodes in the latest - * layer. - * @return lastLayerNodesRange. - */ - public Output lastLayerNodesRange() { - return lastLayerNodesRange; - } - - @OpInputsMetadata( - outputsClass = BoostedTreesGetEnsembleStates.class - ) - public static class Inputs extends RawOpInputs { - /** - * Handle to the tree ensemble. - */ - public final Operand treeEnsembleHandle; - - public Inputs(GraphOperation op) { - super(new BoostedTreesGetEnsembleStates(op), op, Arrays.asList()); - int inputIndex = 0; - treeEnsembleHandle = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeQuantileSummaries.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeQuantileSummaries.java deleted file mode 100644 index 27c220672c3..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeQuantileSummaries.java +++ /dev/null @@ -1,130 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import java.util.Iterator; -import java.util.List; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; - -/** - * Makes the summary of quantiles for the batch. - * An op that takes a list of tensors (one tensor per feature) and outputs the - * quantile summaries for each tensor. - */ -@OpMetadata( - opType = BoostedTreesMakeQuantileSummaries.OP_NAME, - inputsClass = BoostedTreesMakeQuantileSummaries.Inputs.class -) -public final class BoostedTreesMakeQuantileSummaries extends RawOp implements Iterable> { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesMakeQuantileSummaries"; - - private List> summaries; - - @SuppressWarnings("unchecked") - public BoostedTreesMakeQuantileSummaries(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - int summariesLength = operation.outputListLength("summaries"); - summaries = Arrays.asList((Output[]) operation.outputList(outputIdx, summariesLength)); - outputIdx += summariesLength; - } - - /** - * Factory method to create a class wrapping a new BoostedTreesMakeQuantileSummaries operation. - * - * @param scope current scope - * @param floatValues float; List of Rank 1 Tensors each containing values for a single feature. - * @param exampleWeights float; Rank 1 Tensor with weights per instance. - * @param epsilon float; The required maximum approximation error. - * @return a new instance of BoostedTreesMakeQuantileSummaries - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesMakeQuantileSummaries create(Scope scope, - Iterable> floatValues, Operand exampleWeights, - Operand epsilon) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesMakeQuantileSummaries"); - opBuilder.addInputList(Operands.asOutputs(floatValues)); - opBuilder.addInput(exampleWeights.asOutput()); - opBuilder.addInput(epsilon.asOutput()); - return new BoostedTreesMakeQuantileSummaries(opBuilder.build()); - } - - /** - * Gets summaries. - * float; List of Rank 2 Tensors each containing the quantile summary - * (value, weight, min_rank, max_rank) of a single feature. - * @return summaries. - */ - public List> summaries() { - return summaries; - } - - @Override - @SuppressWarnings({"rawtypes", "unchecked"}) - public Iterator> iterator() { - return (Iterator) summaries.iterator(); - } - - @OpInputsMetadata( - outputsClass = BoostedTreesMakeQuantileSummaries.class - ) - public static class Inputs extends RawOpInputs { - /** - * float; List of Rank 1 Tensors each containing values for a single feature. - */ - public final Iterable> floatValues; - - /** - * float; Rank 1 Tensor with weights per instance. - */ - public final Operand exampleWeights; - - /** - * float; The required maximum approximation error. - */ - public final Operand epsilon; - - public Inputs(GraphOperation op) { - super(new BoostedTreesMakeQuantileSummaries(op), op, Arrays.asList()); - int inputIndex = 0; - int floatValuesLength = op.inputListLength("float_values"); - floatValues = Arrays.asList((Operand[]) op.inputList(inputIndex, floatValuesLength)); - inputIndex += floatValuesLength; - exampleWeights = (Operand) op.input(inputIndex++); - epsilon = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeStatsSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeStatsSummary.java deleted file mode 100644 index 4516dfdc8b1..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesMakeStatsSummary.java +++ /dev/null @@ -1,147 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; - -/** - * Makes the summary of accumulated stats for the batch. - * The summary stats contains gradients and hessians accumulated into the corresponding node and bucket for each example. - */ -@OpMetadata( - opType = BoostedTreesMakeStatsSummary.OP_NAME, - inputsClass = BoostedTreesMakeStatsSummary.Inputs.class -) -public final class BoostedTreesMakeStatsSummary extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesMakeStatsSummary"; - - private Output statsSummary; - - public BoostedTreesMakeStatsSummary(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - statsSummary = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesMakeStatsSummary operation. - * - * @param scope current scope - * @param nodeIds int32 Rank 1 Tensor containing node ids, which each example falls into for the requested layer. - * @param gradients float32; Rank 2 Tensor (shape=[#examples, 1]) for gradients. - * @param hessians float32; Rank 2 Tensor (shape=[#examples, 1]) for hessians. - * @param bucketizedFeaturesList int32 list of Rank 1 Tensors, each containing the bucketized feature (for each feature column). - * @param maxSplits int; the maximum number of splits possible in the whole tree. - * @param numBuckets int; equals to the maximum possible value of bucketized feature. - * @return a new instance of BoostedTreesMakeStatsSummary - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesMakeStatsSummary create(Scope scope, Operand nodeIds, - Operand gradients, Operand hessians, - Iterable> bucketizedFeaturesList, Long maxSplits, Long numBuckets) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesMakeStatsSummary"); - opBuilder.addInput(nodeIds.asOutput()); - opBuilder.addInput(gradients.asOutput()); - opBuilder.addInput(hessians.asOutput()); - opBuilder.addInputList(Operands.asOutputs(bucketizedFeaturesList)); - opBuilder.setAttr("max_splits", maxSplits); - opBuilder.setAttr("num_buckets", numBuckets); - return new BoostedTreesMakeStatsSummary(opBuilder.build()); - } - - /** - * Gets statsSummary. - * output Rank 4 Tensor (shape=[#features, #splits, #buckets, 2]) containing accumulated stats put into the corresponding node and bucket. The first index of 4th dimension refers to gradients, and the second to hessians. - * @return statsSummary. - */ - public Output statsSummary() { - return statsSummary; - } - - @Override - public Output asOutput() { - return statsSummary; - } - - @OpInputsMetadata( - outputsClass = BoostedTreesMakeStatsSummary.class - ) - public static class Inputs extends RawOpInputs { - /** - * int32 Rank 1 Tensor containing node ids, which each example falls into for the requested layer. - */ - public final Operand nodeIds; - - /** - * float32; Rank 2 Tensor (shape=[#examples, 1]) for gradients. - */ - public final Operand gradients; - - /** - * float32; Rank 2 Tensor (shape=[#examples, 1]) for hessians. - */ - public final Operand hessians; - - /** - * int32 list of Rank 1 Tensors, each containing the bucketized feature (for each feature column). - */ - public final Iterable> bucketizedFeaturesList; - - /** - * int; the maximum number of splits possible in the whole tree. - */ - public final long maxSplits; - - /** - * int; equals to the maximum possible value of bucketized feature. - */ - public final long numBuckets; - - public Inputs(GraphOperation op) { - super(new BoostedTreesMakeStatsSummary(op), op, Arrays.asList("max_splits", "num_buckets")); - int inputIndex = 0; - nodeIds = (Operand) op.input(inputIndex++); - gradients = (Operand) op.input(inputIndex++); - hessians = (Operand) op.input(inputIndex++); - int bucketizedFeaturesListLength = op.inputListLength("bucketized_features_list"); - bucketizedFeaturesList = Arrays.asList((Operand[]) op.inputList(inputIndex, bucketizedFeaturesListLength)); - inputIndex += bucketizedFeaturesListLength; - maxSplits = op.attributes().getAttrInt("max_splits"); - numBuckets = op.attributes().getAttrInt("num_buckets"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesPredict.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesPredict.java deleted file mode 100644 index a39b3ad733b..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesPredict.java +++ /dev/null @@ -1,128 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; -import org.tensorflow.types.family.TType; - -/** - * Runs multiple additive regression ensemble predictors on input instances and - * computes the logits. It is designed to be used during prediction. - * It traverses all the trees and calculates the final score for each instance. - */ -@OpMetadata( - opType = BoostedTreesPredict.OP_NAME, - inputsClass = BoostedTreesPredict.Inputs.class -) -public final class BoostedTreesPredict extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesPredict"; - - private Output logits; - - public BoostedTreesPredict(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - logits = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesPredict operation. - * - * @param scope current scope - * @param treeEnsembleHandle The treeEnsembleHandle value - * @param bucketizedFeatures A list of rank 1 Tensors containing bucket id for each - * feature. - * @param logitsDimension scalar, dimension of the logits, to be used for partial logits - * shape. - * @return a new instance of BoostedTreesPredict - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesPredict create(Scope scope, Operand treeEnsembleHandle, - Iterable> bucketizedFeatures, Long logitsDimension) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesPredict"); - opBuilder.addInput(treeEnsembleHandle.asOutput()); - opBuilder.addInputList(Operands.asOutputs(bucketizedFeatures)); - opBuilder.setAttr("logits_dimension", logitsDimension); - return new BoostedTreesPredict(opBuilder.build()); - } - - /** - * Gets logits. - * Output rank 2 Tensor containing logits for each example. - * @return logits. - */ - public Output logits() { - return logits; - } - - @Override - public Output asOutput() { - return logits; - } - - @OpInputsMetadata( - outputsClass = BoostedTreesPredict.class - ) - public static class Inputs extends RawOpInputs { - /** - * The treeEnsembleHandle input - */ - public final Operand treeEnsembleHandle; - - /** - * A list of rank 1 Tensors containing bucket id for each - * feature. - */ - public final Iterable> bucketizedFeatures; - - /** - * scalar, dimension of the logits, to be used for partial logits - * shape. - */ - public final long logitsDimension; - - public Inputs(GraphOperation op) { - super(new BoostedTreesPredict(op), op, Arrays.asList("logits_dimension")); - int inputIndex = 0; - treeEnsembleHandle = (Operand) op.input(inputIndex++); - int bucketizedFeaturesLength = op.inputListLength("bucketized_features"); - bucketizedFeatures = Arrays.asList((Operand[]) op.inputList(inputIndex, bucketizedFeaturesLength)); - inputIndex += bucketizedFeaturesLength; - logitsDimension = op.attributes().getAttrInt("logits_dimension"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceAddSummaries.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceAddSummaries.java deleted file mode 100644 index f122432eef1..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceAddSummaries.java +++ /dev/null @@ -1,98 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.family.TType; - -/** - * Add the quantile summaries to each quantile stream resource. - * An op that adds a list of quantile summaries to a quantile stream resource. Each - * summary Tensor is rank 2, containing summaries (value, weight, min_rank, max_rank) - * for a single feature. - */ -@OpMetadata( - opType = BoostedTreesQuantileStreamResourceAddSummaries.OP_NAME, - inputsClass = BoostedTreesQuantileStreamResourceAddSummaries.Inputs.class -) -public final class BoostedTreesQuantileStreamResourceAddSummaries extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesQuantileStreamResourceAddSummaries"; - - public BoostedTreesQuantileStreamResourceAddSummaries(Operation operation) { - super(operation, OP_NAME); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesQuantileStreamResourceAddSummaries operation. - * - * @param scope current scope - * @param quantileStreamResourceHandle resource handle referring to a QuantileStreamResource. - * @param summaries string; List of Rank 2 Tensor each containing the summaries for a single feature. - * @return a new instance of BoostedTreesQuantileStreamResourceAddSummaries - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesQuantileStreamResourceAddSummaries create(Scope scope, - Operand quantileStreamResourceHandle, - Iterable> summaries) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesQuantileStreamResourceAddSummaries"); - opBuilder.addInput(quantileStreamResourceHandle.asOutput()); - opBuilder.addInputList(Operands.asOutputs(summaries)); - return new BoostedTreesQuantileStreamResourceAddSummaries(opBuilder.build()); - } - - @OpInputsMetadata( - outputsClass = BoostedTreesQuantileStreamResourceAddSummaries.class - ) - public static class Inputs extends RawOpInputs { - /** - * resource handle referring to a QuantileStreamResource. - */ - public final Operand quantileStreamResourceHandle; - - /** - * string; List of Rank 2 Tensor each containing the summaries for a single feature. - */ - public final Iterable> summaries; - - public Inputs(GraphOperation op) { - super(new BoostedTreesQuantileStreamResourceAddSummaries(op), op, Arrays.asList()); - int inputIndex = 0; - quantileStreamResourceHandle = (Operand) op.input(inputIndex++); - int summariesLength = op.inputListLength("summaries"); - summaries = Arrays.asList((Operand[]) op.inputList(inputIndex, summariesLength)); - inputIndex += summariesLength; - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceDeserialize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceDeserialize.java deleted file mode 100644 index 691d81c3951..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceDeserialize.java +++ /dev/null @@ -1,96 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.family.TType; - -/** - * Deserialize bucket boundaries and ready flag into current QuantileAccumulator. - * An op that deserializes bucket boundaries and are boundaries ready flag into current QuantileAccumulator. - */ -@OpMetadata( - opType = BoostedTreesQuantileStreamResourceDeserialize.OP_NAME, - inputsClass = BoostedTreesQuantileStreamResourceDeserialize.Inputs.class -) -public final class BoostedTreesQuantileStreamResourceDeserialize extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesQuantileStreamResourceDeserialize"; - - public BoostedTreesQuantileStreamResourceDeserialize(Operation operation) { - super(operation, OP_NAME); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesQuantileStreamResourceDeserialize operation. - * - * @param scope current scope - * @param quantileStreamResourceHandle resource handle referring to a QuantileStreamResource. - * @param bucketBoundaries float; List of Rank 1 Tensors each containing the bucket boundaries for a feature. - * @return a new instance of BoostedTreesQuantileStreamResourceDeserialize - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesQuantileStreamResourceDeserialize create(Scope scope, - Operand quantileStreamResourceHandle, - Iterable> bucketBoundaries) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesQuantileStreamResourceDeserialize"); - opBuilder.addInput(quantileStreamResourceHandle.asOutput()); - opBuilder.addInputList(Operands.asOutputs(bucketBoundaries)); - return new BoostedTreesQuantileStreamResourceDeserialize(opBuilder.build()); - } - - @OpInputsMetadata( - outputsClass = BoostedTreesQuantileStreamResourceDeserialize.class - ) - public static class Inputs extends RawOpInputs { - /** - * resource handle referring to a QuantileStreamResource. - */ - public final Operand quantileStreamResourceHandle; - - /** - * float; List of Rank 1 Tensors each containing the bucket boundaries for a feature. - */ - public final Iterable> bucketBoundaries; - - public Inputs(GraphOperation op) { - super(new BoostedTreesQuantileStreamResourceDeserialize(op), op, Arrays.asList()); - int inputIndex = 0; - quantileStreamResourceHandle = (Operand) op.input(inputIndex++); - int bucketBoundariesLength = op.inputListLength("bucket_boundaries"); - bucketBoundaries = Arrays.asList((Operand[]) op.inputList(inputIndex, bucketBoundariesLength)); - inputIndex += bucketBoundariesLength; - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceFlush.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceFlush.java deleted file mode 100644 index ab971af5a27..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceFlush.java +++ /dev/null @@ -1,153 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TInt64; -import org.tensorflow.types.family.TType; - -/** - * Flush the summaries for a quantile stream resource. - * An op that flushes the summaries for a quantile stream resource. - */ -@OpMetadata( - opType = BoostedTreesQuantileStreamResourceFlush.OP_NAME, - inputsClass = BoostedTreesQuantileStreamResourceFlush.Inputs.class -) -public final class BoostedTreesQuantileStreamResourceFlush extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesQuantileStreamResourceFlush"; - - public BoostedTreesQuantileStreamResourceFlush(Operation operation) { - super(operation, OP_NAME); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesQuantileStreamResourceFlush operation. - * - * @param scope current scope - * @param quantileStreamResourceHandle resource handle referring to a QuantileStreamResource. - * @param numBuckets int; approximate number of buckets unless using generate_quantiles. - * @param options carries optional attribute values - * @return a new instance of BoostedTreesQuantileStreamResourceFlush - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesQuantileStreamResourceFlush create(Scope scope, - Operand quantileStreamResourceHandle, Operand numBuckets, - Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesQuantileStreamResourceFlush"); - opBuilder.addInput(quantileStreamResourceHandle.asOutput()); - opBuilder.addInput(numBuckets.asOutput()); - if (options != null) { - for (Options opts : options) { - if (opts.generateQuantiles != null) { - opBuilder.setAttr("generate_quantiles", opts.generateQuantiles); - } - } - } - return new BoostedTreesQuantileStreamResourceFlush(opBuilder.build()); - } - - /** - * Sets the generateQuantiles option. - * - * @param generateQuantiles bool; If True, the output will be the num_quantiles for each stream where the ith - * entry is the ith quantile of the input with an approximation error of epsilon. - * Duplicate values may be present. - * If False, the output will be the points in the histogram that we got which roughly - * translates to 1/epsilon boundaries and without any duplicates. - * Default to False. - * @return this Options instance. - */ - public static Options generateQuantiles(Boolean generateQuantiles) { - return new Options().generateQuantiles(generateQuantiles); - } - - /** - * Optional attributes for {@link org.tensorflow.op.estimator.BoostedTreesQuantileStreamResourceFlush} - */ - public static class Options { - private Boolean generateQuantiles; - - private Options() { - } - - /** - * Sets the generateQuantiles option. - * - * @param generateQuantiles bool; If True, the output will be the num_quantiles for each stream where the ith - * entry is the ith quantile of the input with an approximation error of epsilon. - * Duplicate values may be present. - * If False, the output will be the points in the histogram that we got which roughly - * translates to 1/epsilon boundaries and without any duplicates. - * Default to False. - * @return this Options instance. - */ - public Options generateQuantiles(Boolean generateQuantiles) { - this.generateQuantiles = generateQuantiles; - return this; - } - } - - @OpInputsMetadata( - outputsClass = BoostedTreesQuantileStreamResourceFlush.class - ) - public static class Inputs extends RawOpInputs { - /** - * resource handle referring to a QuantileStreamResource. - */ - public final Operand quantileStreamResourceHandle; - - /** - * int; approximate number of buckets unless using generate_quantiles. - */ - public final Operand numBuckets; - - /** - * bool; If True, the output will be the num_quantiles for each stream where the ith - * entry is the ith quantile of the input with an approximation error of epsilon. - * Duplicate values may be present. - * If False, the output will be the points in the histogram that we got which roughly - * translates to 1/epsilon boundaries and without any duplicates. - * Default to False. - */ - public final boolean generateQuantiles; - - public Inputs(GraphOperation op) { - super(new BoostedTreesQuantileStreamResourceFlush(op), op, Arrays.asList("generate_quantiles")); - int inputIndex = 0; - quantileStreamResourceHandle = (Operand) op.input(inputIndex++); - numBuckets = (Operand) op.input(inputIndex++); - generateQuantiles = op.attributes().getAttrBool("generate_quantiles"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceGetBucketBoundaries.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceGetBucketBoundaries.java deleted file mode 100644 index 1a472902880..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceGetBucketBoundaries.java +++ /dev/null @@ -1,112 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import java.util.Iterator; -import java.util.List; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.family.TType; - -/** - * Generate the bucket boundaries for each feature based on accumulated summaries. - * An op that returns a list of float tensors for a quantile stream resource. Each - * tensor is Rank 1 containing bucket boundaries for a single feature. - */ -@OpMetadata( - opType = BoostedTreesQuantileStreamResourceGetBucketBoundaries.OP_NAME, - inputsClass = BoostedTreesQuantileStreamResourceGetBucketBoundaries.Inputs.class -) -public final class BoostedTreesQuantileStreamResourceGetBucketBoundaries extends RawOp implements Iterable> { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesQuantileStreamResourceGetBucketBoundaries"; - - private List> bucketBoundaries; - - @SuppressWarnings("unchecked") - public BoostedTreesQuantileStreamResourceGetBucketBoundaries(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - int bucketBoundariesLength = operation.outputListLength("bucket_boundaries"); - bucketBoundaries = Arrays.asList((Output[]) operation.outputList(outputIdx, bucketBoundariesLength)); - outputIdx += bucketBoundariesLength; - } - - /** - * Factory method to create a class wrapping a new BoostedTreesQuantileStreamResourceGetBucketBoundaries operation. - * - * @param scope current scope - * @param quantileStreamResourceHandle resource handle referring to a QuantileStreamResource. - * @param numFeatures inferred int; number of features to get bucket boundaries for. - * @return a new instance of BoostedTreesQuantileStreamResourceGetBucketBoundaries - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesQuantileStreamResourceGetBucketBoundaries create(Scope scope, - Operand quantileStreamResourceHandle, Long numFeatures) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesQuantileStreamResourceGetBucketBoundaries"); - opBuilder.addInput(quantileStreamResourceHandle.asOutput()); - opBuilder.setAttr("num_features", numFeatures); - return new BoostedTreesQuantileStreamResourceGetBucketBoundaries(opBuilder.build()); - } - - /** - * Gets bucketBoundaries. - * float; List of Rank 1 Tensors each containing the bucket boundaries for a feature. - * @return bucketBoundaries. - */ - public List> bucketBoundaries() { - return bucketBoundaries; - } - - @Override - @SuppressWarnings({"rawtypes", "unchecked"}) - public Iterator> iterator() { - return (Iterator) bucketBoundaries.iterator(); - } - - @OpInputsMetadata( - outputsClass = BoostedTreesQuantileStreamResourceGetBucketBoundaries.class - ) - public static class Inputs extends RawOpInputs { - /** - * resource handle referring to a QuantileStreamResource. - */ - public final Operand quantileStreamResourceHandle; - - public Inputs(GraphOperation op) { - super(new BoostedTreesQuantileStreamResourceGetBucketBoundaries(op), op, Arrays.asList()); - int inputIndex = 0; - quantileStreamResourceHandle = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceHandleOp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceHandleOp.java deleted file mode 100644 index bc58dbfcd15..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesQuantileStreamResourceHandleOp.java +++ /dev/null @@ -1,171 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.family.TType; - -/** - * Creates a handle to a BoostedTreesQuantileStreamResource. - */ -@OpMetadata( - opType = BoostedTreesQuantileStreamResourceHandleOp.OP_NAME, - inputsClass = BoostedTreesQuantileStreamResourceHandleOp.Inputs.class -) -public final class BoostedTreesQuantileStreamResourceHandleOp extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesQuantileStreamResourceHandleOp"; - - private Output resource; - - @SuppressWarnings("unchecked") - public BoostedTreesQuantileStreamResourceHandleOp(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - resource = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesQuantileStreamResourceHandleOp operation. - * - * @param scope current scope - * @param options carries optional attribute values - * @return a new instance of BoostedTreesQuantileStreamResourceHandleOp - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesQuantileStreamResourceHandleOp create(Scope scope, Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesQuantileStreamResourceHandleOp"); - if (options != null) { - for (Options opts : options) { - if (opts.container != null) { - opBuilder.setAttr("container", opts.container); - } - if (opts.sharedName != null) { - opBuilder.setAttr("shared_name", opts.sharedName); - } - } - } - return new BoostedTreesQuantileStreamResourceHandleOp(opBuilder.build()); - } - - /** - * Sets the container option. - * - * @param container the container option - * @return this Options instance. - */ - public static Options container(String container) { - return new Options().container(container); - } - - /** - * Sets the sharedName option. - * - * @param sharedName the sharedName option - * @return this Options instance. - */ - public static Options sharedName(String sharedName) { - return new Options().sharedName(sharedName); - } - - /** - * Gets resource. - * - * @return resource. - */ - public Output resource() { - return resource; - } - - @Override - @SuppressWarnings("unchecked") - public Output asOutput() { - return (Output) resource; - } - - /** - * Optional attributes for {@link org.tensorflow.op.estimator.BoostedTreesQuantileStreamResourceHandleOp} - */ - public static class Options { - private String container; - - private String sharedName; - - private Options() { - } - - /** - * Sets the container option. - * - * @param container the container option - * @return this Options instance. - */ - public Options container(String container) { - this.container = container; - return this; - } - - /** - * Sets the sharedName option. - * - * @param sharedName the sharedName option - * @return this Options instance. - */ - public Options sharedName(String sharedName) { - this.sharedName = sharedName; - return this; - } - } - - @OpInputsMetadata( - outputsClass = BoostedTreesQuantileStreamResourceHandleOp.class - ) - public static class Inputs extends RawOpInputs { - /** - * The container attribute - */ - public final String container; - - /** - * The sharedName attribute - */ - public final String sharedName; - - public Inputs(GraphOperation op) { - super(new BoostedTreesQuantileStreamResourceHandleOp(op), op, Arrays.asList("container", "shared_name")); - int inputIndex = 0; - container = op.attributes().getAttrString("container"); - sharedName = op.attributes().getAttrString("shared_name"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSerializeEnsemble.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSerializeEnsemble.java deleted file mode 100644 index 65ed6774a92..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSerializeEnsemble.java +++ /dev/null @@ -1,110 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TInt64; -import org.tensorflow.types.TString; -import org.tensorflow.types.family.TType; - -/** - * Serializes the tree ensemble to a proto. - */ -@OpMetadata( - opType = BoostedTreesSerializeEnsemble.OP_NAME, - inputsClass = BoostedTreesSerializeEnsemble.Inputs.class -) -public final class BoostedTreesSerializeEnsemble extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesSerializeEnsemble"; - - private Output stampToken; - - private Output treeEnsembleSerialized; - - public BoostedTreesSerializeEnsemble(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - stampToken = operation.output(outputIdx++); - treeEnsembleSerialized = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesSerializeEnsemble operation. - * - * @param scope current scope - * @param treeEnsembleHandle Handle to the tree ensemble. - * @return a new instance of BoostedTreesSerializeEnsemble - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesSerializeEnsemble create(Scope scope, - Operand treeEnsembleHandle) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesSerializeEnsemble"); - opBuilder.addInput(treeEnsembleHandle.asOutput()); - return new BoostedTreesSerializeEnsemble(opBuilder.build()); - } - - /** - * Gets stampToken. - * Stamp token of the tree ensemble resource. - * @return stampToken. - */ - public Output stampToken() { - return stampToken; - } - - /** - * Gets treeEnsembleSerialized. - * Serialized proto of the ensemble. - * @return treeEnsembleSerialized. - */ - public Output treeEnsembleSerialized() { - return treeEnsembleSerialized; - } - - @OpInputsMetadata( - outputsClass = BoostedTreesSerializeEnsemble.class - ) - public static class Inputs extends RawOpInputs { - /** - * Handle to the tree ensemble. - */ - public final Operand treeEnsembleHandle; - - public Inputs(GraphOperation op) { - super(new BoostedTreesSerializeEnsemble(op), op, Arrays.asList()); - int inputIndex = 0; - treeEnsembleHandle = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseAggregateStats.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseAggregateStats.java deleted file mode 100644 index 18f56f131e4..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseAggregateStats.java +++ /dev/null @@ -1,199 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; - -/** - * Aggregates the summary of accumulated stats for the batch. - * The summary stats contains gradients and hessians accumulated for each node, bucket and dimension id. - */ -@OpMetadata( - opType = BoostedTreesSparseAggregateStats.OP_NAME, - inputsClass = BoostedTreesSparseAggregateStats.Inputs.class -) -public final class BoostedTreesSparseAggregateStats extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesSparseAggregateStats"; - - private Output statsSummaryIndices; - - private Output statsSummaryValues; - - private Output statsSummaryShape; - - public BoostedTreesSparseAggregateStats(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - statsSummaryIndices = operation.output(outputIdx++); - statsSummaryValues = operation.output(outputIdx++); - statsSummaryShape = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesSparseAggregateStats operation. - * - * @param scope current scope - * @param nodeIds int32; Rank 1 Tensor containing node ids for each example, shape [batch_size]. - * @param gradients float32; Rank 2 Tensor (shape=[batch_size, logits_dimension]) with gradients for each example. - * @param hessians float32; Rank 2 Tensor (shape=[batch_size, hessian_dimension]) with hessians for each example. - * @param featureIndices int32; Rank 2 indices of feature sparse Tensors (shape=[number of sparse entries, 2]). - * Number of sparse entries across all instances from the batch. The first value is - * the index of the instance, the second is dimension of the feature. The second axis - * can only have 2 values, i.e., the input dense version of Tensor can only be matrix. - * @param featureValues int32; Rank 1 values of feature sparse Tensors (shape=[number of sparse entries]). - * Number of sparse entries across all instances from the batch. The first value is - * the index of the instance, the second is dimension of the feature. - * @param featureShape int32; Rank 1 dense shape of feature sparse Tensors (shape=[2]). - * The first axis can only have 2 values, [batch_size, feature_dimension]. - * @param maxSplits int; the maximum number of splits possible in the whole tree. - * @param numBuckets int; equals to the maximum possible value of bucketized feature + 1. - * @return a new instance of BoostedTreesSparseAggregateStats - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesSparseAggregateStats create(Scope scope, Operand nodeIds, - Operand gradients, Operand hessians, Operand featureIndices, - Operand featureValues, Operand featureShape, Long maxSplits, - Long numBuckets) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesSparseAggregateStats"); - opBuilder.addInput(nodeIds.asOutput()); - opBuilder.addInput(gradients.asOutput()); - opBuilder.addInput(hessians.asOutput()); - opBuilder.addInput(featureIndices.asOutput()); - opBuilder.addInput(featureValues.asOutput()); - opBuilder.addInput(featureShape.asOutput()); - opBuilder.setAttr("max_splits", maxSplits); - opBuilder.setAttr("num_buckets", numBuckets); - return new BoostedTreesSparseAggregateStats(opBuilder.build()); - } - - /** - * Gets statsSummaryIndices. - * int32; Rank 2 indices of summary sparse Tensors (shape=[number of non zero statistics, 4]) - * The second axis can only be 4 including node id, feature dimension, bucket id, and statistics_dimension. - * statistics_dimension = logits_dimension + hessian_dimension. - * @return statsSummaryIndices. - */ - public Output statsSummaryIndices() { - return statsSummaryIndices; - } - - /** - * Gets statsSummaryValues. - * output Rank 1 Tensor (shape=[number of non zero statistics]) - * @return statsSummaryValues. - */ - public Output statsSummaryValues() { - return statsSummaryValues; - } - - /** - * Gets statsSummaryShape. - * output Rank 1 Tensor (shape=[4]) - * The tensor has following 4 values: [max_splits, feature_dimension, num_buckets, statistics_dimension], - * where statistics_dimension = gradient_dimension + hessian_dimension. gradient_dimension - * is the same as label_dimension, i.e., the output space. hessian_dimension can be the same - * as logits dimension when diagonal hessian is used, or label_dimension^2 when full - * hessian is used. - * @return statsSummaryShape. - */ - public Output statsSummaryShape() { - return statsSummaryShape; - } - - @OpInputsMetadata( - outputsClass = BoostedTreesSparseAggregateStats.class - ) - public static class Inputs extends RawOpInputs { - /** - * int32; Rank 1 Tensor containing node ids for each example, shape [batch_size]. - */ - public final Operand nodeIds; - - /** - * float32; Rank 2 Tensor (shape=[batch_size, logits_dimension]) with gradients for each example. - */ - public final Operand gradients; - - /** - * float32; Rank 2 Tensor (shape=[batch_size, hessian_dimension]) with hessians for each example. - */ - public final Operand hessians; - - /** - * int32; Rank 2 indices of feature sparse Tensors (shape=[number of sparse entries, 2]). - * Number of sparse entries across all instances from the batch. The first value is - * the index of the instance, the second is dimension of the feature. The second axis - * can only have 2 values, i.e., the input dense version of Tensor can only be matrix. - */ - public final Operand featureIndices; - - /** - * int32; Rank 1 values of feature sparse Tensors (shape=[number of sparse entries]). - * Number of sparse entries across all instances from the batch. The first value is - * the index of the instance, the second is dimension of the feature. - */ - public final Operand featureValues; - - /** - * int32; Rank 1 dense shape of feature sparse Tensors (shape=[2]). - * The first axis can only have 2 values, [batch_size, feature_dimension]. - */ - public final Operand featureShape; - - /** - * int; the maximum number of splits possible in the whole tree. - */ - public final long maxSplits; - - /** - * int; equals to the maximum possible value of bucketized feature + 1. - */ - public final long numBuckets; - - public Inputs(GraphOperation op) { - super(new BoostedTreesSparseAggregateStats(op), op, Arrays.asList("max_splits", "num_buckets")); - int inputIndex = 0; - nodeIds = (Operand) op.input(inputIndex++); - gradients = (Operand) op.input(inputIndex++); - hessians = (Operand) op.input(inputIndex++); - featureIndices = (Operand) op.input(inputIndex++); - featureValues = (Operand) op.input(inputIndex++); - featureShape = (Operand) op.input(inputIndex++); - maxSplits = op.attributes().getAttrInt("max_splits"); - numBuckets = op.attributes().getAttrInt("num_buckets"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseCalculateBestFeatureSplit.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseCalculateBestFeatureSplit.java deleted file mode 100644 index 3803c2aa456..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesSparseCalculateBestFeatureSplit.java +++ /dev/null @@ -1,290 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; -import org.tensorflow.types.TString; - -/** - * Calculates gains for each feature and returns the best possible split information for the feature. - * The split information is the best threshold (bucket id), gains and left/right node contributions per node for each feature. - *

It is possible that not all nodes can be split on each feature. Hence, the list of possible nodes can differ between the features. Therefore, we return {@code node_ids_list} for each feature, containing the list of nodes that this feature can be used to split. - *

In this manner, the output is the best split per features and per node, so that it needs to be combined later to produce the best split for each node (among all possible features). - *

The output shapes are compatible in a way that the first dimension of all tensors are the same and equal to the number of possible split nodes for each feature. - */ -@OpMetadata( - opType = BoostedTreesSparseCalculateBestFeatureSplit.OP_NAME, - inputsClass = BoostedTreesSparseCalculateBestFeatureSplit.Inputs.class -) -public final class BoostedTreesSparseCalculateBestFeatureSplit extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesSparseCalculateBestFeatureSplit"; - - private Output nodeIds; - - private Output gains; - - private Output featureDimensions; - - private Output thresholds; - - private Output leftNodeContribs; - - private Output rightNodeContribs; - - private Output splitWithDefaultDirections; - - public BoostedTreesSparseCalculateBestFeatureSplit(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - nodeIds = operation.output(outputIdx++); - gains = operation.output(outputIdx++); - featureDimensions = operation.output(outputIdx++); - thresholds = operation.output(outputIdx++); - leftNodeContribs = operation.output(outputIdx++); - rightNodeContribs = operation.output(outputIdx++); - splitWithDefaultDirections = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesSparseCalculateBestFeatureSplit operation. - * - * @param scope current scope - * @param nodeIdRange A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). - * @param statsSummaryIndices A Rank 2 int64 tensor of dense shape [N, 4] (N specifies the number of non-zero values) for accumulated stats summary (gradient/hessian) per node per bucket for each feature. The second dimension contains node id, feature dimension, bucket id, and stats dim. - * stats dim is the sum of logits dimension and hessian dimension, hessian dimension can either be logits dimension if diagonal hessian is used, or logits dimension^2 if full hessian is used. - * @param statsSummaryValues A Rank 1 float tensor of dense shape [N] (N specifies the number of non-zero values), which supplies the values for each element in summary_indices. - * @param statsSummaryShape A Rank 1 float tensor of dense shape [4], which specifies the dense shape of the sparse tensor, which is [num tree nodes, feature dimensions, num buckets, stats dim]. - * @param l1 l1 regularization factor on leaf weights, per instance based. - * @param l2 l2 regularization factor on leaf weights, per instance based. - * @param treeComplexity adjustment to the gain, per leaf based. - * @param minNodeWeight minimum avg of hessians in a node before required for the node to be considered for splitting. - * @param logitsDimension The dimension of logit, i.e., number of classes. - * @param options carries optional attribute values - * @return a new instance of BoostedTreesSparseCalculateBestFeatureSplit - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesSparseCalculateBestFeatureSplit create(Scope scope, - Operand nodeIdRange, Operand statsSummaryIndices, - Operand statsSummaryValues, Operand statsSummaryShape, Operand l1, - Operand l2, Operand treeComplexity, Operand minNodeWeight, - Long logitsDimension, Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesSparseCalculateBestFeatureSplit"); - opBuilder.addInput(nodeIdRange.asOutput()); - opBuilder.addInput(statsSummaryIndices.asOutput()); - opBuilder.addInput(statsSummaryValues.asOutput()); - opBuilder.addInput(statsSummaryShape.asOutput()); - opBuilder.addInput(l1.asOutput()); - opBuilder.addInput(l2.asOutput()); - opBuilder.addInput(treeComplexity.asOutput()); - opBuilder.addInput(minNodeWeight.asOutput()); - opBuilder.setAttr("logits_dimension", logitsDimension); - if (options != null) { - for (Options opts : options) { - if (opts.splitType != null) { - opBuilder.setAttr("split_type", opts.splitType); - } - } - } - return new BoostedTreesSparseCalculateBestFeatureSplit(opBuilder.build()); - } - - /** - * Sets the splitType option. - * - * @param splitType A string indicating if this Op should perform inequality split or equality split. - * @return this Options instance. - */ - public static Options splitType(String splitType) { - return new Options().splitType(splitType); - } - - /** - * Gets nodeIds. - * A Rank 1 tensor indicating possible node ids that can be split. - * @return nodeIds. - */ - public Output nodeIds() { - return nodeIds; - } - - /** - * Gets gains. - * A Rank 1 tensor indicating the best gains to split each node. - * @return gains. - */ - public Output gains() { - return gains; - } - - /** - * Gets featureDimensions. - * A Rank 1 tensor indicating the best feature dimension for each feature to split for each node. - * @return featureDimensions. - */ - public Output featureDimensions() { - return featureDimensions; - } - - /** - * Gets thresholds. - * A Rank 1 tensor indicating the bucket id to compare with (as a threshold) for split in each node. - * @return thresholds. - */ - public Output thresholds() { - return thresholds; - } - - /** - * Gets leftNodeContribs. - * A Rank 2 tensor indicating the contribution of the left nodes when branching from parent nodes to the left direction by the given threshold for each feature. - * This value will be used to make the left node value by adding to the parent node value. Second dimension size is logits dimension. - * @return leftNodeContribs. - */ - public Output leftNodeContribs() { - return leftNodeContribs; - } - - /** - * Gets rightNodeContribs. - * A Rank 2 tensor, with the same shape/conditions as left_node_contribs_list, but just that the value is for the right node. - * @return rightNodeContribs. - */ - public Output rightNodeContribs() { - return rightNodeContribs; - } - - /** - * Gets splitWithDefaultDirections. - * A Rank 1 tensor indicating which direction to go if data is missing. - * Inequality with default left returns 0, inequality with default right returns 1, equality with default right returns 2. - * @return splitWithDefaultDirections. - */ - public Output splitWithDefaultDirections() { - return splitWithDefaultDirections; - } - - /** - * Optional attributes for {@link org.tensorflow.op.estimator.BoostedTreesSparseCalculateBestFeatureSplit} - */ - public static class Options { - private String splitType; - - private Options() { - } - - /** - * Sets the splitType option. - * - * @param splitType A string indicating if this Op should perform inequality split or equality split. - * @return this Options instance. - */ - public Options splitType(String splitType) { - this.splitType = splitType; - return this; - } - } - - @OpInputsMetadata( - outputsClass = BoostedTreesSparseCalculateBestFeatureSplit.class - ) - public static class Inputs extends RawOpInputs { - /** - * A Rank 1 tensor (shape=[2]) to specify the range [first, last) of node ids to process within {@code stats_summary_list}. The nodes are iterated between the two nodes specified by the tensor, as like {@code for node_id in range(node_id_range[0], node_id_range[1])} (Note that the last index node_id_range[1] is exclusive). - */ - public final Operand nodeIdRange; - - /** - * A Rank 2 int64 tensor of dense shape [N, 4] (N specifies the number of non-zero values) for accumulated stats summary (gradient/hessian) per node per bucket for each feature. The second dimension contains node id, feature dimension, bucket id, and stats dim. - * stats dim is the sum of logits dimension and hessian dimension, hessian dimension can either be logits dimension if diagonal hessian is used, or logits dimension^2 if full hessian is used. - */ - public final Operand statsSummaryIndices; - - /** - * A Rank 1 float tensor of dense shape [N] (N specifies the number of non-zero values), which supplies the values for each element in summary_indices. - */ - public final Operand statsSummaryValues; - - /** - * A Rank 1 float tensor of dense shape [4], which specifies the dense shape of the sparse tensor, which is [num tree nodes, feature dimensions, num buckets, stats dim]. - */ - public final Operand statsSummaryShape; - - /** - * l1 regularization factor on leaf weights, per instance based. - */ - public final Operand l1; - - /** - * l2 regularization factor on leaf weights, per instance based. - */ - public final Operand l2; - - /** - * adjustment to the gain, per leaf based. - */ - public final Operand treeComplexity; - - /** - * minimum avg of hessians in a node before required for the node to be considered for splitting. - */ - public final Operand minNodeWeight; - - /** - * The dimension of logit, i.e., number of classes. - */ - public final long logitsDimension; - - /** - * A string indicating if this Op should perform inequality split or equality split. - */ - public final String splitType; - - public Inputs(GraphOperation op) { - super(new BoostedTreesSparseCalculateBestFeatureSplit(op), op, Arrays.asList("logits_dimension", "split_type")); - int inputIndex = 0; - nodeIdRange = (Operand) op.input(inputIndex++); - statsSummaryIndices = (Operand) op.input(inputIndex++); - statsSummaryValues = (Operand) op.input(inputIndex++); - statsSummaryShape = (Operand) op.input(inputIndex++); - l1 = (Operand) op.input(inputIndex++); - l2 = (Operand) op.input(inputIndex++); - treeComplexity = (Operand) op.input(inputIndex++); - minNodeWeight = (Operand) op.input(inputIndex++); - logitsDimension = op.attributes().getAttrInt("logits_dimension"); - splitType = op.attributes().getAttrString("split_type"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesTrainingPredict.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesTrainingPredict.java deleted file mode 100644 index 8fb083ec5e5..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesTrainingPredict.java +++ /dev/null @@ -1,171 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; -import org.tensorflow.types.family.TType; - -/** - * Runs multiple additive regression ensemble predictors on input instances and - * computes the update to cached logits. It is designed to be used during training. - * It traverses the trees starting from cached tree id and cached node id and - * calculates the updates to be pushed to the cache. - */ -@OpMetadata( - opType = BoostedTreesTrainingPredict.OP_NAME, - inputsClass = BoostedTreesTrainingPredict.Inputs.class -) -public final class BoostedTreesTrainingPredict extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesTrainingPredict"; - - private Output partialLogits; - - private Output treeIds; - - private Output nodeIds; - - public BoostedTreesTrainingPredict(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - partialLogits = operation.output(outputIdx++); - treeIds = operation.output(outputIdx++); - nodeIds = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesTrainingPredict operation. - * - * @param scope current scope - * @param treeEnsembleHandle The treeEnsembleHandle value - * @param cachedTreeIds Rank 1 Tensor containing cached tree ids which is the starting - * tree of prediction. - * @param cachedNodeIds Rank 1 Tensor containing cached node id which is the starting - * node of prediction. - * @param bucketizedFeatures A list of rank 1 Tensors containing bucket id for each - * feature. - * @param logitsDimension scalar, dimension of the logits, to be used for partial logits - * shape. - * @return a new instance of BoostedTreesTrainingPredict - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesTrainingPredict create(Scope scope, - Operand treeEnsembleHandle, Operand cachedTreeIds, - Operand cachedNodeIds, Iterable> bucketizedFeatures, - Long logitsDimension) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesTrainingPredict"); - opBuilder.addInput(treeEnsembleHandle.asOutput()); - opBuilder.addInput(cachedTreeIds.asOutput()); - opBuilder.addInput(cachedNodeIds.asOutput()); - opBuilder.addInputList(Operands.asOutputs(bucketizedFeatures)); - opBuilder.setAttr("logits_dimension", logitsDimension); - return new BoostedTreesTrainingPredict(opBuilder.build()); - } - - /** - * Gets partialLogits. - * Rank 2 Tensor containing logits update (with respect to cached - * values stored) for each example. - * @return partialLogits. - */ - public Output partialLogits() { - return partialLogits; - } - - /** - * Gets treeIds. - * Rank 1 Tensor containing new tree ids for each example. - * @return treeIds. - */ - public Output treeIds() { - return treeIds; - } - - /** - * Gets nodeIds. - * Rank 1 Tensor containing new node ids in the new tree_ids. - * @return nodeIds. - */ - public Output nodeIds() { - return nodeIds; - } - - @OpInputsMetadata( - outputsClass = BoostedTreesTrainingPredict.class - ) - public static class Inputs extends RawOpInputs { - /** - * The treeEnsembleHandle input - */ - public final Operand treeEnsembleHandle; - - /** - * Rank 1 Tensor containing cached tree ids which is the starting - * tree of prediction. - */ - public final Operand cachedTreeIds; - - /** - * Rank 1 Tensor containing cached node id which is the starting - * node of prediction. - */ - public final Operand cachedNodeIds; - - /** - * A list of rank 1 Tensors containing bucket id for each - * feature. - */ - public final Iterable> bucketizedFeatures; - - /** - * scalar, dimension of the logits, to be used for partial logits - * shape. - */ - public final long logitsDimension; - - public Inputs(GraphOperation op) { - super(new BoostedTreesTrainingPredict(op), op, Arrays.asList("logits_dimension")); - int inputIndex = 0; - treeEnsembleHandle = (Operand) op.input(inputIndex++); - cachedTreeIds = (Operand) op.input(inputIndex++); - cachedNodeIds = (Operand) op.input(inputIndex++); - int bucketizedFeaturesLength = op.inputListLength("bucketized_features"); - bucketizedFeatures = Arrays.asList((Operand[]) op.inputList(inputIndex, bucketizedFeaturesLength)); - inputIndex += bucketizedFeaturesLength; - logitsDimension = op.attributes().getAttrInt("logits_dimension"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsemble.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsemble.java deleted file mode 100644 index df6017e180b..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsemble.java +++ /dev/null @@ -1,188 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; -import org.tensorflow.types.family.TType; - -/** - * Updates the tree ensemble by either adding a layer to the last tree being grown - * or by starting a new tree. - */ -@OpMetadata( - opType = BoostedTreesUpdateEnsemble.OP_NAME, - inputsClass = BoostedTreesUpdateEnsemble.Inputs.class -) -public final class BoostedTreesUpdateEnsemble extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesUpdateEnsemble"; - - public BoostedTreesUpdateEnsemble(Operation operation) { - super(operation, OP_NAME); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesUpdateEnsemble operation. - * - * @param scope current scope - * @param treeEnsembleHandle Handle to the ensemble variable. - * @param featureIds Rank 1 tensor with ids for each feature. This is the real id of - * the feature that will be used in the split. - * @param nodeIds List of rank 1 tensors representing the nodes for which this feature - * has a split. - * @param gains List of rank 1 tensors representing the gains for each of the feature's - * split. - * @param thresholds List of rank 1 tensors representing the thesholds for each of the - * feature's split. - * @param leftNodeContribs List of rank 2 tensors with left leaf contribs for each of - * the feature's splits. Will be added to the previous node values to constitute - * the values of the left nodes. - * @param rightNodeContribs List of rank 2 tensors with right leaf contribs for each - * of the feature's splits. Will be added to the previous node values to constitute - * the values of the right nodes. - * @param maxDepth Max depth of the tree to build. - * @param learningRate shrinkage const for each new tree. - * @param pruningMode 0-No pruning, 1-Pre-pruning, 2-Post-pruning. - * @return a new instance of BoostedTreesUpdateEnsemble - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesUpdateEnsemble create(Scope scope, - Operand treeEnsembleHandle, Operand featureIds, - Iterable> nodeIds, Iterable> gains, - Iterable> thresholds, Iterable> leftNodeContribs, - Iterable> rightNodeContribs, Operand maxDepth, - Operand learningRate, Long pruningMode) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesUpdateEnsemble"); - opBuilder.addInput(treeEnsembleHandle.asOutput()); - opBuilder.addInput(featureIds.asOutput()); - opBuilder.addInputList(Operands.asOutputs(nodeIds)); - opBuilder.addInputList(Operands.asOutputs(gains)); - opBuilder.addInputList(Operands.asOutputs(thresholds)); - opBuilder.addInputList(Operands.asOutputs(leftNodeContribs)); - opBuilder.addInputList(Operands.asOutputs(rightNodeContribs)); - opBuilder.addInput(maxDepth.asOutput()); - opBuilder.addInput(learningRate.asOutput()); - opBuilder.setAttr("pruning_mode", pruningMode); - return new BoostedTreesUpdateEnsemble(opBuilder.build()); - } - - @OpInputsMetadata( - outputsClass = BoostedTreesUpdateEnsemble.class - ) - public static class Inputs extends RawOpInputs { - /** - * Handle to the ensemble variable. - */ - public final Operand treeEnsembleHandle; - - /** - * Rank 1 tensor with ids for each feature. This is the real id of - * the feature that will be used in the split. - */ - public final Operand featureIds; - - /** - * List of rank 1 tensors representing the nodes for which this feature - * has a split. - */ - public final Iterable> nodeIds; - - /** - * List of rank 1 tensors representing the gains for each of the feature's - * split. - */ - public final Iterable> gains; - - /** - * List of rank 1 tensors representing the thesholds for each of the - * feature's split. - */ - public final Iterable> thresholds; - - /** - * List of rank 2 tensors with left leaf contribs for each of - * the feature's splits. Will be added to the previous node values to constitute - * the values of the left nodes. - */ - public final Iterable> leftNodeContribs; - - /** - * List of rank 2 tensors with right leaf contribs for each - * of the feature's splits. Will be added to the previous node values to constitute - * the values of the right nodes. - */ - public final Iterable> rightNodeContribs; - - /** - * Max depth of the tree to build. - */ - public final Operand maxDepth; - - /** - * shrinkage const for each new tree. - */ - public final Operand learningRate; - - /** - * 0-No pruning, 1-Pre-pruning, 2-Post-pruning. - */ - public final long pruningMode; - - public Inputs(GraphOperation op) { - super(new BoostedTreesUpdateEnsemble(op), op, Arrays.asList("pruning_mode")); - int inputIndex = 0; - treeEnsembleHandle = (Operand) op.input(inputIndex++); - featureIds = (Operand) op.input(inputIndex++); - int nodeIdsLength = op.inputListLength("node_ids"); - nodeIds = Arrays.asList((Operand[]) op.inputList(inputIndex, nodeIdsLength)); - inputIndex += nodeIdsLength; - int gainsLength = op.inputListLength("gains"); - gains = Arrays.asList((Operand[]) op.inputList(inputIndex, gainsLength)); - inputIndex += gainsLength; - int thresholdsLength = op.inputListLength("thresholds"); - thresholds = Arrays.asList((Operand[]) op.inputList(inputIndex, thresholdsLength)); - inputIndex += thresholdsLength; - int leftNodeContribsLength = op.inputListLength("left_node_contribs"); - leftNodeContribs = Arrays.asList((Operand[]) op.inputList(inputIndex, leftNodeContribsLength)); - inputIndex += leftNodeContribsLength; - int rightNodeContribsLength = op.inputListLength("right_node_contribs"); - rightNodeContribs = Arrays.asList((Operand[]) op.inputList(inputIndex, rightNodeContribsLength)); - inputIndex += rightNodeContribsLength; - maxDepth = (Operand) op.input(inputIndex++); - learningRate = (Operand) op.input(inputIndex++); - pruningMode = op.attributes().getAttrInt("pruning_mode"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsembleV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsembleV2.java deleted file mode 100644 index c1c35fd932c..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/BoostedTreesUpdateEnsembleV2.java +++ /dev/null @@ -1,287 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.TInt32; -import org.tensorflow.types.TString; -import org.tensorflow.types.family.TType; - -/** - * Updates the tree ensemble by adding a layer to the last tree being grown - * or by starting a new tree. - */ -@OpMetadata( - opType = BoostedTreesUpdateEnsembleV2.OP_NAME, - inputsClass = BoostedTreesUpdateEnsembleV2.Inputs.class -) -public final class BoostedTreesUpdateEnsembleV2 extends RawOp { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "BoostedTreesUpdateEnsembleV2"; - - public BoostedTreesUpdateEnsembleV2(Operation operation) { - super(operation, OP_NAME); - } - - /** - * Factory method to create a class wrapping a new BoostedTreesUpdateEnsembleV2 operation. - * - * @param scope current scope - * @param treeEnsembleHandle Handle to the ensemble variable. - * @param featureIds Rank 1 tensor with ids for each feature. This is the real id of - * the feature that will be used in the split. - * @param dimensionIds List of rank 1 tensors representing the dimension in each feature. - * @param nodeIds List of rank 1 tensors representing the nodes for which this feature - * has a split. - * @param gains List of rank 1 tensors representing the gains for each of the feature's - * split. - * @param thresholds List of rank 1 tensors representing the thesholds for each of the - * feature's split. - * @param leftNodeContribs List of rank 2 tensors with left leaf contribs for each of - * the feature's splits. Will be added to the previous node values to constitute - * the values of the left nodes. - * @param rightNodeContribs List of rank 2 tensors with right leaf contribs for each - * of the feature's splits. Will be added to the previous node values to constitute - * the values of the right nodes. - * @param splitTypes List of rank 1 tensors representing the split type for each feature. - * @param maxDepth Max depth of the tree to build. - * @param learningRate shrinkage const for each new tree. - * @param pruningMode 0-No pruning, 1-Pre-pruning, 2-Post-pruning. - * @param options carries optional attribute values - * @return a new instance of BoostedTreesUpdateEnsembleV2 - */ - @Endpoint( - describeByClass = true - ) - public static BoostedTreesUpdateEnsembleV2 create(Scope scope, - Operand treeEnsembleHandle, Iterable> featureIds, - Iterable> dimensionIds, Iterable> nodeIds, - Iterable> gains, Iterable> thresholds, - Iterable> leftNodeContribs, Iterable> rightNodeContribs, - Iterable> splitTypes, Operand maxDepth, - Operand learningRate, Operand pruningMode, Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "BoostedTreesUpdateEnsembleV2"); - opBuilder.addInput(treeEnsembleHandle.asOutput()); - opBuilder.addInputList(Operands.asOutputs(featureIds)); - opBuilder.addInputList(Operands.asOutputs(dimensionIds)); - opBuilder.addInputList(Operands.asOutputs(nodeIds)); - opBuilder.addInputList(Operands.asOutputs(gains)); - opBuilder.addInputList(Operands.asOutputs(thresholds)); - opBuilder.addInputList(Operands.asOutputs(leftNodeContribs)); - opBuilder.addInputList(Operands.asOutputs(rightNodeContribs)); - opBuilder.addInputList(Operands.asOutputs(splitTypes)); - opBuilder.addInput(maxDepth.asOutput()); - opBuilder.addInput(learningRate.asOutput()); - opBuilder.addInput(pruningMode.asOutput()); - if (options != null) { - for (Options opts : options) { - if (opts.logitsDimension != null) { - opBuilder.setAttr("logits_dimension", opts.logitsDimension); - } - if (opts.numGroups != null) { - opBuilder.setAttr("num_groups", opts.numGroups); - } - } - } - return new BoostedTreesUpdateEnsembleV2(opBuilder.build()); - } - - /** - * Sets the logitsDimension option. - * - * @param logitsDimension scalar, dimension of the logits - * @return this Options instance. - */ - public static Options logitsDimension(Long logitsDimension) { - return new Options().logitsDimension(logitsDimension); - } - - /** - * Sets the numGroups option. - * - * @param numGroups Number of groups of split information to process, where a group contains feature - * ids that are processed together in BoostedTreesCalculateBestFeatureSplitOpV2. - * INFERRED. - * @return this Options instance. - */ - public static Options numGroups(Long numGroups) { - return new Options().numGroups(numGroups); - } - - /** - * Optional attributes for {@link org.tensorflow.op.estimator.BoostedTreesUpdateEnsembleV2} - */ - public static class Options { - private Long logitsDimension; - - private Long numGroups; - - private Options() { - } - - /** - * Sets the logitsDimension option. - * - * @param logitsDimension scalar, dimension of the logits - * @return this Options instance. - */ - public Options logitsDimension(Long logitsDimension) { - this.logitsDimension = logitsDimension; - return this; - } - - /** - * Sets the numGroups option. - * - * @param numGroups Number of groups of split information to process, where a group contains feature - * ids that are processed together in BoostedTreesCalculateBestFeatureSplitOpV2. - * INFERRED. - * @return this Options instance. - */ - public Options numGroups(Long numGroups) { - this.numGroups = numGroups; - return this; - } - } - - @OpInputsMetadata( - outputsClass = BoostedTreesUpdateEnsembleV2.class - ) - public static class Inputs extends RawOpInputs { - /** - * Handle to the ensemble variable. - */ - public final Operand treeEnsembleHandle; - - /** - * Rank 1 tensor with ids for each feature. This is the real id of - * the feature that will be used in the split. - */ - public final Iterable> featureIds; - - /** - * List of rank 1 tensors representing the dimension in each feature. - */ - public final Iterable> dimensionIds; - - /** - * List of rank 1 tensors representing the nodes for which this feature - * has a split. - */ - public final Iterable> nodeIds; - - /** - * List of rank 1 tensors representing the gains for each of the feature's - * split. - */ - public final Iterable> gains; - - /** - * List of rank 1 tensors representing the thesholds for each of the - * feature's split. - */ - public final Iterable> thresholds; - - /** - * List of rank 2 tensors with left leaf contribs for each of - * the feature's splits. Will be added to the previous node values to constitute - * the values of the left nodes. - */ - public final Iterable> leftNodeContribs; - - /** - * List of rank 2 tensors with right leaf contribs for each - * of the feature's splits. Will be added to the previous node values to constitute - * the values of the right nodes. - */ - public final Iterable> rightNodeContribs; - - /** - * List of rank 1 tensors representing the split type for each feature. - */ - public final Iterable> splitTypes; - - /** - * Max depth of the tree to build. - */ - public final Operand maxDepth; - - /** - * shrinkage const for each new tree. - */ - public final Operand learningRate; - - /** - * 0-No pruning, 1-Pre-pruning, 2-Post-pruning. - */ - public final Operand pruningMode; - - /** - * scalar, dimension of the logits - */ - public final long logitsDimension; - - public Inputs(GraphOperation op) { - super(new BoostedTreesUpdateEnsembleV2(op), op, Arrays.asList("logits_dimension")); - int inputIndex = 0; - treeEnsembleHandle = (Operand) op.input(inputIndex++); - int featureIdsLength = op.inputListLength("feature_ids"); - featureIds = Arrays.asList((Operand[]) op.inputList(inputIndex, featureIdsLength)); - inputIndex += featureIdsLength; - int dimensionIdsLength = op.inputListLength("dimension_ids"); - dimensionIds = Arrays.asList((Operand[]) op.inputList(inputIndex, dimensionIdsLength)); - inputIndex += dimensionIdsLength; - int nodeIdsLength = op.inputListLength("node_ids"); - nodeIds = Arrays.asList((Operand[]) op.inputList(inputIndex, nodeIdsLength)); - inputIndex += nodeIdsLength; - int gainsLength = op.inputListLength("gains"); - gains = Arrays.asList((Operand[]) op.inputList(inputIndex, gainsLength)); - inputIndex += gainsLength; - int thresholdsLength = op.inputListLength("thresholds"); - thresholds = Arrays.asList((Operand[]) op.inputList(inputIndex, thresholdsLength)); - inputIndex += thresholdsLength; - int leftNodeContribsLength = op.inputListLength("left_node_contribs"); - leftNodeContribs = Arrays.asList((Operand[]) op.inputList(inputIndex, leftNodeContribsLength)); - inputIndex += leftNodeContribsLength; - int rightNodeContribsLength = op.inputListLength("right_node_contribs"); - rightNodeContribs = Arrays.asList((Operand[]) op.inputList(inputIndex, rightNodeContribsLength)); - inputIndex += rightNodeContribsLength; - int splitTypesLength = op.inputListLength("split_types"); - splitTypes = Arrays.asList((Operand[]) op.inputList(inputIndex, splitTypesLength)); - inputIndex += splitTypesLength; - maxDepth = (Operand) op.input(inputIndex++); - learningRate = (Operand) op.input(inputIndex++); - pruningMode = (Operand) op.input(inputIndex++); - logitsDimension = op.attributes().getAttrInt("logits_dimension"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesEnsembleInitialized.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesEnsembleInitialized.java deleted file mode 100644 index 23ed27e107a..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesEnsembleInitialized.java +++ /dev/null @@ -1,102 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TBool; -import org.tensorflow.types.family.TType; - -/** - * Checks whether a tree ensemble has been initialized. - */ -@OpMetadata( - opType = IsBoostedTreesEnsembleInitialized.OP_NAME, - inputsClass = IsBoostedTreesEnsembleInitialized.Inputs.class -) -public final class IsBoostedTreesEnsembleInitialized extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "IsBoostedTreesEnsembleInitialized"; - - private Output isInitialized; - - public IsBoostedTreesEnsembleInitialized(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - isInitialized = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new IsBoostedTreesEnsembleInitialized operation. - * - * @param scope current scope - * @param treeEnsembleHandle Handle to the tree ensemble resource. - * @return a new instance of IsBoostedTreesEnsembleInitialized - */ - @Endpoint( - describeByClass = true - ) - public static IsBoostedTreesEnsembleInitialized create(Scope scope, - Operand treeEnsembleHandle) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "IsBoostedTreesEnsembleInitialized"); - opBuilder.addInput(treeEnsembleHandle.asOutput()); - return new IsBoostedTreesEnsembleInitialized(opBuilder.build()); - } - - /** - * Gets isInitialized. - * output boolean on whether it is initialized or not. - * @return isInitialized. - */ - public Output isInitialized() { - return isInitialized; - } - - @Override - public Output asOutput() { - return isInitialized; - } - - @OpInputsMetadata( - outputsClass = IsBoostedTreesEnsembleInitialized.class - ) - public static class Inputs extends RawOpInputs { - /** - * Handle to the tree ensemble resource. - */ - public final Operand treeEnsembleHandle; - - public Inputs(GraphOperation op) { - super(new IsBoostedTreesEnsembleInitialized(op), op, Arrays.asList()); - int inputIndex = 0; - treeEnsembleHandle = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesQuantileStreamResourceInitialized.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesQuantileStreamResourceInitialized.java deleted file mode 100644 index 6145156ddbe..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/estimator/IsBoostedTreesQuantileStreamResourceInitialized.java +++ /dev/null @@ -1,103 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.estimator; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TBool; -import org.tensorflow.types.family.TType; - -/** - * Checks whether a quantile stream has been initialized. - * An Op that checks if quantile stream resource is initialized. - */ -@OpMetadata( - opType = IsBoostedTreesQuantileStreamResourceInitialized.OP_NAME, - inputsClass = IsBoostedTreesQuantileStreamResourceInitialized.Inputs.class -) -public final class IsBoostedTreesQuantileStreamResourceInitialized extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "IsBoostedTreesQuantileStreamResourceInitialized"; - - private Output isInitialized; - - public IsBoostedTreesQuantileStreamResourceInitialized(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - isInitialized = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new IsBoostedTreesQuantileStreamResourceInitialized operation. - * - * @param scope current scope - * @param quantileStreamResourceHandle resource; The reference to quantile stream resource handle. - * @return a new instance of IsBoostedTreesQuantileStreamResourceInitialized - */ - @Endpoint( - describeByClass = true - ) - public static IsBoostedTreesQuantileStreamResourceInitialized create(Scope scope, - Operand quantileStreamResourceHandle) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "IsBoostedTreesQuantileStreamResourceInitialized"); - opBuilder.addInput(quantileStreamResourceHandle.asOutput()); - return new IsBoostedTreesQuantileStreamResourceInitialized(opBuilder.build()); - } - - /** - * Gets isInitialized. - * bool; True if the resource is initialized, False otherwise. - * @return isInitialized. - */ - public Output isInitialized() { - return isInitialized; - } - - @Override - public Output asOutput() { - return isInitialized; - } - - @OpInputsMetadata( - outputsClass = IsBoostedTreesQuantileStreamResourceInitialized.class - ) - public static class Inputs extends RawOpInputs { - /** - * resource; The reference to quantile stream resource handle. - */ - public final Operand quantileStreamResourceHandle; - - public Inputs(GraphOperation op) { - super(new IsBoostedTreesQuantileStreamResourceInitialized(op), op, Arrays.asList()); - int inputIndex = 0; - quantileStreamResourceHandle = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractGlimpse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractGlimpse.java index f86cf256d0c..ee24391b920 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractGlimpse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ExtractGlimpse.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -58,6 +59,9 @@ opType = ExtractGlimpse.OP_NAME, inputsClass = ExtractGlimpse.Inputs.class ) +@Operator( + group = "image" +) public final class ExtractGlimpse extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/GenerateBoundingBoxProposals.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/GenerateBoundingBoxProposals.java index fddbed0f2a5..9b8b8c1d7dd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/GenerateBoundingBoxProposals.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/GenerateBoundingBoxProposals.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -52,6 +53,9 @@ opType = GenerateBoundingBoxProposals.OP_NAME, inputsClass = GenerateBoundingBoxProposals.Inputs.class ) +@Operator( + group = "image" +) public final class GenerateBoundingBoxProposals extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV2.java index 715813065f8..f18c2ebd219 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV2.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -48,6 +49,9 @@ opType = ImageProjectiveTransformV2.OP_NAME, inputsClass = ImageProjectiveTransformV2.Inputs.class ) +@Operator( + group = "image" +) public final class ImageProjectiveTransformV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV3.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV3.java index 2ae29fbfa2b..9b9d948a5d9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV3.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ImageProjectiveTransformV3.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -48,6 +49,9 @@ opType = ImageProjectiveTransformV3.OP_NAME, inputsClass = ImageProjectiveTransformV3.Inputs.class ) +@Operator( + group = "image" +) public final class ImageProjectiveTransformV3 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NearestNeighbors.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NearestNeighbors.java index d986293c625..c6be173a44c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NearestNeighbors.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/NearestNeighbors.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -42,6 +43,9 @@ opType = NearestNeighbors.OP_NAME, inputsClass = NearestNeighbors.Inputs.class ) +@Operator( + group = "image" +) public final class NearestNeighbors extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubicGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubicGrad.java index 2da91473929..16d5af61802 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubicGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBicubicGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -42,6 +43,9 @@ opType = ResizeBicubicGrad.OP_NAME, inputsClass = ResizeBicubicGrad.Inputs.class ) +@Operator( + group = "image" +) public final class ResizeBicubicGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinearGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinearGrad.java index 9e8d82364a6..dbd172bfbf2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinearGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeBilinearGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -42,6 +43,9 @@ opType = ResizeBilinearGrad.OP_NAME, inputsClass = ResizeBilinearGrad.Inputs.class ) +@Operator( + group = "image" +) public final class ResizeBilinearGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighborGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighborGrad.java index 6b5bbcd2c21..485aa4ba63b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighborGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ResizeNearestNeighborGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -42,6 +43,9 @@ opType = ResizeNearestNeighborGrad.OP_NAME, inputsClass = ResizeNearestNeighborGrad.Inputs.class ) +@Operator( + group = "image" +) public final class ResizeNearestNeighborGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslateGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslateGrad.java index e7e10f7d70d..55dae2a4ae8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslateGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/image/ScaleAndTranslateGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -42,6 +43,9 @@ opType = ScaleAndTranslateGrad.OP_NAME, inputsClass = ScaleAndTranslateGrad.Inputs.class ) +@Operator( + group = "image" +) public final class ScaleAndTranslateGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DisableCopyOnRead.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DisableCopyOnRead.java index 0e64df025ff..b3117dbe119 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DisableCopyOnRead.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/DisableCopyOnRead.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -38,6 +39,9 @@ opType = DisableCopyOnRead.OP_NAME, inputsClass = DisableCopyOnRead.Inputs.class ) +@Operator( + group = "io" +) public final class DisableCopyOnRead extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalNot.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/FakeQueue.java similarity index 53% rename from tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalNot.java rename to tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/FakeQueue.java index 1da8a281e85..9000deaef1a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalNot.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/io/FakeQueue.java @@ -15,7 +15,7 @@ // This class has been generated, DO NOT EDIT! -package org.tensorflow.op.risc; +package org.tensorflow.op.io; import java.util.Arrays; import org.tensorflow.GraphOperation; @@ -29,72 +29,77 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TBool; +import org.tensorflow.op.annotation.Operator; +import org.tensorflow.types.TString; +import org.tensorflow.types.family.TType; /** - * The RiscLogicalNot operation + * Deprecated. Do not use. */ @OpMetadata( - opType = RiscLogicalNot.OP_NAME, - inputsClass = RiscLogicalNot.Inputs.class + opType = FakeQueue.OP_NAME, + inputsClass = FakeQueue.Inputs.class ) -public final class RiscLogicalNot extends RawOp implements Operand { +@Operator( + group = "io" +) +public final class FakeQueue extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine */ - public static final String OP_NAME = "RiscLogicalNot"; + public static final String OP_NAME = "FakeQueue"; - private Output z; + private Output handle; - public RiscLogicalNot(Operation operation) { + public FakeQueue(Operation operation) { super(operation, OP_NAME); int outputIdx = 0; - z = operation.output(outputIdx++); + handle = operation.output(outputIdx++); } /** - * Factory method to create a class wrapping a new RiscLogicalNot operation. + * Factory method to create a class wrapping a new FakeQueue operation. * * @param scope current scope - * @param x The x value - * @return a new instance of RiscLogicalNot + * @param resource The resource value + * @return a new instance of FakeQueue */ @Endpoint( describeByClass = true ) - public static RiscLogicalNot create(Scope scope, Operand x) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscLogicalNot"); - opBuilder.addInput(x.asOutput()); - return new RiscLogicalNot(opBuilder.build()); + public static FakeQueue create(Scope scope, Operand resource) { + OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "FakeQueue"); + opBuilder.addInput(resource.asOutput()); + return new FakeQueue(opBuilder.build()); } /** - * Gets z. + * Gets handle. * - * @return z. + * @return handle. */ - public Output z() { - return z; + public Output handle() { + return handle; } @Override - public Output asOutput() { - return z; + public Output asOutput() { + return handle; } @OpInputsMetadata( - outputsClass = RiscLogicalNot.class + outputsClass = FakeQueue.class ) - public static class Inputs extends RawOpInputs { + public static class Inputs extends RawOpInputs { /** - * The x input + * The resource input */ - public final Operand x; + public final Operand resource; public Inputs(GraphOperation op) { - super(new RiscLogicalNot(op), op, Arrays.asList()); + super(new FakeQueue(op), op, Arrays.asList()); int inputIndex = 0; - x = (Operand) op.input(inputIndex++); + resource = (Operand) op.input(inputIndex++); } } } diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandedTriangularSolve.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandedTriangularSolve.java index a5fba205939..9dc6dba4348 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandedTriangularSolve.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/BandedTriangularSolve.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +42,9 @@ opType = BandedTriangularSolve.OP_NAME, inputsClass = BandedTriangularSolve.Inputs.class ) +@Operator( + group = "linalg" +) public final class BandedTriangularSolve extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFft.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixExponential.java similarity index 65% rename from tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFft.java rename to tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixExponential.java index c8f1fe664c6..961f57037f4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFft.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixExponential.java @@ -15,7 +15,7 @@ // This class has been generated, DO NOT EDIT! -package org.tensorflow.op.risc; +package org.tensorflow.op.linalg; import java.util.Arrays; import org.tensorflow.GraphOperation; @@ -29,47 +29,51 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; /** - * The RiscFft operation + * Deprecated, use python implementation tf.linalg.matrix_exponential. * * @param data type for {@code output} output */ @OpMetadata( - opType = RiscFft.OP_NAME, - inputsClass = RiscFft.Inputs.class + opType = MatrixExponential.OP_NAME, + inputsClass = MatrixExponential.Inputs.class ) -public final class RiscFft extends RawOp implements Operand { +@Operator( + group = "linalg" +) +public final class MatrixExponential extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine */ - public static final String OP_NAME = "RiscFft"; + public static final String OP_NAME = "MatrixExponential"; private Output output; - public RiscFft(Operation operation) { + public MatrixExponential(Operation operation) { super(operation, OP_NAME); int outputIdx = 0; output = operation.output(outputIdx++); } /** - * Factory method to create a class wrapping a new RiscFft operation. + * Factory method to create a class wrapping a new MatrixExponential operation. * * @param scope current scope * @param input The input value - * @param data type for {@code RiscFft} output and operands - * @return a new instance of RiscFft + * @param data type for {@code MatrixExponential} output and operands + * @return a new instance of MatrixExponential */ @Endpoint( describeByClass = true ) - public static RiscFft create(Scope scope, Operand input) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscFft"); + public static MatrixExponential create(Scope scope, Operand input) { + OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "MatrixExponential"); opBuilder.addInput(input.asOutput()); - return new RiscFft<>(opBuilder.build()); + return new MatrixExponential<>(opBuilder.build()); } /** @@ -87,24 +91,24 @@ public Output asOutput() { } @OpInputsMetadata( - outputsClass = RiscFft.class + outputsClass = MatrixExponential.class ) - public static class Inputs extends RawOpInputs> { + public static class Inputs extends RawOpInputs> { /** * The input input */ public final Operand input; /** - * The Tcomplex attribute + * The T attribute */ - public final DataType Tcomplex; + public final DataType T; public Inputs(GraphOperation op) { - super(new RiscFft<>(op), op, Arrays.asList("Tcomplex")); + super(new MatrixExponential<>(op), op, Arrays.asList("T")); int inputIndex = 0; input = (Operand) op.input(inputIndex++); - Tcomplex = op.attributes().getAttrType("Tcomplex"); + T = op.attributes().getAttrType("T"); } } } diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixLogarithm.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixLogarithm.java index aa349aa1792..b3876d3a572 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixLogarithm.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/MatrixLogarithm.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -52,6 +53,9 @@ opType = MatrixLogarithm.OP_NAME, inputsClass = MatrixLogarithm.Inputs.class ) +@Operator( + group = "linalg" +) public final class MatrixLogarithm extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBias.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBias.java index 7d0559a981b..9e9f8fa1653 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBias.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBias.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -49,6 +50,9 @@ opType = QuantizedMatMulWithBias.OP_NAME, inputsClass = QuantizedMatMulWithBias.Inputs.class ) +@Operator( + group = "linalg" +) public final class QuantizedMatMulWithBias extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndRelu.java index c857d310fde..f8496d61bc8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndRelu.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -50,6 +51,9 @@ opType = QuantizedMatMulWithBiasAndRelu.OP_NAME, inputsClass = QuantizedMatMulWithBiasAndRelu.Inputs.class ) +@Operator( + group = "linalg" +) public final class QuantizedMatMulWithBiasAndRelu extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndReluAndRequantize.java index 035a5d89982..f2042376788 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/QuantizedMatMulWithBiasAndReluAndRequantize.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -51,6 +52,9 @@ opType = QuantizedMatMulWithBiasAndReluAndRequantize.OP_NAME, inputsClass = QuantizedMatMulWithBiasAndReluAndRequantize.Inputs.class ) +@Operator( + group = "linalg" +) public final class QuantizedMatMulWithBiasAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalMatMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalMatMul.java index e098ce81667..bd69ed483e4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalMatMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalMatMul.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = TridiagonalMatMul.OP_NAME, inputsClass = TridiagonalMatMul.Inputs.class ) +@Operator( + group = "linalg" +) public final class TridiagonalMatMul extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalSolve.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalSolve.java index 34d0b1ba4ac..57c0864ef7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalSolve.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/TridiagonalSolve.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -48,6 +49,9 @@ opType = TridiagonalSolve.OP_NAME, inputsClass = TridiagonalSolve.Inputs.class ) +@Operator( + group = "linalg" +) public final class TridiagonalSolve extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixComponents.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixComponents.java index 08bccb5d80e..27d77557bfb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixComponents.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixComponents.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -45,6 +46,9 @@ opType = CSRSparseMatrixComponents.OP_NAME, inputsClass = CSRSparseMatrixComponents.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class CSRSparseMatrixComponents extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToDense.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToDense.java index 71c35b31ad7..51bed06f6ba 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToDense.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToDense.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = CSRSparseMatrixToDense.OP_NAME, inputsClass = CSRSparseMatrixToDense.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class CSRSparseMatrixToDense extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToSparseTensor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToSparseTensor.java index 34144514ad0..5c111887894 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToSparseTensor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/CSRSparseMatrixToSparseTensor.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = CSRSparseMatrixToSparseTensor.OP_NAME, inputsClass = CSRSparseMatrixToSparseTensor.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class CSRSparseMatrixToSparseTensor extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/DenseToCSRSparseMatrix.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/DenseToCSRSparseMatrix.java index f8181163ece..dfd4cdefccc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/DenseToCSRSparseMatrix.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/DenseToCSRSparseMatrix.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -40,6 +41,9 @@ opType = DenseToCSRSparseMatrix.OP_NAME, inputsClass = DenseToCSRSparseMatrix.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class DenseToCSRSparseMatrix extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixAdd.java index 6c6b6ba285d..fab4f97ab71 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixAdd.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +42,9 @@ opType = SparseMatrixAdd.OP_NAME, inputsClass = SparseMatrixAdd.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseMatrixAdd extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMatMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMatMul.java index a4d7ed766b9..5fafe452b17 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMatMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMatMul.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -62,6 +63,9 @@ opType = SparseMatrixMatMul.OP_NAME, inputsClass = SparseMatrixMatMul.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseMatrixMatMul extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMul.java index 5bd6ea37bbf..8b5a5e779c2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixMul.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -45,6 +46,9 @@ opType = SparseMatrixMul.OP_NAME, inputsClass = SparseMatrixMul.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseMatrixMul extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixNNZ.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixNNZ.java index c6f7974ff9c..2499a1060fc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixNNZ.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixNNZ.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = SparseMatrixNNZ.OP_NAME, inputsClass = SparseMatrixNNZ.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseMatrixNNZ extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixOrderingAMD.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixOrderingAMD.java index c6144467b24..d9287ea0611 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixOrderingAMD.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixOrderingAMD.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -80,6 +81,9 @@ opType = SparseMatrixOrderingAMD.OP_NAME, inputsClass = SparseMatrixOrderingAMD.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseMatrixOrderingAMD extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmax.java index 64b3648cea4..35bd0313c79 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmax.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -45,6 +46,9 @@ opType = SparseMatrixSoftmax.OP_NAME, inputsClass = SparseMatrixSoftmax.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseMatrixSoftmax extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmaxGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmaxGrad.java index 5ea02b16ecc..0dba5334bcb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmaxGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSoftmaxGrad.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -41,6 +42,9 @@ opType = SparseMatrixSoftmaxGrad.OP_NAME, inputsClass = SparseMatrixSoftmaxGrad.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseMatrixSoftmaxGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseCholesky.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseCholesky.java index b589f323f5c..6542ed2c6cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseCholesky.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseCholesky.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -105,6 +106,9 @@ opType = SparseMatrixSparseCholesky.OP_NAME, inputsClass = SparseMatrixSparseCholesky.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseMatrixSparseCholesky extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseMatMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseMatMul.java index 7cfd0afd589..4fcde3a77bf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseMatMul.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixSparseMatMul.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -106,6 +107,9 @@ opType = SparseMatrixSparseMatMul.OP_NAME, inputsClass = SparseMatrixSparseMatMul.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseMatrixSparseMatMul extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixTranspose.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixTranspose.java index abb25dd4e2c..dc1c3bb6833 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixTranspose.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixTranspose.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = SparseMatrixTranspose.OP_NAME, inputsClass = SparseMatrixTranspose.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseMatrixTranspose extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixZeros.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixZeros.java index faede600cc5..7e4fecb5c60 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixZeros.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseMatrixZeros.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -41,6 +42,9 @@ opType = SparseMatrixZeros.OP_NAME, inputsClass = SparseMatrixZeros.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseMatrixZeros extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseTensorToCSRSparseMatrix.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseTensorToCSRSparseMatrix.java index 143ec3fd3aa..cec991e5159 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseTensorToCSRSparseMatrix.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/linalg/sparse/SparseTensorToCSRSparseMatrix.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -40,6 +41,9 @@ opType = SparseTensorToCSRSparseMatrix.OP_NAME, inputsClass = SparseTensorToCSRSparseMatrix.Inputs.class ) +@Operator( + group = "linalg.sparse" +) public final class SparseTensorToCSRSparseMatrix extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0.java index a39144d1e94..d3782706f20 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselI0.OP_NAME, inputsClass = BesselI0.Inputs.class ) +@Operator( + group = "math" +) public final class BesselI0 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0e.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0e.java index dfebad42475..eec8b3281a3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0e.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI0e.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselI0e.OP_NAME, inputsClass = BesselI0e.Inputs.class ) +@Operator( + group = "math" +) public final class BesselI0e extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1.java index dec28f14920..bb59dc19f5c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselI1.OP_NAME, inputsClass = BesselI1.Inputs.class ) +@Operator( + group = "math" +) public final class BesselI1 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1e.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1e.java index 0cf7fcd63fb..fe929e32eb1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1e.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/BesselI1e.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselI1e.OP_NAME, inputsClass = BesselI1e.Inputs.class ) +@Operator( + group = "math" +) public final class BesselI1e extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/CumulativeLogsumexp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/CumulativeLogsumexp.java index 6cf26d3e4eb..ac645ccaaab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/CumulativeLogsumexp.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/CumulativeLogsumexp.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -57,6 +58,9 @@ opType = CumulativeLogsumexp.OP_NAME, inputsClass = CumulativeLogsumexp.Inputs.class ) +@Operator( + group = "math" +) public final class CumulativeLogsumexp extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IgammaGradA.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IgammaGradA.java index 8be3c723c18..f9e7aced432 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IgammaGradA.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/IgammaGradA.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = IgammaGradA.OP_NAME, inputsClass = IgammaGradA.Inputs.class ) +@Operator( + group = "math" +) public final class IgammaGradA extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ReciprocalGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ReciprocalGrad.java index 76522727a1b..13b7b7592ab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ReciprocalGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/ReciprocalGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = ReciprocalGrad.OP_NAME, inputsClass = ReciprocalGrad.Inputs.class ) +@Operator( + group = "math" +) public final class ReciprocalGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizationRangePerChannel.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizationRangePerChannel.java index 500e112079b..a0681e950ab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizationRangePerChannel.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizationRangePerChannel.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -40,6 +41,9 @@ opType = RequantizationRangePerChannel.OP_NAME, inputsClass = RequantizationRangePerChannel.Inputs.class ) +@Operator( + group = "math" +) public final class RequantizationRangePerChannel extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizePerChannel.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizePerChannel.java index a08d758e335..c2a71d1d594 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizePerChannel.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RequantizePerChannel.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -43,6 +44,9 @@ opType = RequantizePerChannel.OP_NAME, inputsClass = RequantizePerChannel.Inputs.class ) +@Operator( + group = "math" +) public final class RequantizePerChannel extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RsqrtGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RsqrtGrad.java index 6044722f85c..f92da40a82b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RsqrtGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/RsqrtGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = RsqrtGrad.OP_NAME, inputsClass = RsqrtGrad.Inputs.class ) +@Operator( + group = "math" +) public final class RsqrtGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SigmoidGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SigmoidGrad.java index a787d25809d..8f4b7cfe45c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SigmoidGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SigmoidGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = SigmoidGrad.OP_NAME, inputsClass = SigmoidGrad.Inputs.class ) +@Operator( + group = "math" +) public final class SigmoidGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SobolSample.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SobolSample.java index 75ca95262bf..d6de1a9ada4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SobolSample.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SobolSample.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -46,6 +47,9 @@ opType = SobolSample.OP_NAME, inputsClass = SobolSample.Inputs.class ) +@Operator( + group = "math" +) public final class SobolSample extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SoftplusGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SoftplusGrad.java index c0718a4fdff..5a8445dad45 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SoftplusGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SoftplusGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = SoftplusGrad.OP_NAME, inputsClass = SoftplusGrad.Inputs.class ) +@Operator( + group = "math" +) public final class SoftplusGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SqrtGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SqrtGrad.java index 893814519ad..451143c16e4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SqrtGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/SqrtGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = SqrtGrad.OP_NAME, inputsClass = SqrtGrad.Inputs.class ) +@Operator( + group = "math" +) public final class SqrtGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TanhGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TanhGrad.java index ca83e939fe4..c638f78b3fe 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TanhGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/TanhGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = TanhGrad.OP_NAME, inputsClass = TanhGrad.Inputs.class ) +@Operator( + group = "math" +) public final class TanhGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UniformQuantizedAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UniformQuantizedAdd.java index 84c58201f70..a3f9a486aad 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UniformQuantizedAdd.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/UniformQuantizedAdd.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -58,6 +59,9 @@ opType = UniformQuantizedAdd.OP_NAME, inputsClass = UniformQuantizedAdd.Inputs.class ) +@Operator( + group = "math" +) public final class UniformQuantizedAdd extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ0.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ0.java index bda86750e13..6ef1d289c7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ0.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ0.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselJ0.OP_NAME, inputsClass = BesselJ0.Inputs.class ) +@Operator( + group = "math.special" +) public final class BesselJ0 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ1.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ1.java index f193f6ffa69..5e7718f4144 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ1.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselJ1.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselJ1.OP_NAME, inputsClass = BesselJ1.Inputs.class ) +@Operator( + group = "math.special" +) public final class BesselJ1 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0.java index 9b47bab8dc1..338b5759a10 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselK0.OP_NAME, inputsClass = BesselK0.Inputs.class ) +@Operator( + group = "math.special" +) public final class BesselK0 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0e.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0e.java index cc8c267674c..f2a01b68ba8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0e.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK0e.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselK0e.OP_NAME, inputsClass = BesselK0e.Inputs.class ) +@Operator( + group = "math.special" +) public final class BesselK0e extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1.java index 1247c91aeee..8143c8107d5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselK1.OP_NAME, inputsClass = BesselK1.Inputs.class ) +@Operator( + group = "math.special" +) public final class BesselK1 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1e.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1e.java index 578ad729543..08ea2073dab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1e.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselK1e.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselK1e.OP_NAME, inputsClass = BesselK1e.Inputs.class ) +@Operator( + group = "math.special" +) public final class BesselK1e extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY0.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY0.java index 9fda9433a0c..c82e15022db 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY0.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY0.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselY0.OP_NAME, inputsClass = BesselY0.Inputs.class ) +@Operator( + group = "math.special" +) public final class BesselY0 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY1.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY1.java index dc8ddd9700d..5b86f1987e3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY1.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/BesselY1.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = BesselY1.OP_NAME, inputsClass = BesselY1.Inputs.class ) +@Operator( + group = "math.special" +) public final class BesselY1 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Dawsn.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Dawsn.java index 0d920d5a275..045ffc0d94c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Dawsn.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Dawsn.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = Dawsn.OP_NAME, inputsClass = Dawsn.Inputs.class ) +@Operator( + group = "math.special" +) public final class Dawsn extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Expint.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Expint.java index e5dc1567219..bcdff92cb07 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Expint.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Expint.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = Expint.OP_NAME, inputsClass = Expint.Inputs.class ) +@Operator( + group = "math.special" +) public final class Expint extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelCos.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelCos.java index ec38bd4e34b..790daad9115 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelCos.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelCos.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = FresnelCos.OP_NAME, inputsClass = FresnelCos.Inputs.class ) +@Operator( + group = "math.special" +) public final class FresnelCos extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelSin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelSin.java index 6819ad7842b..a148cb42bff 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelSin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/FresnelSin.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = FresnelSin.OP_NAME, inputsClass = FresnelSin.Inputs.class ) +@Operator( + group = "math.special" +) public final class FresnelSin extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Spence.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Spence.java index 26e610a9a2c..7835a2fca79 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Spence.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/math/special/Spence.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = Spence.OP_NAME, inputsClass = Spence.Inputs.class ) +@Operator( + group = "math.special" +) public final class Spence extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPoolGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPoolGrad.java index f093b0c4bc4..9355617a8d5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPoolGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/AvgPoolGrad.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -43,6 +44,9 @@ opType = AvgPoolGrad.OP_NAME, inputsClass = AvgPoolGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class AvgPoolGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTM.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTM.java index 7afdca6853f..3363a371d20 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTM.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTM.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -62,6 +63,9 @@ opType = BlockLSTM.OP_NAME, inputsClass = BlockLSTM.Inputs.class ) +@Operator( + group = "nn" +) public final class BlockLSTM extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTMGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTMGrad.java index f65b4eba384..2684ae60017 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTMGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/BlockLSTMGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -43,6 +44,9 @@ opType = BlockLSTMGrad.OP_NAME, inputsClass = BlockLSTMGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class BlockLSTMGrad extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CTCLossV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CTCLossV2.java index d28bc428eca..f09ff645b5a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CTCLossV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CTCLossV2.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -42,6 +43,9 @@ opType = CTCLossV2.OP_NAME, inputsClass = CTCLossV2.Inputs.class ) +@Operator( + group = "nn" +) public final class CTCLossV2 extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNN.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNN.java index 60ad5093171..0525df86f45 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNN.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNN.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -79,6 +80,9 @@ opType = CudnnRNN.OP_NAME, inputsClass = CudnnRNN.Inputs.class ) +@Operator( + group = "nn" +) public final class CudnnRNN extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNBackprop.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNBackprop.java index 2e0300fb057..d76dd629918 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNBackprop.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/CudnnRNNBackprop.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -89,6 +90,9 @@ opType = CudnnRNNBackprop.OP_NAME, inputsClass = CudnnRNNBackprop.Inputs.class ) +@Operator( + group = "nn" +) public final class CudnnRNNBackprop extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/EluGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/EluGrad.java index 911c0b92978..2df99ce5c8f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/EluGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/EluGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = EluGrad.OP_NAME, inputsClass = EluGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class EluGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPoolGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPoolGrad.java index b5abcf98128..952b982148b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPoolGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalAvgPoolGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -47,6 +48,9 @@ opType = FractionalAvgPoolGrad.OP_NAME, inputsClass = FractionalAvgPoolGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class FractionalAvgPoolGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPoolGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPoolGrad.java index 29c318e3770..c77591034a8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPoolGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/FractionalMaxPoolGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -42,6 +43,9 @@ opType = FractionalMaxPoolGrad.OP_NAME, inputsClass = FractionalMaxPoolGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class FractionalMaxPoolGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCell.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCell.java index 8374083d380..413c9db45cf 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCell.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCell.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -79,6 +80,9 @@ opType = GRUBlockCell.OP_NAME, inputsClass = GRUBlockCell.Inputs.class ) +@Operator( + group = "nn" +) public final class GRUBlockCell extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCellGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCellGrad.java index 07959198356..108aa910427 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCellGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/GRUBlockCellGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -114,6 +115,9 @@ opType = GRUBlockCellGrad.OP_NAME, inputsClass = GRUBlockCellGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class GRUBlockCellGrad extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InvGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InvGrad.java index 44fb6f47765..37f66b92878 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InvGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/InvGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = InvGrad.OP_NAME, inputsClass = InvGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class InvGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/IsotonicRegression.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/IsotonicRegression.java index 25508047cf7..8936770d8b7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/IsotonicRegression.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/IsotonicRegression.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -44,6 +45,9 @@ opType = IsotonicRegression.OP_NAME, inputsClass = IsotonicRegression.Inputs.class ) +@Operator( + group = "nn" +) public final class IsotonicRegression extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCell.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCell.java index e0fe0976803..12d4402e70f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCell.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCell.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -63,6 +64,9 @@ opType = LSTMBlockCell.OP_NAME, inputsClass = LSTMBlockCell.Inputs.class ) +@Operator( + group = "nn" +) public final class LSTMBlockCell extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCellGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCellGrad.java index 7b96ccfbbc0..e22e2241718 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCellGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LSTMBlockCellGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +43,9 @@ opType = LSTMBlockCellGrad.OP_NAME, inputsClass = LSTMBlockCellGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class LSTMBlockCellGrad extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalizationGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalizationGrad.java index 2f32d0cb241..ebf451c2f9c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalizationGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/LocalResponseNormalizationGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = LocalResponseNormalizationGrad.OP_NAME, inputsClass = LocalResponseNormalizationGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class LocalResponseNormalizationGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradWithArgmax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradWithArgmax.java index e705d69475e..3f3f754b7d6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradWithArgmax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/MaxPoolGradWithArgmax.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -42,6 +43,9 @@ opType = MaxPoolGradWithArgmax.OP_NAME, inputsClass = MaxPoolGradWithArgmax.Inputs.class ) +@Operator( + group = "nn" +) public final class MaxPoolGradWithArgmax extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRelu.java index a09b4b1ccdf..4594e0401cc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRelu.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedConv2DAndRelu.OP_NAME, inputsClass = QuantizedConv2DAndRelu.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedConv2DAndRelu extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndReluAndRequantize.java index 4c39a3d850b..0104cbf9908 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndReluAndRequantize.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedConv2DAndReluAndRequantize.OP_NAME, inputsClass = QuantizedConv2DAndReluAndRequantize.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedConv2DAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRequantize.java index 52ad5aad565..5fe5999adab 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DAndRequantize.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedConv2DAndRequantize.OP_NAME, inputsClass = QuantizedConv2DAndRequantize.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedConv2DAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DPerChannel.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DPerChannel.java index bbc48628147..134449aba91 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DPerChannel.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DPerChannel.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedConv2DPerChannel.OP_NAME, inputsClass = QuantizedConv2DPerChannel.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedConv2DPerChannel extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBias.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBias.java index a1b8bba235e..27f5343c6ff 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBias.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBias.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedConv2DWithBias.OP_NAME, inputsClass = QuantizedConv2DWithBias.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedConv2DWithBias extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRelu.java index e9e3fc45325..61c9bb31b45 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRelu.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedConv2DWithBiasAndRelu.OP_NAME, inputsClass = QuantizedConv2DWithBiasAndRelu.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedConv2DWithBiasAndRelu extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndReluAndRequantize.java index 84787fb55ca..081b8ac3863 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndReluAndRequantize.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedConv2DWithBiasAndReluAndRequantize.OP_NAME, inputsClass = QuantizedConv2DWithBiasAndReluAndRequantize.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedConv2DWithBiasAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRequantize.java index 82bbd0dab73..21f4eef7826 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasAndRequantize.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedConv2DWithBiasAndRequantize.OP_NAME, inputsClass = QuantizedConv2DWithBiasAndRequantize.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedConv2DWithBiasAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.java index 38e1c2b09d3..afdd8b87219 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.OP_NAME, inputsClass = QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedConv2DWithBiasSignedSumAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndRelu.java index 0a982087a43..d92782f88bb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndRelu.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedConv2DWithBiasSumAndRelu.OP_NAME, inputsClass = QuantizedConv2DWithBiasSumAndRelu.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedConv2DWithBiasSumAndRelu extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndReluAndRequantize.java index 67a018207b2..0d9c4fab0f6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedConv2DWithBiasSumAndReluAndRequantize.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedConv2DWithBiasSumAndReluAndRequantize.OP_NAME, inputsClass = QuantizedConv2DWithBiasSumAndReluAndRequantize.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedConv2DWithBiasSumAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2D.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2D.java index 87106dc7ecd..19c05799f1f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2D.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2D.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedDepthwiseConv2D.OP_NAME, inputsClass = QuantizedDepthwiseConv2D.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedDepthwiseConv2D extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBias.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBias.java index c85c050c25b..9414fd9e015 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBias.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBias.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedDepthwiseConv2DWithBias.OP_NAME, inputsClass = QuantizedDepthwiseConv2DWithBias.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedDepthwiseConv2DWithBias extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndRelu.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndRelu.java index d85101d528e..c8d6a30445b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndRelu.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndRelu.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedDepthwiseConv2DWithBiasAndRelu.OP_NAME, inputsClass = QuantizedDepthwiseConv2DWithBiasAndRelu.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedDepthwiseConv2DWithBiasAndRelu extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.java index 2a70d8f3bd2..b23311716d2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.OP_NAME, inputsClass = QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.Inputs.class ) +@Operator( + group = "nn" +) public final class QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6Grad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6Grad.java index 34ccb4a740f..48ec9cb7037 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6Grad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/Relu6Grad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = Relu6Grad.OP_NAME, inputsClass = Relu6Grad.Inputs.class ) +@Operator( + group = "nn" +) public final class Relu6Grad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ReluGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ReluGrad.java index 15b361b3924..5e7103853f3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ReluGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/ReluGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = ReluGrad.OP_NAME, inputsClass = ReluGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class ReluGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SeluGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SeluGrad.java index a1f1f50785a..7a2e0656275 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SeluGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SeluGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = SeluGrad.OP_NAME, inputsClass = SeluGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class SeluGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftsignGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftsignGrad.java index 129e475474c..b16c933ffe0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftsignGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/SoftsignGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = SoftsignGrad.OP_NAME, inputsClass = SoftsignGrad.Inputs.class ) +@Operator( + group = "nn" +) public final class SoftsignGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/UniformQuantizedConvolution.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/UniformQuantizedConvolution.java index fbc664f3e68..bcc2dec4789 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/UniformQuantizedConvolution.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/UniformQuantizedConvolution.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -61,6 +62,9 @@ opType = UniformQuantizedConvolution.OP_NAME, inputsClass = UniformQuantizedConvolution.Inputs.class ) +@Operator( + group = "nn" +) public final class UniformQuantizedConvolution extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/UniformQuantizedConvolutionHybrid.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/UniformQuantizedConvolutionHybrid.java index 5a40bfa376f..80d0d92a223 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/UniformQuantizedConvolutionHybrid.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/nn/UniformQuantizedConvolutionHybrid.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -61,6 +62,9 @@ opType = UniformQuantizedConvolutionHybrid.OP_NAME, inputsClass = UniformQuantizedConvolutionHybrid.Inputs.class ) +@Operator( + group = "nn" +) public final class UniformQuantizedConvolutionHybrid extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndDequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndDequantize.java index bf2f1e9193c..69827ccd019 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndDequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndDequantize.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -43,6 +44,9 @@ opType = QuantizedMatMulWithBiasAndDequantize.OP_NAME, inputsClass = QuantizedMatMulWithBiasAndDequantize.Inputs.class ) +@Operator( + group = "quantization" +) public final class QuantizedMatMulWithBiasAndDequantize extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndRequantize.java index 32f81098f3c..cd48b07ac48 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/QuantizedMatMulWithBiasAndRequantize.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TNumber; @@ -43,6 +44,9 @@ opType = QuantizedMatMulWithBiasAndRequantize.OP_NAME, inputsClass = QuantizedMatMulWithBiasAndRequantize.Inputs.class ) +@Operator( + group = "quantization" +) public final class QuantizedMatMulWithBiasAndRequantize extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformDequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformDequantize.java index 6a960b8cc44..d9d5594ae49 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformDequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformDequantize.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -46,6 +47,9 @@ opType = UniformDequantize.OP_NAME, inputsClass = UniformDequantize.Inputs.class ) +@Operator( + group = "quantization" +) public final class UniformDequantize extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantize.java index 866745f64ad..fd5f44970be 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantize.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -46,6 +47,9 @@ opType = UniformQuantize.OP_NAME, inputsClass = UniformQuantize.Inputs.class ) +@Operator( + group = "quantization" +) public final class UniformQuantize extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantizedDot.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantizedDot.java index 7ad36ff574c..16768a99b22 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantizedDot.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantizedDot.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -50,6 +51,9 @@ opType = UniformQuantizedDot.OP_NAME, inputsClass = UniformQuantizedDot.Inputs.class ) +@Operator( + group = "quantization" +) public final class UniformQuantizedDot extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantizedDotHybrid.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantizedDotHybrid.java index c690ea43e4d..ed8c67f9a53 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantizedDotHybrid.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformQuantizedDotHybrid.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -49,6 +50,9 @@ opType = UniformQuantizedDotHybrid.OP_NAME, inputsClass = UniformQuantizedDotHybrid.Inputs.class ) +@Operator( + group = "quantization" +) public final class UniformQuantizedDotHybrid extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformRequantize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformRequantize.java index 8f97998d23c..c1ef14c7ae7 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformRequantize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/quantization/UniformRequantize.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -58,6 +59,9 @@ opType = UniformRequantize.OP_NAME, inputsClass = UniformRequantize.Inputs.class ) +@Operator( + group = "quantization" +) public final class UniformRequantize extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCountSparseOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCountSparseOutput.java index d060e7baa74..1e654d1665b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCountSparseOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCountSparseOutput.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -43,6 +44,9 @@ opType = RaggedCountSparseOutput.OP_NAME, inputsClass = RaggedCountSparseOutput.Inputs.class ) +@Operator( + group = "ragged" +) public final class RaggedCountSparseOutput extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCross.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCross.java index 821c0f609d4..816bdc2b4d2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCross.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedCross.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -47,6 +48,9 @@ opType = RaggedCross.OP_NAME, inputsClass = RaggedCross.Inputs.class ) +@Operator( + group = "ragged" +) public final class RaggedCross extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedGather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedGather.java index 98a29562a70..059c102f6ed 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedGather.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedGather.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -64,6 +65,9 @@ opType = RaggedGather.OP_NAME, inputsClass = RaggedGather.Inputs.class ) +@Operator( + group = "ragged" +) public final class RaggedGather extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedRange.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedRange.java index 67426f99801..52d8d2d66b9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedRange.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedRange.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -58,6 +59,9 @@ opType = RaggedRange.OP_NAME, inputsClass = RaggedRange.Inputs.class ) +@Operator( + group = "ragged" +) public final class RaggedRange extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorFromVariant.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorFromVariant.java index 3f3146fe837..677c642582f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorFromVariant.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorFromVariant.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -58,6 +59,9 @@ opType = RaggedTensorFromVariant.OP_NAME, inputsClass = RaggedTensorFromVariant.Inputs.class ) +@Operator( + group = "ragged" +) public final class RaggedTensorFromVariant extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToSparse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToSparse.java index 4472f907b24..510cab39924 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToSparse.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToSparse.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -47,6 +48,9 @@ opType = RaggedTensorToSparse.OP_NAME, inputsClass = RaggedTensorToSparse.Inputs.class ) +@Operator( + group = "ragged" +) public final class RaggedTensorToSparse extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToTensor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToTensor.java index 0c0f474e590..7a3f212b56b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToTensor.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToTensor.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -60,6 +61,9 @@ opType = RaggedTensorToTensor.OP_NAME, inputsClass = RaggedTensorToTensor.Inputs.class ) +@Operator( + group = "ragged" +) public final class RaggedTensorToTensor extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariant.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariant.java index a3d57f0c9b1..ab616b0379d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariant.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariant.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -51,6 +52,9 @@ opType = RaggedTensorToVariant.OP_NAME, inputsClass = RaggedTensorToVariant.Inputs.class ) +@Operator( + group = "ragged" +) public final class RaggedTensorToVariant extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariantGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariantGradient.java index 42836ec7aeb..d8e57336a0e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariantGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/ragged/RaggedTensorToVariantGradient.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -48,6 +49,9 @@ opType = RaggedTensorToVariantGradient.OP_NAME, inputsClass = RaggedTensorToVariantGradient.Inputs.class ) +@Operator( + group = "ragged" +) public final class RaggedTensorToVariantGradient extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousRandomSeedGenerator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousRandomSeedGenerator.java index 525be34462d..e7557a8d4c8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousRandomSeedGenerator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousRandomSeedGenerator.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = AnonymousRandomSeedGenerator.OP_NAME, inputsClass = AnonymousRandomSeedGenerator.Inputs.class ) +@Operator( + group = "random" +) public final class AnonymousRandomSeedGenerator extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousSeedGenerator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousSeedGenerator.java index 976012740cc..1344c1ad9e4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousSeedGenerator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/AnonymousSeedGenerator.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBool; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -40,6 +41,9 @@ opType = AnonymousSeedGenerator.OP_NAME, inputsClass = AnonymousSeedGenerator.Inputs.class ) +@Operator( + group = "random" +) public final class AnonymousSeedGenerator extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteRandomSeedGenerator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteRandomSeedGenerator.java index dd2caddc2f4..32b867106f9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteRandomSeedGenerator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteRandomSeedGenerator.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -37,6 +38,9 @@ opType = DeleteRandomSeedGenerator.OP_NAME, inputsClass = DeleteRandomSeedGenerator.Inputs.class ) +@Operator( + group = "random" +) public final class DeleteRandomSeedGenerator extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteSeedGenerator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteSeedGenerator.java index 81e91b1242c..463a9ba87a0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteSeedGenerator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DeleteSeedGenerator.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -37,6 +38,9 @@ opType = DeleteSeedGenerator.OP_NAME, inputsClass = DeleteSeedGenerator.Inputs.class ) +@Operator( + group = "random" +) public final class DeleteSeedGenerator extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DummySeedGenerator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DummySeedGenerator.java index 2fef5b989e7..a73f248e303 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DummySeedGenerator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/DummySeedGenerator.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -38,6 +39,9 @@ opType = DummySeedGenerator.OP_NAME, inputsClass = DummySeedGenerator.Inputs.class ) +@Operator( + group = "random" +) public final class DummySeedGenerator extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/NonDeterministicInts.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/NonDeterministicInts.java index a417ab3d6f7..6008cd03718 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/NonDeterministicInts.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/NonDeterministicInts.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -44,6 +45,9 @@ opType = NonDeterministicInts.OP_NAME, inputsClass = NonDeterministicInts.Inputs.class ) +@Operator( + group = "random" +) public final class NonDeterministicInts extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGammaGrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGammaGrad.java index 9f1e99374f9..7baaab08ee4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGammaGrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RandomGammaGrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = RandomGammaGrad.OP_NAME, inputsClass = RandomGammaGrad.Inputs.class ) +@Operator( + group = "random" +) public final class RandomGammaGrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngReadAndSkip.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngReadAndSkip.java index d54a45c272f..78b596678fa 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngReadAndSkip.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngReadAndSkip.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -45,6 +46,9 @@ opType = RngReadAndSkip.OP_NAME, inputsClass = RngReadAndSkip.Inputs.class ) +@Operator( + group = "random" +) public final class RngReadAndSkip extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngSkip.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngSkip.java index 0ecbe4d0c43..41cd0aeff19 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngSkip.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/RngSkip.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = RngSkip.OP_NAME, inputsClass = RngSkip.Inputs.class ) +@Operator( + group = "random" +) public final class RngSkip extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulTruncatedNormal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulTruncatedNormal.java index 110501e2a87..e623baabf5c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulTruncatedNormal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulTruncatedNormal.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -47,6 +48,9 @@ opType = StatefulTruncatedNormal.OP_NAME, inputsClass = StatefulTruncatedNormal.Inputs.class ) +@Operator( + group = "random" +) public final class StatefulTruncatedNormal extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniform.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniform.java index 3321a260aa3..a0e85b0458f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniform.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniform.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -46,6 +47,9 @@ opType = StatefulUniform.OP_NAME, inputsClass = StatefulUniform.Inputs.class ) +@Operator( + group = "random" +) public final class StatefulUniform extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformFullInt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformFullInt.java index 6f22d3a3ec7..a43b26418ea 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformFullInt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformFullInt.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -44,6 +45,9 @@ opType = StatefulUniformFullInt.OP_NAME, inputsClass = StatefulUniformFullInt.Inputs.class ) +@Operator( + group = "random" +) public final class StatefulUniformFullInt extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformInt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformInt.java index acf5028ed32..154f3bd2841 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformInt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatefulUniformInt.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -48,6 +49,9 @@ opType = StatefulUniformInt.OP_NAME, inputsClass = StatefulUniformInt.Inputs.class ) +@Operator( + group = "random" +) public final class StatefulUniformInt extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessParameterizedTruncatedNormal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessParameterizedTruncatedNormal.java index 9b72e0ad1b9..b10e961aab2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessParameterizedTruncatedNormal.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessParameterizedTruncatedNormal.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -41,6 +42,9 @@ opType = StatelessParameterizedTruncatedNormal.OP_NAME, inputsClass = StatelessParameterizedTruncatedNormal.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessParameterizedTruncatedNormal extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomBinomial.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomBinomial.java index 171fcd66f25..71a3cb24cf9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomBinomial.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomBinomial.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -45,6 +46,9 @@ opType = StatelessRandomBinomial.OP_NAME, inputsClass = StatelessRandomBinomial.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomBinomial extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGamma.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGamma.java index 36b8e0ffa39..e57dfcf90f6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGamma.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGamma.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -45,6 +46,9 @@ opType = StatelessRandomGamma.OP_NAME, inputsClass = StatelessRandomGamma.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomGamma extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetAlg.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetAlg.java index cc16090aef1..30e7dd10837 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetAlg.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetAlg.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; /** @@ -39,6 +40,9 @@ opType = StatelessRandomGetAlg.OP_NAME, inputsClass = StatelessRandomGetAlg.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomGetAlg extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounter.java index 23d1bcad77f..db52e5ba0d4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounter.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -41,6 +42,9 @@ opType = StatelessRandomGetKeyCounter.OP_NAME, inputsClass = StatelessRandomGetKeyCounter.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomGetKeyCounter extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounterAlg.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounterAlg.java index 212a30743b7..d98e7a1e935 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounterAlg.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomGetKeyCounterAlg.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -42,6 +43,9 @@ opType = StatelessRandomGetKeyCounterAlg.OP_NAME, inputsClass = StatelessRandomGetKeyCounterAlg.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomGetKeyCounterAlg extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormalV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormalV2.java index 9dcd8e490ea..b1e9dcb4439 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormalV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomNormalV2.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -47,6 +48,9 @@ opType = StatelessRandomNormalV2.OP_NAME, inputsClass = StatelessRandomNormalV2.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomNormalV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomPoisson.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomPoisson.java index a3a152635aa..3a55731c32d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomPoisson.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomPoisson.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = StatelessRandomPoisson.OP_NAME, inputsClass = StatelessRandomPoisson.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomPoisson extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullInt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullInt.java index 76bd5fb79ee..ef2bf5e7884 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullInt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullInt.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -44,6 +45,9 @@ opType = StatelessRandomUniformFullInt.OP_NAME, inputsClass = StatelessRandomUniformFullInt.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomUniformFullInt extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullIntV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullIntV2.java index 2d94bfc0d0e..50fb67d6fe1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullIntV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformFullIntV2.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -46,6 +47,9 @@ opType = StatelessRandomUniformFullIntV2.OP_NAME, inputsClass = StatelessRandomUniformFullIntV2.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomUniformFullIntV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformInt.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformInt.java index 335e2700a56..8bce8bc129e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformInt.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformInt.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; @@ -43,6 +44,9 @@ opType = StatelessRandomUniformInt.OP_NAME, inputsClass = StatelessRandomUniformInt.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomUniformInt extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformIntV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformIntV2.java index 93fcc9b2685..aa3e3d0de83 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformIntV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformIntV2.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -45,6 +46,9 @@ opType = StatelessRandomUniformIntV2.OP_NAME, inputsClass = StatelessRandomUniformIntV2.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomUniformIntV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformV2.java index 74f97ca2dc6..8b0e106cb95 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessRandomUniformV2.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -48,6 +49,9 @@ opType = StatelessRandomUniformV2.OP_NAME, inputsClass = StatelessRandomUniformV2.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessRandomUniformV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormalV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormalV2.java index 11685b3b2fc..6505cd06561 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormalV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/StatelessTruncatedNormalV2.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt32; @@ -49,6 +50,9 @@ opType = StatelessTruncatedNormalV2.OP_NAME, inputsClass = StatelessTruncatedNormalV2.Inputs.class ) +@Operator( + group = "random" +) public final class StatelessTruncatedNormalV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/ThreadUnsafeUnigramCandidateSampler.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/ThreadUnsafeUnigramCandidateSampler.java new file mode 100644 index 00000000000..1aff75975d1 --- /dev/null +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/ThreadUnsafeUnigramCandidateSampler.java @@ -0,0 +1,259 @@ +/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +=======================================================================*/ + +// This class has been generated, DO NOT EDIT! + +package org.tensorflow.op.random; + +import java.util.Arrays; +import org.tensorflow.GraphOperation; +import org.tensorflow.Operand; +import org.tensorflow.Operation; +import org.tensorflow.OperationBuilder; +import org.tensorflow.Output; +import org.tensorflow.op.RawOp; +import org.tensorflow.op.RawOpInputs; +import org.tensorflow.op.Scope; +import org.tensorflow.op.annotation.Endpoint; +import org.tensorflow.op.annotation.OpInputsMetadata; +import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; +import org.tensorflow.types.TFloat32; +import org.tensorflow.types.TInt64; + +/** + * Generates labels for candidate sampling with a learned unigram distribution. + * See explanations of candidate sampling and the data formats at + * go/candidate-sampling. + *

For each batch, this op picks a single set of sampled candidate labels. + *

The advantages of sampling candidates per-batch are simplicity and the + * possibility of efficient dense matrix multiplication. The disadvantage is that + * the sampled candidates must be chosen independently of the context and of the + * true labels. + */ +@OpMetadata( + opType = ThreadUnsafeUnigramCandidateSampler.OP_NAME, + inputsClass = ThreadUnsafeUnigramCandidateSampler.Inputs.class +) +@Operator( + group = "random" +) +public final class ThreadUnsafeUnigramCandidateSampler extends RawOp { + /** + * The name of this op, as known by TensorFlow core engine + */ + public static final String OP_NAME = "ThreadUnsafeUnigramCandidateSampler"; + + private Output sampledCandidates; + + private Output trueExpectedCount; + + private Output sampledExpectedCount; + + public ThreadUnsafeUnigramCandidateSampler(Operation operation) { + super(operation, OP_NAME); + int outputIdx = 0; + sampledCandidates = operation.output(outputIdx++); + trueExpectedCount = operation.output(outputIdx++); + sampledExpectedCount = operation.output(outputIdx++); + } + + /** + * Factory method to create a class wrapping a new ThreadUnsafeUnigramCandidateSampler operation. + * + * @param scope current scope + * @param trueClasses A batch_size * num_true matrix, in which each row contains the + * IDs of the num_true target_classes in the corresponding original label. + * @param numTrue Number of true labels per context. + * @param numSampled Number of candidates to randomly sample. + * @param unique If unique is true, we sample with rejection, so that all sampled + * candidates in a batch are unique. This requires some approximation to + * estimate the post-rejection sampling probabilities. + * @param rangeMax The sampler will sample integers from the interval [0, range_max). + * @param options carries optional attribute values + * @return a new instance of ThreadUnsafeUnigramCandidateSampler + */ + @Endpoint( + describeByClass = true + ) + public static ThreadUnsafeUnigramCandidateSampler create(Scope scope, Operand trueClasses, + Long numTrue, Long numSampled, Boolean unique, Long rangeMax, Options... options) { + OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "ThreadUnsafeUnigramCandidateSampler"); + opBuilder.addInput(trueClasses.asOutput()); + opBuilder.setAttr("num_true", numTrue); + opBuilder.setAttr("num_sampled", numSampled); + opBuilder.setAttr("unique", unique); + opBuilder.setAttr("range_max", rangeMax); + if (options != null) { + for (Options opts : options) { + if (opts.seed != null) { + opBuilder.setAttr("seed", opts.seed); + } + if (opts.seed2 != null) { + opBuilder.setAttr("seed2", opts.seed2); + } + } + } + return new ThreadUnsafeUnigramCandidateSampler(opBuilder.build()); + } + + /** + * Sets the seed option. + * + * @param seed If either seed or seed2 are set to be non-zero, the random number + * generator is seeded by the given seed. Otherwise, it is seeded by a + * random seed. + * @return this Options instance. + */ + public static Options seed(Long seed) { + return new Options().seed(seed); + } + + /** + * Sets the seed2 option. + * + * @param seed2 An second seed to avoid seed collision. + * @return this Options instance. + */ + public static Options seed2(Long seed2) { + return new Options().seed2(seed2); + } + + /** + * Gets sampledCandidates. + * A vector of length num_sampled, in which each element is + * the ID of a sampled candidate. + * @return sampledCandidates. + */ + public Output sampledCandidates() { + return sampledCandidates; + } + + /** + * Gets trueExpectedCount. + * A batch_size * num_true matrix, representing + * the number of times each candidate is expected to occur in a batch + * of sampled candidates. If unique=true, then this is a probability. + * @return trueExpectedCount. + */ + public Output trueExpectedCount() { + return trueExpectedCount; + } + + /** + * Gets sampledExpectedCount. + * A vector of length num_sampled, for each sampled + * candidate representing the number of times the candidate is expected + * to occur in a batch of sampled candidates. If unique=true, then this is a + * probability. + * @return sampledExpectedCount. + */ + public Output sampledExpectedCount() { + return sampledExpectedCount; + } + + /** + * Optional attributes for {@link org.tensorflow.op.random.ThreadUnsafeUnigramCandidateSampler} + */ + public static class Options { + private Long seed; + + private Long seed2; + + private Options() { + } + + /** + * Sets the seed option. + * + * @param seed If either seed or seed2 are set to be non-zero, the random number + * generator is seeded by the given seed. Otherwise, it is seeded by a + * random seed. + * @return this Options instance. + */ + public Options seed(Long seed) { + this.seed = seed; + return this; + } + + /** + * Sets the seed2 option. + * + * @param seed2 An second seed to avoid seed collision. + * @return this Options instance. + */ + public Options seed2(Long seed2) { + this.seed2 = seed2; + return this; + } + } + + @OpInputsMetadata( + outputsClass = ThreadUnsafeUnigramCandidateSampler.class + ) + public static class Inputs extends RawOpInputs { + /** + * A batch_size * num_true matrix, in which each row contains the + * IDs of the num_true target_classes in the corresponding original label. + */ + public final Operand trueClasses; + + /** + * Number of true labels per context. + */ + public final long numTrue; + + /** + * Number of candidates to randomly sample. + */ + public final long numSampled; + + /** + * If unique is true, we sample with rejection, so that all sampled + * candidates in a batch are unique. This requires some approximation to + * estimate the post-rejection sampling probabilities. + */ + public final boolean unique; + + /** + * The sampler will sample integers from the interval [0, range_max). + */ + public final long rangeMax; + + /** + * If either seed or seed2 are set to be non-zero, the random number + * generator is seeded by the given seed. Otherwise, it is seeded by a + * random seed. + */ + public final long seed; + + /** + * An second seed to avoid seed collision. + */ + public final long seed2; + + public Inputs(GraphOperation op) { + super(new ThreadUnsafeUnigramCandidateSampler(op), op, Arrays.asList("num_true", "num_sampled", "unique", "range_max", "seed", "seed2")); + int inputIndex = 0; + trueClasses = (Operand) op.input(inputIndex++); + numTrue = op.attributes().getAttrInt("num_true"); + numSampled = op.attributes().getAttrInt("num_sampled"); + unique = op.attributes().getAttrBool("unique"); + rangeMax = op.attributes().getAttrInt("range_max"); + seed = op.attributes().getAttrInt("seed"); + seed2 = op.attributes().getAttrInt("seed2"); + } + } +} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/experimental/StatelessShuffle.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/experimental/StatelessShuffle.java index 44d41784c45..5100d0ef8c6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/experimental/StatelessShuffle.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/random/experimental/StatelessShuffle.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -51,6 +52,9 @@ opType = StatelessShuffle.OP_NAME, inputsClass = StatelessShuffle.Inputs.class ) +@Operator( + group = "random.experimental" +) public final class StatelessShuffle extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAbs.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAbs.java deleted file mode 100644 index 17034cab166..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAbs.java +++ /dev/null @@ -1,110 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscAbs operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscAbs.OP_NAME, - inputsClass = RiscAbs.Inputs.class -) -public final class RiscAbs extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscAbs"; - - private Output y; - - public RiscAbs(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscAbs operation. - * - * @param scope current scope - * @param x The x value - * @param data type for {@code RiscAbs} output and operands - * @return a new instance of RiscAbs - */ - @Endpoint( - describeByClass = true - ) - public static RiscAbs create(Scope scope, Operand x) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscAbs"); - opBuilder.addInput(x.asOutput()); - return new RiscAbs<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscAbs.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscAbs<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAdd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAdd.java deleted file mode 100644 index aaa550bc070..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscAdd.java +++ /dev/null @@ -1,121 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * Returns x + y element-wise. - * NOTE: {@code risc.RiscAdd} does not supports broadcasting. - *

Given two input tensors, the {@code tf.risc_add} operation computes the sum for every element in the tensor. - *

Both input and output have a range {@code (-inf, inf)}. - * - * @param data type for {@code z} output - */ -@OpMetadata( - opType = RiscAdd.OP_NAME, - inputsClass = RiscAdd.Inputs.class -) -public final class RiscAdd extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscAdd"; - - private Output z; - - public RiscAdd(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - z = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscAdd operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @param data type for {@code RiscAdd} output and operands - * @return a new instance of RiscAdd - */ - @Endpoint( - describeByClass = true - ) - public static RiscAdd create(Scope scope, Operand x, Operand y) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscAdd"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - return new RiscAdd<>(opBuilder.build()); - } - - /** - * Gets z. - * - * @return z. - */ - public Output z() { - return z; - } - - @Override - public Output asOutput() { - return z; - } - - @OpInputsMetadata( - outputsClass = RiscAdd.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscAdd<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryArithmetic.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryArithmetic.java deleted file mode 100644 index 5ab050f4c6b..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryArithmetic.java +++ /dev/null @@ -1,127 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscBinaryArithmetic operation - * - * @param data type for {@code z} output - */ -@OpMetadata( - opType = RiscBinaryArithmetic.OP_NAME, - inputsClass = RiscBinaryArithmetic.Inputs.class -) -public final class RiscBinaryArithmetic extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscBinaryArithmetic"; - - private Output z; - - public RiscBinaryArithmetic(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - z = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscBinaryArithmetic operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @param opType The value of the opType attribute - * @param data type for {@code RiscBinaryArithmetic} output and operands - * @return a new instance of RiscBinaryArithmetic - */ - @Endpoint( - describeByClass = true - ) - public static RiscBinaryArithmetic create(Scope scope, Operand x, - Operand y, String opType) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscBinaryArithmetic"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - opBuilder.setAttr("op_type", opType); - return new RiscBinaryArithmetic<>(opBuilder.build()); - } - - /** - * Gets z. - * - * @return z. - */ - public Output z() { - return z; - } - - @Override - public Output asOutput() { - return z; - } - - @OpInputsMetadata( - outputsClass = RiscBinaryArithmetic.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - /** - * The opType attribute - */ - public final String opType; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscBinaryArithmetic<>(op), op, Arrays.asList("op_type", "T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - opType = op.attributes().getAttrString("op_type"); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryComparison.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryComparison.java deleted file mode 100644 index 5881c9bf342..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBinaryComparison.java +++ /dev/null @@ -1,126 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.TBool; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscBinaryComparison operation - */ -@OpMetadata( - opType = RiscBinaryComparison.OP_NAME, - inputsClass = RiscBinaryComparison.Inputs.class -) -public final class RiscBinaryComparison extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscBinaryComparison"; - - private Output z; - - public RiscBinaryComparison(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - z = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscBinaryComparison operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @param opType The value of the opType attribute - * @param data type for {@code RiscBinaryComparison} output and operands - * @return a new instance of RiscBinaryComparison - */ - @Endpoint( - describeByClass = true - ) - public static RiscBinaryComparison create(Scope scope, Operand x, - Operand y, String opType) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscBinaryComparison"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - opBuilder.setAttr("op_type", opType); - return new RiscBinaryComparison(opBuilder.build()); - } - - /** - * Gets z. - * - * @return z. - */ - public Output z() { - return z; - } - - @Override - public Output asOutput() { - return z; - } - - @OpInputsMetadata( - outputsClass = RiscBinaryComparison.class - ) - public static class Inputs extends RawOpInputs { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - /** - * The opType attribute - */ - public final String opType; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscBinaryComparison(op), op, Arrays.asList("op_type", "T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - opType = op.attributes().getAttrString("op_type"); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBitcast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBitcast.java deleted file mode 100644 index 89d34941e52..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBitcast.java +++ /dev/null @@ -1,120 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TType; - -/** - * The RiscBitcast operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscBitcast.OP_NAME, - inputsClass = RiscBitcast.Inputs.class -) -public final class RiscBitcast extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscBitcast"; - - private Output y; - - public RiscBitcast(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscBitcast operation. - * - * @param scope current scope - * @param x The x value - * @param DstT The value of the DstT attribute - * @param data type for {@code RiscBitcast} output and operands - * @return a new instance of RiscBitcast - */ - @Endpoint( - describeByClass = true - ) - public static RiscBitcast create(Scope scope, Operand x, - Class DstT) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscBitcast"); - opBuilder.addInput(x.asOutput()); - opBuilder.setAttr("DstT", Operands.toDataType(DstT)); - return new RiscBitcast<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscBitcast.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The SrcT attribute - */ - public final DataType SrcT; - - /** - * The DstT attribute - */ - public final DataType DstT; - - public Inputs(GraphOperation op) { - super(new RiscBitcast<>(op), op, Arrays.asList("SrcT", "DstT")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - SrcT = op.attributes().getAttrType("SrcT"); - DstT = op.attributes().getAttrType("DstT"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBroadcast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBroadcast.java deleted file mode 100644 index 873f1ddde89..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscBroadcast.java +++ /dev/null @@ -1,126 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; -import org.tensorflow.types.family.TType; - -/** - * The RiscBroadcast operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscBroadcast.OP_NAME, - inputsClass = RiscBroadcast.Inputs.class -) -public final class RiscBroadcast extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscBroadcast"; - - private Output output; - - public RiscBroadcast(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscBroadcast operation. - * - * @param scope current scope - * @param input The input value - * @param shape The shape value - * @param data type for {@code RiscBroadcast} output and operands - * @return a new instance of RiscBroadcast - */ - @Endpoint( - describeByClass = true - ) - public static RiscBroadcast create(Scope scope, Operand input, - Operand shape) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscBroadcast"); - opBuilder.addInput(input.asOutput()); - opBuilder.addInput(shape.asOutput()); - return new RiscBroadcast<>(opBuilder.build()); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscBroadcast.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The input input - */ - public final Operand input; - - /** - * The shape input - */ - public final Operand shape; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The Tidx attribute - */ - public final DataType Tidx; - - public Inputs(GraphOperation op) { - super(new RiscBroadcast<>(op), op, Arrays.asList("T", "Tidx")); - int inputIndex = 0; - input = (Operand) op.input(inputIndex++); - shape = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - Tidx = op.attributes().getAttrType("Tidx"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCast.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCast.java deleted file mode 100644 index 764a86428d8..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCast.java +++ /dev/null @@ -1,120 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TType; - -/** - * The RiscCast operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscCast.OP_NAME, - inputsClass = RiscCast.Inputs.class -) -public final class RiscCast extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscCast"; - - private Output y; - - public RiscCast(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscCast operation. - * - * @param scope current scope - * @param x The x value - * @param DstT The value of the DstT attribute - * @param data type for {@code RiscCast} output and operands - * @return a new instance of RiscCast - */ - @Endpoint( - describeByClass = true - ) - public static RiscCast create(Scope scope, Operand x, - Class DstT) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscCast"); - opBuilder.addInput(x.asOutput()); - opBuilder.setAttr("DstT", Operands.toDataType(DstT)); - return new RiscCast<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscCast.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The SrcT attribute - */ - public final DataType SrcT; - - /** - * The DstT attribute - */ - public final DataType DstT; - - public Inputs(GraphOperation op) { - super(new RiscCast<>(op), op, Arrays.asList("SrcT", "DstT")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - SrcT = op.attributes().getAttrType("SrcT"); - DstT = op.attributes().getAttrType("DstT"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCeil.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCeil.java deleted file mode 100644 index c252b4e31d2..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCeil.java +++ /dev/null @@ -1,110 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscCeil operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscCeil.OP_NAME, - inputsClass = RiscCeil.Inputs.class -) -public final class RiscCeil extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscCeil"; - - private Output y; - - public RiscCeil(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscCeil operation. - * - * @param scope current scope - * @param x The x value - * @param data type for {@code RiscCeil} output and operands - * @return a new instance of RiscCeil - */ - @Endpoint( - describeByClass = true - ) - public static RiscCeil create(Scope scope, Operand x) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscCeil"); - opBuilder.addInput(x.asOutput()); - return new RiscCeil<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscCeil.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscCeil<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConcat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConcat.java deleted file mode 100644 index 4fb947711c0..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConcat.java +++ /dev/null @@ -1,129 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; -import org.tensorflow.types.family.TType; - -/** - * The RiscConcat operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscConcat.OP_NAME, - inputsClass = RiscConcat.Inputs.class -) -public final class RiscConcat extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscConcat"; - - private Output output; - - public RiscConcat(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscConcat operation. - * - * @param scope current scope - * @param values The values value - * @param axis The axis value - * @param data type for {@code RiscConcat} output and operands - * @return a new instance of RiscConcat - */ - @Endpoint( - describeByClass = true - ) - public static RiscConcat create(Scope scope, Iterable> values, - Operand axis) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscConcat"); - opBuilder.addInputList(Operands.asOutputs(values)); - opBuilder.addInput(axis.asOutput()); - return new RiscConcat<>(opBuilder.build()); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscConcat.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The values input - */ - public final Iterable> values; - - /** - * The axis input - */ - public final Operand axis; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The Tidx attribute - */ - public final DataType Tidx; - - public Inputs(GraphOperation op) { - super(new RiscConcat<>(op), op, Arrays.asList("T", "Tidx")); - int inputIndex = 0; - int valuesLength = op.inputListLength("values"); - values = Arrays.asList((Operand[]) op.inputList(inputIndex, valuesLength)); - inputIndex += valuesLength; - axis = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - Tidx = op.attributes().getAttrType("Tidx"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCondition.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCondition.java deleted file mode 100644 index 53e2c186ba3..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCondition.java +++ /dev/null @@ -1,144 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.ConcreteFunction; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.TBool; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscCondition operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscCondition.OP_NAME, - inputsClass = RiscCondition.Inputs.class -) -public final class RiscCondition extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscCondition"; - - private Output output; - - public RiscCondition(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscCondition operation. - * - * @param scope current scope - * @param pred The pred value - * @param inputTrue The inputTrue value - * @param inputFalse The inputFalse value - * @param funcTrue The value of the funcTrue attribute - * @param funcFalse The value of the funcFalse attribute - * @param DstT The value of the DstT attribute - * @param data type for {@code RiscCondition} output and operands - * @param data type for {@code RiscCondition} output and operands - * @return a new instance of RiscCondition - */ - @Endpoint( - describeByClass = true - ) - public static RiscCondition create(Scope scope, - Operand pred, Operand inputTrue, Operand inputFalse, ConcreteFunction funcTrue, - ConcreteFunction funcFalse, Class DstT) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscCondition"); - opBuilder.addInput(pred.asOutput()); - opBuilder.addInput(inputTrue.asOutput()); - opBuilder.addInput(inputFalse.asOutput()); - opBuilder.setAttr("func_true", funcTrue); - opBuilder.setAttr("func_false", funcFalse); - opBuilder.setAttr("DstT", Operands.toDataType(DstT)); - return new RiscCondition<>(opBuilder.build()); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscCondition.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The pred input - */ - public final Operand pred; - - /** - * The inputTrue input - */ - public final Operand inputTrue; - - /** - * The inputFalse input - */ - public final Operand inputFalse; - - /** - * The SrcT attribute - */ - public final DataType SrcT; - - /** - * The DstT attribute - */ - public final DataType DstT; - - public Inputs(GraphOperation op) { - super(new RiscCondition<>(op), op, Arrays.asList("SrcT", "DstT")); - int inputIndex = 0; - pred = (Operand) op.input(inputIndex++); - inputTrue = (Operand) op.input(inputIndex++); - inputFalse = (Operand) op.input(inputIndex++); - SrcT = op.attributes().getAttrType("SrcT"); - DstT = op.attributes().getAttrType("DstT"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConv.java deleted file mode 100644 index 0b910dc463b..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscConv.java +++ /dev/null @@ -1,234 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import java.util.List; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscConv operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscConv.OP_NAME, - inputsClass = RiscConv.Inputs.class -) -public final class RiscConv extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscConv"; - - private Output output; - - public RiscConv(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscConv operation. - * - * @param scope current scope - * @param input The input value - * @param filter The filter value - * @param strides The value of the strides attribute - * @param options carries optional attribute values - * @param data type for {@code RiscConv} output and operands - * @return a new instance of RiscConv - */ - @Endpoint( - describeByClass = true - ) - public static RiscConv create(Scope scope, Operand input, - Operand filter, List strides, Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscConv"); - opBuilder.addInput(input.asOutput()); - opBuilder.addInput(filter.asOutput()); - long[] stridesArray = new long[strides.size()]; - for (int i = 0 ; i < stridesArray.length ; i++) { - stridesArray[i] = strides.get(i); - } - opBuilder.setAttr("strides", stridesArray); - if (options != null) { - for (Options opts : options) { - if (opts.dataFormat != null) { - opBuilder.setAttr("data_format", opts.dataFormat); - } - if (opts.dilations != null) { - long[] dilationsArray = new long[opts.dilations.size()]; - for (int i = 0 ; i < dilationsArray.length ; i++) { - dilationsArray[i] = opts.dilations.get(i); - } - opBuilder.setAttr("dilations", dilationsArray); - } - } - } - return new RiscConv<>(opBuilder.build()); - } - - /** - * Sets the dataFormat option. - * - * @param dataFormat the dataFormat option - * @return this Options instance. - */ - public static Options dataFormat(String dataFormat) { - return new Options().dataFormat(dataFormat); - } - - /** - * Sets the dilations option. - * - * @param dilations the dilations option - * @return this Options instance. - */ - public static Options dilations(List dilations) { - return new Options().dilations(dilations); - } - - /** - * Sets the dilations option. - * - * @param dilations the dilations option - * @return this Options instance. - */ - public static Options dilations(Long... dilations) { - return new Options().dilations(dilations); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - /** - * Optional attributes for {@link org.tensorflow.op.risc.RiscConv} - */ - public static class Options { - private String dataFormat; - - private List dilations; - - private Options() { - } - - /** - * Sets the dataFormat option. - * - * @param dataFormat the dataFormat option - * @return this Options instance. - */ - public Options dataFormat(String dataFormat) { - this.dataFormat = dataFormat; - return this; - } - - /** - * Sets the dilations option. - * - * @param dilations the dilations option - * @return this Options instance. - */ - public Options dilations(List dilations) { - this.dilations = dilations; - return this; - } - - /** - * Sets the dilations option. - * - * @param dilations the dilations option - * @return this Options instance. - */ - public Options dilations(Long... dilations) { - this.dilations = Arrays.asList(dilations); - return this; - } - } - - @OpInputsMetadata( - outputsClass = RiscConv.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The input input - */ - public final Operand input; - - /** - * The filter input - */ - public final Operand filter; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The strides attribute - */ - public final long[] strides; - - /** - * The dataFormat attribute - */ - public final String dataFormat; - - /** - * The dilations attribute - */ - public final long[] dilations; - - public Inputs(GraphOperation op) { - super(new RiscConv<>(op), op, Arrays.asList("T", "strides", "data_format", "dilations")); - int inputIndex = 0; - input = (Operand) op.input(inputIndex++); - filter = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - strides = op.attributes().getAttrIntList("strides"); - dataFormat = op.attributes().getAttrString("data_format"); - dilations = op.attributes().getAttrIntList("dilations"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCos.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCos.java deleted file mode 100644 index c150d5397b2..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscCos.java +++ /dev/null @@ -1,110 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscCos operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscCos.OP_NAME, - inputsClass = RiscCos.Inputs.class -) -public final class RiscCos extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscCos"; - - private Output y; - - public RiscCos(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscCos operation. - * - * @param scope current scope - * @param x The x value - * @param data type for {@code RiscCos} output and operands - * @return a new instance of RiscCos - */ - @Endpoint( - describeByClass = true - ) - public static RiscCos create(Scope scope, Operand x) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscCos"); - opBuilder.addInput(x.asOutput()); - return new RiscCos<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscCos.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscCos<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDiv.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDiv.java deleted file mode 100644 index 6556fe5c2dd..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDiv.java +++ /dev/null @@ -1,118 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscDiv operation - * - * @param data type for {@code z} output - */ -@OpMetadata( - opType = RiscDiv.OP_NAME, - inputsClass = RiscDiv.Inputs.class -) -public final class RiscDiv extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscDiv"; - - private Output z; - - public RiscDiv(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - z = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscDiv operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @param data type for {@code RiscDiv} output and operands - * @return a new instance of RiscDiv - */ - @Endpoint( - describeByClass = true - ) - public static RiscDiv create(Scope scope, Operand x, Operand y) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscDiv"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - return new RiscDiv<>(opBuilder.build()); - } - - /** - * Gets z. - * - * @return z. - */ - public Output z() { - return z; - } - - @Override - public Output asOutput() { - return z; - } - - @OpInputsMetadata( - outputsClass = RiscDiv.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscDiv<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDot.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDot.java deleted file mode 100644 index 97d34e57a55..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscDot.java +++ /dev/null @@ -1,196 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscDot operation - * - * @param data type for {@code product} output - */ -@OpMetadata( - opType = RiscDot.OP_NAME, - inputsClass = RiscDot.Inputs.class -) -public final class RiscDot extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscDot"; - - private Output product; - - public RiscDot(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - product = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscDot operation. - * - * @param scope current scope - * @param a The a value - * @param b The b value - * @param options carries optional attribute values - * @param data type for {@code RiscDot} output and operands - * @return a new instance of RiscDot - */ - @Endpoint( - describeByClass = true - ) - public static RiscDot create(Scope scope, Operand a, Operand b, - Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscDot"); - opBuilder.addInput(a.asOutput()); - opBuilder.addInput(b.asOutput()); - if (options != null) { - for (Options opts : options) { - if (opts.transposeA != null) { - opBuilder.setAttr("transpose_a", opts.transposeA); - } - if (opts.transposeB != null) { - opBuilder.setAttr("transpose_b", opts.transposeB); - } - } - } - return new RiscDot<>(opBuilder.build()); - } - - /** - * Sets the transposeA option. - * - * @param transposeA the transposeA option - * @return this Options instance. - */ - public static Options transposeA(Boolean transposeA) { - return new Options().transposeA(transposeA); - } - - /** - * Sets the transposeB option. - * - * @param transposeB the transposeB option - * @return this Options instance. - */ - public static Options transposeB(Boolean transposeB) { - return new Options().transposeB(transposeB); - } - - /** - * Gets product. - * - * @return product. - */ - public Output product() { - return product; - } - - @Override - public Output asOutput() { - return product; - } - - /** - * Optional attributes for {@link org.tensorflow.op.risc.RiscDot} - */ - public static class Options { - private Boolean transposeA; - - private Boolean transposeB; - - private Options() { - } - - /** - * Sets the transposeA option. - * - * @param transposeA the transposeA option - * @return this Options instance. - */ - public Options transposeA(Boolean transposeA) { - this.transposeA = transposeA; - return this; - } - - /** - * Sets the transposeB option. - * - * @param transposeB the transposeB option - * @return this Options instance. - */ - public Options transposeB(Boolean transposeB) { - this.transposeB = transposeB; - return this; - } - } - - @OpInputsMetadata( - outputsClass = RiscDot.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The a input - */ - public final Operand a; - - /** - * The b input - */ - public final Operand b; - - /** - * The transposeA attribute - */ - public final boolean transposeA; - - /** - * The transposeB attribute - */ - public final boolean transposeB; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscDot<>(op), op, Arrays.asList("transpose_a", "transpose_b", "T")); - int inputIndex = 0; - a = (Operand) op.input(inputIndex++); - b = (Operand) op.input(inputIndex++); - transposeA = op.attributes().getAttrBool("transpose_a"); - transposeB = op.attributes().getAttrBool("transpose_b"); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscExp.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscExp.java deleted file mode 100644 index f0b7b04ac98..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscExp.java +++ /dev/null @@ -1,110 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscExp operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscExp.OP_NAME, - inputsClass = RiscExp.Inputs.class -) -public final class RiscExp extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscExp"; - - private Output y; - - public RiscExp(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscExp operation. - * - * @param scope current scope - * @param x The x value - * @param data type for {@code RiscExp} output and operands - * @return a new instance of RiscExp - */ - @Endpoint( - describeByClass = true - ) - public static RiscExp create(Scope scope, Operand x) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscExp"); - opBuilder.addInput(x.asOutput()); - return new RiscExp<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscExp.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscExp<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFloor.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFloor.java deleted file mode 100644 index 20ec1a8ba17..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscFloor.java +++ /dev/null @@ -1,110 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscFloor operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscFloor.OP_NAME, - inputsClass = RiscFloor.Inputs.class -) -public final class RiscFloor extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscFloor"; - - private Output y; - - public RiscFloor(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscFloor operation. - * - * @param scope current scope - * @param x The x value - * @param data type for {@code RiscFloor} output and operands - * @return a new instance of RiscFloor - */ - @Endpoint( - describeByClass = true - ) - public static RiscFloor create(Scope scope, Operand x) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscFloor"); - opBuilder.addInput(x.asOutput()); - return new RiscFloor<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscFloor.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscFloor<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscGather.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscGather.java deleted file mode 100644 index ff6b5418760..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscGather.java +++ /dev/null @@ -1,185 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; -import org.tensorflow.types.family.TType; - -/** - * The RiscGather operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscGather.OP_NAME, - inputsClass = RiscGather.Inputs.class -) -public final class RiscGather extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscGather"; - - private Output output; - - public RiscGather(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscGather operation. - * - * @param scope current scope - * @param params The params value - * @param indices The indices value - * @param axis The axis value - * @param options carries optional attribute values - * @param data type for {@code RiscGather} output and operands - * @return a new instance of RiscGather - */ - @Endpoint( - describeByClass = true - ) - public static RiscGather create(Scope scope, Operand params, - Operand indices, Operand axis, Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscGather"); - opBuilder.addInput(params.asOutput()); - opBuilder.addInput(indices.asOutput()); - opBuilder.addInput(axis.asOutput()); - if (options != null) { - for (Options opts : options) { - if (opts.batchDims != null) { - opBuilder.setAttr("batch_dims", opts.batchDims); - } - } - } - return new RiscGather<>(opBuilder.build()); - } - - /** - * Sets the batchDims option. - * - * @param batchDims the batchDims option - * @return this Options instance. - */ - public static Options batchDims(Long batchDims) { - return new Options().batchDims(batchDims); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - /** - * Optional attributes for {@link org.tensorflow.op.risc.RiscGather} - */ - public static class Options { - private Long batchDims; - - private Options() { - } - - /** - * Sets the batchDims option. - * - * @param batchDims the batchDims option - * @return this Options instance. - */ - public Options batchDims(Long batchDims) { - this.batchDims = batchDims; - return this; - } - } - - @OpInputsMetadata( - outputsClass = RiscGather.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The params input - */ - public final Operand params; - - /** - * The indices input - */ - public final Operand indices; - - /** - * The axis input - */ - public final Operand axis; - - /** - * The batchDims attribute - */ - public final long batchDims; - - /** - * The Tparams attribute - */ - public final DataType Tparams; - - /** - * The Tindices attribute - */ - public final DataType Tindices; - - /** - * The Taxis attribute - */ - public final DataType Taxis; - - public Inputs(GraphOperation op) { - super(new RiscGather<>(op), op, Arrays.asList("batch_dims", "Tparams", "Tindices", "Taxis")); - int inputIndex = 0; - params = (Operand) op.input(inputIndex++); - indices = (Operand) op.input(inputIndex++); - axis = (Operand) op.input(inputIndex++); - batchDims = op.attributes().getAttrInt("batch_dims"); - Tparams = op.attributes().getAttrType("Tparams"); - Tindices = op.attributes().getAttrType("Tindices"); - Taxis = op.attributes().getAttrType("Taxis"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscImag.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscImag.java deleted file mode 100644 index 6fa65b2f301..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscImag.java +++ /dev/null @@ -1,136 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.family.TNumber; -import org.tensorflow.types.family.TType; - -/** - * The RiscImag operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscImag.OP_NAME, - inputsClass = RiscImag.Inputs.class -) -public final class RiscImag extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscImag"; - - private Output output; - - public RiscImag(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscImag operation. - * - * @param scope current scope - * @param input The input value - * @param Tout The value of the Tout attribute - * @param data type for {@code RiscImag} output and operands - * @return a new instance of RiscImag - */ - @Endpoint( - describeByClass = true - ) - public static RiscImag create(Scope scope, Operand input, - Class Tout) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscImag"); - opBuilder.addInput(input.asOutput()); - opBuilder.setAttr("Tout", Operands.toDataType(Tout)); - return new RiscImag<>(opBuilder.build()); - } - - /** - * Factory method to create a class wrapping a new RiscImag operation, with the default output types. - * - * @param scope current scope - * @param input The input value - * @return a new instance of RiscImag, with default output types - */ - @Endpoint( - describeByClass = true - ) - public static RiscImag create(Scope scope, Operand input) { - return create(scope, input, TFloat32.class); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscImag.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The input input - */ - public final Operand input; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The Tout attribute - */ - public final DataType Tout; - - public Inputs(GraphOperation op) { - super(new RiscImag<>(op), op, Arrays.asList("T", "Tout")); - int inputIndex = 0; - input = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - Tout = op.attributes().getAttrType("Tout"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscIsFinite.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscIsFinite.java deleted file mode 100644 index 45772d5b001..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscIsFinite.java +++ /dev/null @@ -1,108 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.TBool; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscIsFinite operation - */ -@OpMetadata( - opType = RiscIsFinite.OP_NAME, - inputsClass = RiscIsFinite.Inputs.class -) -public final class RiscIsFinite extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscIsFinite"; - - private Output y; - - public RiscIsFinite(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscIsFinite operation. - * - * @param scope current scope - * @param x The x value - * @return a new instance of RiscIsFinite - */ - @Endpoint( - describeByClass = true - ) - public static RiscIsFinite create(Scope scope, Operand x) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscIsFinite"); - opBuilder.addInput(x.asOutput()); - return new RiscIsFinite(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscIsFinite.class - ) - public static class Inputs extends RawOpInputs { - /** - * The x input - */ - public final Operand x; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscIsFinite(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLog.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLog.java deleted file mode 100644 index 2418350f067..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLog.java +++ /dev/null @@ -1,110 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscLog operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscLog.OP_NAME, - inputsClass = RiscLog.Inputs.class -) -public final class RiscLog extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscLog"; - - private Output y; - - public RiscLog(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscLog operation. - * - * @param scope current scope - * @param x The x value - * @param data type for {@code RiscLog} output and operands - * @return a new instance of RiscLog - */ - @Endpoint( - describeByClass = true - ) - public static RiscLog create(Scope scope, Operand x) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscLog"); - opBuilder.addInput(x.asOutput()); - return new RiscLog<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscLog.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscLog<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalAnd.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalAnd.java deleted file mode 100644 index 5119ddc0d74..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalAnd.java +++ /dev/null @@ -1,108 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TBool; - -/** - * The RiscLogicalAnd operation - */ -@OpMetadata( - opType = RiscLogicalAnd.OP_NAME, - inputsClass = RiscLogicalAnd.Inputs.class -) -public final class RiscLogicalAnd extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscLogicalAnd"; - - private Output z; - - public RiscLogicalAnd(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - z = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscLogicalAnd operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @return a new instance of RiscLogicalAnd - */ - @Endpoint( - describeByClass = true - ) - public static RiscLogicalAnd create(Scope scope, Operand x, Operand y) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscLogicalAnd"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - return new RiscLogicalAnd(opBuilder.build()); - } - - /** - * Gets z. - * - * @return z. - */ - public Output z() { - return z; - } - - @Override - public Output asOutput() { - return z; - } - - @OpInputsMetadata( - outputsClass = RiscLogicalAnd.class - ) - public static class Inputs extends RawOpInputs { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - public Inputs(GraphOperation op) { - super(new RiscLogicalAnd(op), op, Arrays.asList()); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalOr.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalOr.java deleted file mode 100644 index 85ca9fc3508..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscLogicalOr.java +++ /dev/null @@ -1,108 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.types.TBool; - -/** - * The RiscLogicalOr operation - */ -@OpMetadata( - opType = RiscLogicalOr.OP_NAME, - inputsClass = RiscLogicalOr.Inputs.class -) -public final class RiscLogicalOr extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscLogicalOr"; - - private Output z; - - public RiscLogicalOr(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - z = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscLogicalOr operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @return a new instance of RiscLogicalOr - */ - @Endpoint( - describeByClass = true - ) - public static RiscLogicalOr create(Scope scope, Operand x, Operand y) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscLogicalOr"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - return new RiscLogicalOr(opBuilder.build()); - } - - /** - * Gets z. - * - * @return z. - */ - public Output z() { - return z; - } - - @Override - public Output asOutput() { - return z; - } - - @OpInputsMetadata( - outputsClass = RiscLogicalOr.class - ) - public static class Inputs extends RawOpInputs { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - public Inputs(GraphOperation op) { - super(new RiscLogicalOr(op), op, Arrays.asList()); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMax.java deleted file mode 100644 index f3709d295b6..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMax.java +++ /dev/null @@ -1,120 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * Returns max(x, y) element-wise. - * NOTE: {@code risc.RiscMax} does not supports broadcasting. - *

Given two input tensors, the {@code tf.risc_max} operation computes the maximum for every element in the tensor. - * - * @param data type for {@code max} output - */ -@OpMetadata( - opType = RiscMax.OP_NAME, - inputsClass = RiscMax.Inputs.class -) -public final class RiscMax extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscMax"; - - private Output max; - - public RiscMax(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - max = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscMax operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @param data type for {@code RiscMax} output and operands - * @return a new instance of RiscMax - */ - @Endpoint( - describeByClass = true - ) - public static RiscMax create(Scope scope, Operand x, Operand y) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscMax"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - return new RiscMax<>(opBuilder.build()); - } - - /** - * Gets max. - * - * @return max. - */ - public Output max() { - return max; - } - - @Override - public Output asOutput() { - return max; - } - - @OpInputsMetadata( - outputsClass = RiscMax.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscMax<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMin.java deleted file mode 100644 index 33b088561b7..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMin.java +++ /dev/null @@ -1,118 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscMin operation - * - * @param data type for {@code z} output - */ -@OpMetadata( - opType = RiscMin.OP_NAME, - inputsClass = RiscMin.Inputs.class -) -public final class RiscMin extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscMin"; - - private Output z; - - public RiscMin(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - z = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscMin operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @param data type for {@code RiscMin} output and operands - * @return a new instance of RiscMin - */ - @Endpoint( - describeByClass = true - ) - public static RiscMin create(Scope scope, Operand x, Operand y) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscMin"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - return new RiscMin<>(opBuilder.build()); - } - - /** - * Gets z. - * - * @return z. - */ - public Output z() { - return z; - } - - @Override - public Output asOutput() { - return z; - } - - @OpInputsMetadata( - outputsClass = RiscMin.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscMin<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMul.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMul.java deleted file mode 100644 index bc51b7fd9ca..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscMul.java +++ /dev/null @@ -1,118 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscMul operation - * - * @param data type for {@code z} output - */ -@OpMetadata( - opType = RiscMul.OP_NAME, - inputsClass = RiscMul.Inputs.class -) -public final class RiscMul extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscMul"; - - private Output z; - - public RiscMul(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - z = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscMul operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @param data type for {@code RiscMul} output and operands - * @return a new instance of RiscMul - */ - @Endpoint( - describeByClass = true - ) - public static RiscMul create(Scope scope, Operand x, Operand y) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscMul"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - return new RiscMul<>(opBuilder.build()); - } - - /** - * Gets z. - * - * @return z. - */ - public Output z() { - return z; - } - - @Override - public Output asOutput() { - return z; - } - - @OpInputsMetadata( - outputsClass = RiscMul.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscMul<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscNeg.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscNeg.java deleted file mode 100644 index 0eacdcd5c08..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscNeg.java +++ /dev/null @@ -1,110 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscNeg operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscNeg.OP_NAME, - inputsClass = RiscNeg.Inputs.class -) -public final class RiscNeg extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscNeg"; - - private Output y; - - public RiscNeg(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscNeg operation. - * - * @param scope current scope - * @param x The x value - * @param data type for {@code RiscNeg} output and operands - * @return a new instance of RiscNeg - */ - @Endpoint( - describeByClass = true - ) - public static RiscNeg create(Scope scope, Operand x) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscNeg"); - opBuilder.addInput(x.asOutput()); - return new RiscNeg<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscNeg.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscNeg<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPad.java deleted file mode 100644 index 7c9178eb5cf..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPad.java +++ /dev/null @@ -1,133 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscPad operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscPad.OP_NAME, - inputsClass = RiscPad.Inputs.class -) -public final class RiscPad extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscPad"; - - private Output output; - - public RiscPad(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscPad operation. - * - * @param scope current scope - * @param input The input value - * @param paddings The paddings value - * @param constantValues The constantValues value - * @param data type for {@code RiscPad} output and operands - * @return a new instance of RiscPad - */ - @Endpoint( - describeByClass = true - ) - public static RiscPad create(Scope scope, Operand input, - Operand paddings, Operand constantValues) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscPad"); - opBuilder.addInput(input.asOutput()); - opBuilder.addInput(paddings.asOutput()); - opBuilder.addInput(constantValues.asOutput()); - return new RiscPad<>(opBuilder.build()); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscPad.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The input input - */ - public final Operand input; - - /** - * The paddings input - */ - public final Operand paddings; - - /** - * The constantValues input - */ - public final Operand constantValues; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The Tpaddings attribute - */ - public final DataType Tpaddings; - - public Inputs(GraphOperation op) { - super(new RiscPad<>(op), op, Arrays.asList("T", "Tpaddings")); - int inputIndex = 0; - input = (Operand) op.input(inputIndex++); - paddings = (Operand) op.input(inputIndex++); - constantValues = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - Tpaddings = op.attributes().getAttrType("Tpaddings"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPool.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPool.java deleted file mode 100644 index fe9574f672d..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPool.java +++ /dev/null @@ -1,189 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import java.util.List; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscPool operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscPool.OP_NAME, - inputsClass = RiscPool.Inputs.class -) -public final class RiscPool extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscPool"; - - private Output output; - - public RiscPool(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscPool operation. - * - * @param scope current scope - * @param value The value value - * @param ksize The value of the ksize attribute - * @param strides The value of the strides attribute - * @param poolingType The value of the poolingType attribute - * @param options carries optional attribute values - * @param data type for {@code RiscPool} output and operands - * @return a new instance of RiscPool - */ - @Endpoint( - describeByClass = true - ) - public static RiscPool create(Scope scope, Operand value, - List ksize, List strides, String poolingType, Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscPool"); - opBuilder.addInput(value.asOutput()); - long[] ksizeArray = new long[ksize.size()]; - for (int i = 0 ; i < ksizeArray.length ; i++) { - ksizeArray[i] = ksize.get(i); - } - opBuilder.setAttr("ksize", ksizeArray); - long[] stridesArray = new long[strides.size()]; - for (int i = 0 ; i < stridesArray.length ; i++) { - stridesArray[i] = strides.get(i); - } - opBuilder.setAttr("strides", stridesArray); - opBuilder.setAttr("pooling_type", poolingType); - if (options != null) { - for (Options opts : options) { - if (opts.dataFormat != null) { - opBuilder.setAttr("data_format", opts.dataFormat); - } - } - } - return new RiscPool<>(opBuilder.build()); - } - - /** - * Sets the dataFormat option. - * - * @param dataFormat the dataFormat option - * @return this Options instance. - */ - public static Options dataFormat(String dataFormat) { - return new Options().dataFormat(dataFormat); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - /** - * Optional attributes for {@link org.tensorflow.op.risc.RiscPool} - */ - public static class Options { - private String dataFormat; - - private Options() { - } - - /** - * Sets the dataFormat option. - * - * @param dataFormat the dataFormat option - * @return this Options instance. - */ - public Options dataFormat(String dataFormat) { - this.dataFormat = dataFormat; - return this; - } - } - - @OpInputsMetadata( - outputsClass = RiscPool.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The value input - */ - public final Operand value; - - /** - * The ksize attribute - */ - public final long[] ksize; - - /** - * The strides attribute - */ - public final long[] strides; - - /** - * The poolingType attribute - */ - public final String poolingType; - - /** - * The dataFormat attribute - */ - public final String dataFormat; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscPool<>(op), op, Arrays.asList("ksize", "strides", "pooling_type", "data_format", "T")); - int inputIndex = 0; - value = (Operand) op.input(inputIndex++); - ksize = op.attributes().getAttrIntList("ksize"); - strides = op.attributes().getAttrIntList("strides"); - poolingType = op.attributes().getAttrString("pooling_type"); - dataFormat = op.attributes().getAttrString("data_format"); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPow.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPow.java deleted file mode 100644 index c33db0add83..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscPow.java +++ /dev/null @@ -1,118 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscPow operation - * - * @param data type for {@code z} output - */ -@OpMetadata( - opType = RiscPow.OP_NAME, - inputsClass = RiscPow.Inputs.class -) -public final class RiscPow extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscPow"; - - private Output z; - - public RiscPow(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - z = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscPow operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @param data type for {@code RiscPow} output and operands - * @return a new instance of RiscPow - */ - @Endpoint( - describeByClass = true - ) - public static RiscPow create(Scope scope, Operand x, Operand y) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscPow"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - return new RiscPow<>(opBuilder.build()); - } - - /** - * Gets z. - * - * @return z. - */ - public Output z() { - return z; - } - - @Override - public Output asOutput() { - return z; - } - - @OpInputsMetadata( - outputsClass = RiscPow.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscPow<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRandomUniform.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRandomUniform.java deleted file mode 100644 index 5c7bee6d0fe..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRandomUniform.java +++ /dev/null @@ -1,154 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscRandomUniform operation - */ -@OpMetadata( - opType = RiscRandomUniform.OP_NAME, - inputsClass = RiscRandomUniform.Inputs.class -) -public final class RiscRandomUniform extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscRandomUniform"; - - private Output output; - - public RiscRandomUniform(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscRandomUniform operation. - * - * @param scope current scope - * @param shape The shape value - * @param options carries optional attribute values - * @return a new instance of RiscRandomUniform - */ - @Endpoint( - describeByClass = true - ) - public static RiscRandomUniform create(Scope scope, Operand shape, - Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscRandomUniform"); - opBuilder.addInput(shape.asOutput()); - if (options != null) { - for (Options opts : options) { - if (opts.seed != null) { - opBuilder.setAttr("seed", opts.seed); - } - } - } - return new RiscRandomUniform(opBuilder.build()); - } - - /** - * Sets the seed option. - * - * @param seed the seed option - * @return this Options instance. - */ - public static Options seed(Long seed) { - return new Options().seed(seed); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - /** - * Optional attributes for {@link org.tensorflow.op.risc.RiscRandomUniform} - */ - public static class Options { - private Long seed; - - private Options() { - } - - /** - * Sets the seed option. - * - * @param seed the seed option - * @return this Options instance. - */ - public Options seed(Long seed) { - this.seed = seed; - return this; - } - } - - @OpInputsMetadata( - outputsClass = RiscRandomUniform.class - ) - public static class Inputs extends RawOpInputs { - /** - * The shape input - */ - public final Operand shape; - - /** - * The seed attribute - */ - public final long seed; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscRandomUniform(op), op, Arrays.asList("seed", "T")); - int inputIndex = 0; - shape = (Operand) op.input(inputIndex++); - seed = op.attributes().getAttrInt("seed"); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReal.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReal.java deleted file mode 100644 index 66e96fabc27..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReal.java +++ /dev/null @@ -1,136 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.TFloat32; -import org.tensorflow.types.family.TNumber; -import org.tensorflow.types.family.TType; - -/** - * The RiscReal operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscReal.OP_NAME, - inputsClass = RiscReal.Inputs.class -) -public final class RiscReal extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscReal"; - - private Output output; - - public RiscReal(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscReal operation. - * - * @param scope current scope - * @param input The input value - * @param Tout The value of the Tout attribute - * @param data type for {@code RiscReal} output and operands - * @return a new instance of RiscReal - */ - @Endpoint( - describeByClass = true - ) - public static RiscReal create(Scope scope, Operand input, - Class Tout) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscReal"); - opBuilder.addInput(input.asOutput()); - opBuilder.setAttr("Tout", Operands.toDataType(Tout)); - return new RiscReal<>(opBuilder.build()); - } - - /** - * Factory method to create a class wrapping a new RiscReal operation, with the default output types. - * - * @param scope current scope - * @param input The input value - * @return a new instance of RiscReal, with default output types - */ - @Endpoint( - describeByClass = true - ) - public static RiscReal create(Scope scope, Operand input) { - return create(scope, input, TFloat32.class); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscReal.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The input input - */ - public final Operand input; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The Tout attribute - */ - public final DataType Tout; - - public Inputs(GraphOperation op) { - super(new RiscReal<>(op), op, Arrays.asList("T", "Tout")); - int inputIndex = 0; - input = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - Tout = op.attributes().getAttrType("Tout"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReduce.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReduce.java deleted file mode 100644 index a978e967ab5..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReduce.java +++ /dev/null @@ -1,133 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscReduce operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscReduce.OP_NAME, - inputsClass = RiscReduce.Inputs.class -) -public final class RiscReduce extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscReduce"; - - private Output output; - - public RiscReduce(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscReduce operation. - * - * @param scope current scope - * @param tensor The tensor value - * @param axis The axis value - * @param reduceType The value of the reduceType attribute - * @param data type for {@code RiscReduce} output and operands - * @return a new instance of RiscReduce - */ - @Endpoint( - describeByClass = true - ) - public static RiscReduce create(Scope scope, Operand tensor, - Operand axis, String reduceType) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscReduce"); - opBuilder.addInput(tensor.asOutput()); - opBuilder.addInput(axis.asOutput()); - opBuilder.setAttr("reduce_type", reduceType); - return new RiscReduce<>(opBuilder.build()); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscReduce.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The tensor input - */ - public final Operand tensor; - - /** - * The axis input - */ - public final Operand axis; - - /** - * The reduceType attribute - */ - public final String reduceType; - - /** - * The Index attribute - */ - public final DataType Index; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscReduce<>(op), op, Arrays.asList("reduce_type", "Index", "T")); - int inputIndex = 0; - tensor = (Operand) op.input(inputIndex++); - axis = (Operand) op.input(inputIndex++); - reduceType = op.attributes().getAttrString("reduce_type"); - Index = op.attributes().getAttrType("Index"); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRem.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRem.java deleted file mode 100644 index 4e87ed630c9..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscRem.java +++ /dev/null @@ -1,118 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscRem operation - * - * @param data type for {@code z} output - */ -@OpMetadata( - opType = RiscRem.OP_NAME, - inputsClass = RiscRem.Inputs.class -) -public final class RiscRem extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscRem"; - - private Output z; - - public RiscRem(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - z = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscRem operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @param data type for {@code RiscRem} output and operands - * @return a new instance of RiscRem - */ - @Endpoint( - describeByClass = true - ) - public static RiscRem create(Scope scope, Operand x, Operand y) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscRem"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - return new RiscRem<>(opBuilder.build()); - } - - /** - * Gets z. - * - * @return z. - */ - public Output z() { - return z; - } - - @Override - public Output asOutput() { - return z; - } - - @OpInputsMetadata( - outputsClass = RiscRem.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscRem<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReshape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReshape.java deleted file mode 100644 index 89fe6b2f6fe..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReshape.java +++ /dev/null @@ -1,125 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscReshape operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscReshape.OP_NAME, - inputsClass = RiscReshape.Inputs.class -) -public final class RiscReshape extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscReshape"; - - private Output output; - - public RiscReshape(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscReshape operation. - * - * @param scope current scope - * @param tensor The tensor value - * @param shape The shape value - * @param data type for {@code RiscReshape} output and operands - * @return a new instance of RiscReshape - */ - @Endpoint( - describeByClass = true - ) - public static RiscReshape create(Scope scope, Operand tensor, - Operand shape) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscReshape"); - opBuilder.addInput(tensor.asOutput()); - opBuilder.addInput(shape.asOutput()); - return new RiscReshape<>(opBuilder.build()); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscReshape.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The tensor input - */ - public final Operand tensor; - - /** - * The shape input - */ - public final Operand shape; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The Tshape attribute - */ - public final DataType Tshape; - - public Inputs(GraphOperation op) { - super(new RiscReshape<>(op), op, Arrays.asList("T", "Tshape")); - int inputIndex = 0; - tensor = (Operand) op.input(inputIndex++); - shape = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - Tshape = op.attributes().getAttrType("Tshape"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReverse.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReverse.java deleted file mode 100644 index b5045c2673d..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscReverse.java +++ /dev/null @@ -1,125 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscReverse operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscReverse.OP_NAME, - inputsClass = RiscReverse.Inputs.class -) -public final class RiscReverse extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscReverse"; - - private Output output; - - public RiscReverse(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscReverse operation. - * - * @param scope current scope - * @param tensor The tensor value - * @param axis The axis value - * @param data type for {@code RiscReverse} output and operands - * @return a new instance of RiscReverse - */ - @Endpoint( - describeByClass = true - ) - public static RiscReverse create(Scope scope, Operand tensor, - Operand axis) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscReverse"); - opBuilder.addInput(tensor.asOutput()); - opBuilder.addInput(axis.asOutput()); - return new RiscReverse<>(opBuilder.build()); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscReverse.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The tensor input - */ - public final Operand tensor; - - /** - * The axis input - */ - public final Operand axis; - - /** - * The Tidx attribute - */ - public final DataType Tidx; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscReverse<>(op), op, Arrays.asList("Tidx", "T")); - int inputIndex = 0; - tensor = (Operand) op.input(inputIndex++); - axis = (Operand) op.input(inputIndex++); - Tidx = op.attributes().getAttrType("Tidx"); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscScatter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscScatter.java deleted file mode 100644 index 37b5c69d7c6..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscScatter.java +++ /dev/null @@ -1,134 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscScatter operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscScatter.OP_NAME, - inputsClass = RiscScatter.Inputs.class -) -public final class RiscScatter extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscScatter"; - - private Output output; - - public RiscScatter(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscScatter operation. - * - * @param scope current scope - * @param indices The indices value - * @param updates The updates value - * @param shape The shape value - * @param data type for {@code RiscScatter} output and operands - * @param data type for {@code RiscScatter} output and operands - * @return a new instance of RiscScatter - */ - @Endpoint( - describeByClass = true - ) - public static RiscScatter create(Scope scope, - Operand indices, Operand updates, Operand shape) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscScatter"); - opBuilder.addInput(indices.asOutput()); - opBuilder.addInput(updates.asOutput()); - opBuilder.addInput(shape.asOutput()); - return new RiscScatter<>(opBuilder.build()); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscScatter.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The indices input - */ - public final Operand indices; - - /** - * The updates input - */ - public final Operand updates; - - /** - * The shape input - */ - public final Operand shape; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The Tindices attribute - */ - public final DataType Tindices; - - public Inputs(GraphOperation op) { - super(new RiscScatter<>(op), op, Arrays.asList("T", "Tindices")); - int inputIndex = 0; - indices = (Operand) op.input(inputIndex++); - updates = (Operand) op.input(inputIndex++); - shape = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - Tindices = op.attributes().getAttrType("Tindices"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscShape.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscShape.java deleted file mode 100644 index 51614887315..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscShape.java +++ /dev/null @@ -1,135 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.TInt32; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscShape operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscShape.OP_NAME, - inputsClass = RiscShape.Inputs.class -) -public final class RiscShape extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscShape"; - - private Output output; - - public RiscShape(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscShape operation. - * - * @param scope current scope - * @param input The input value - * @param outType The value of the outType attribute - * @param data type for {@code RiscShape} output and operands - * @return a new instance of RiscShape - */ - @Endpoint( - describeByClass = true - ) - public static RiscShape create(Scope scope, - Operand input, Class outType) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscShape"); - opBuilder.addInput(input.asOutput()); - opBuilder.setAttr("out_type", Operands.toDataType(outType)); - return new RiscShape<>(opBuilder.build()); - } - - /** - * Factory method to create a class wrapping a new RiscShape operation, with the default output types. - * - * @param scope current scope - * @param input The input value - * @return a new instance of RiscShape, with default output types - */ - @Endpoint( - describeByClass = true - ) - public static RiscShape create(Scope scope, Operand input) { - return create(scope, input, TInt32.class); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscShape.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The input input - */ - public final Operand input; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The outType attribute - */ - public final DataType outType; - - public Inputs(GraphOperation op) { - super(new RiscShape<>(op), op, Arrays.asList("T", "out_type")); - int inputIndex = 0; - input = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - outType = op.attributes().getAttrType("out_type"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSign.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSign.java deleted file mode 100644 index 8b84667e605..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSign.java +++ /dev/null @@ -1,110 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscSign operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscSign.OP_NAME, - inputsClass = RiscSign.Inputs.class -) -public final class RiscSign extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscSign"; - - private Output y; - - public RiscSign(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscSign operation. - * - * @param scope current scope - * @param x The x value - * @param data type for {@code RiscSign} output and operands - * @return a new instance of RiscSign - */ - @Endpoint( - describeByClass = true - ) - public static RiscSign create(Scope scope, Operand x) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscSign"); - opBuilder.addInput(x.asOutput()); - return new RiscSign<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscSign.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscSign<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSlice.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSlice.java deleted file mode 100644 index 808b9645db1..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSlice.java +++ /dev/null @@ -1,134 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscSlice operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscSlice.OP_NAME, - inputsClass = RiscSlice.Inputs.class -) -public final class RiscSlice extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscSlice"; - - private Output output; - - public RiscSlice(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscSlice operation. - * - * @param scope current scope - * @param input The input value - * @param begin The begin value - * @param sizeOutput The sizeOutput value - * @param data type for {@code RiscSlice} output and operands - * @param data type for {@code RiscSlice} output and operands - * @return a new instance of RiscSlice - */ - @Endpoint( - describeByClass = true - ) - public static RiscSlice create(Scope scope, - Operand input, Operand begin, Operand sizeOutput) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscSlice"); - opBuilder.addInput(input.asOutput()); - opBuilder.addInput(begin.asOutput()); - opBuilder.addInput(sizeOutput.asOutput()); - return new RiscSlice<>(opBuilder.build()); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscSlice.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The input input - */ - public final Operand input; - - /** - * The begin input - */ - public final Operand begin; - - /** - * The sizeOutput input - */ - public final Operand sizeOutput; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The Index attribute - */ - public final DataType Index; - - public Inputs(GraphOperation op) { - super(new RiscSlice<>(op), op, Arrays.asList("T", "Index")); - int inputIndex = 0; - input = (Operand) op.input(inputIndex++); - begin = (Operand) op.input(inputIndex++); - sizeOutput = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - Index = op.attributes().getAttrType("Index"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSort.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSort.java deleted file mode 100644 index 9cd18955703..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSort.java +++ /dev/null @@ -1,133 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscSort operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscSort.OP_NAME, - inputsClass = RiscSort.Inputs.class -) -public final class RiscSort extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscSort"; - - private Output output; - - public RiscSort(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscSort operation. - * - * @param scope current scope - * @param input The input value - * @param axis The axis value - * @param direction The value of the direction attribute - * @param data type for {@code RiscSort} output and operands - * @return a new instance of RiscSort - */ - @Endpoint( - describeByClass = true - ) - public static RiscSort create(Scope scope, Operand input, - Operand axis, String direction) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscSort"); - opBuilder.addInput(input.asOutput()); - opBuilder.addInput(axis.asOutput()); - opBuilder.setAttr("direction", direction); - return new RiscSort<>(opBuilder.build()); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - @OpInputsMetadata( - outputsClass = RiscSort.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The input input - */ - public final Operand input; - - /** - * The axis input - */ - public final Operand axis; - - /** - * The Index attribute - */ - public final DataType Index; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The direction attribute - */ - public final String direction; - - public Inputs(GraphOperation op) { - super(new RiscSort<>(op), op, Arrays.asList("Index", "T", "direction")); - int inputIndex = 0; - input = (Operand) op.input(inputIndex++); - axis = (Operand) op.input(inputIndex++); - Index = op.attributes().getAttrType("Index"); - T = op.attributes().getAttrType("T"); - direction = op.attributes().getAttrString("direction"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSqueeze.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSqueeze.java deleted file mode 100644 index d63a3018ff0..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSqueeze.java +++ /dev/null @@ -1,182 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import java.util.List; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TType; - -/** - * The RiscSqueeze operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscSqueeze.OP_NAME, - inputsClass = RiscSqueeze.Inputs.class -) -public final class RiscSqueeze extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscSqueeze"; - - private Output output; - - public RiscSqueeze(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscSqueeze operation. - * - * @param scope current scope - * @param input The input value - * @param options carries optional attribute values - * @param data type for {@code RiscSqueeze} output and operands - * @return a new instance of RiscSqueeze - */ - @Endpoint( - describeByClass = true - ) - public static RiscSqueeze create(Scope scope, Operand input, - Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscSqueeze"); - opBuilder.addInput(input.asOutput()); - if (options != null) { - for (Options opts : options) { - if (opts.squeezeDims != null) { - long[] squeezeDimsArray = new long[opts.squeezeDims.size()]; - for (int i = 0 ; i < squeezeDimsArray.length ; i++) { - squeezeDimsArray[i] = opts.squeezeDims.get(i); - } - opBuilder.setAttr("squeeze_dims", squeezeDimsArray); - } - } - } - return new RiscSqueeze<>(opBuilder.build()); - } - - /** - * Sets the squeezeDims option. - * - * @param squeezeDims the squeezeDims option - * @return this Options instance. - */ - public static Options squeezeDims(List squeezeDims) { - return new Options().squeezeDims(squeezeDims); - } - - /** - * Sets the squeezeDims option. - * - * @param squeezeDims the squeezeDims option - * @return this Options instance. - */ - public static Options squeezeDims(Long... squeezeDims) { - return new Options().squeezeDims(squeezeDims); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - /** - * Optional attributes for {@link org.tensorflow.op.risc.RiscSqueeze} - */ - public static class Options { - private List squeezeDims; - - private Options() { - } - - /** - * Sets the squeezeDims option. - * - * @param squeezeDims the squeezeDims option - * @return this Options instance. - */ - public Options squeezeDims(List squeezeDims) { - this.squeezeDims = squeezeDims; - return this; - } - - /** - * Sets the squeezeDims option. - * - * @param squeezeDims the squeezeDims option - * @return this Options instance. - */ - public Options squeezeDims(Long... squeezeDims) { - this.squeezeDims = Arrays.asList(squeezeDims); - return this; - } - } - - @OpInputsMetadata( - outputsClass = RiscSqueeze.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The input input - */ - public final Operand input; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The squeezeDims attribute - */ - public final long[] squeezeDims; - - public Inputs(GraphOperation op) { - super(new RiscSqueeze<>(op), op, Arrays.asList("T", "squeeze_dims")); - int inputIndex = 0; - input = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - squeezeDims = op.attributes().getAttrIntList("squeeze_dims"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSub.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSub.java deleted file mode 100644 index 55277160351..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscSub.java +++ /dev/null @@ -1,118 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscSub operation - * - * @param data type for {@code z} output - */ -@OpMetadata( - opType = RiscSub.OP_NAME, - inputsClass = RiscSub.Inputs.class -) -public final class RiscSub extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscSub"; - - private Output z; - - public RiscSub(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - z = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscSub operation. - * - * @param scope current scope - * @param x The x value - * @param y The y value - * @param data type for {@code RiscSub} output and operands - * @return a new instance of RiscSub - */ - @Endpoint( - describeByClass = true - ) - public static RiscSub create(Scope scope, Operand x, Operand y) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscSub"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(y.asOutput()); - return new RiscSub<>(opBuilder.build()); - } - - /** - * Gets z. - * - * @return z. - */ - public Output z() { - return z; - } - - @Override - public Output asOutput() { - return z; - } - - @OpInputsMetadata( - outputsClass = RiscSub.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The y input - */ - public final Operand y; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscSub<>(op), op, Arrays.asList("T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - y = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTranspose.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTranspose.java deleted file mode 100644 index 32be8a3725c..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTranspose.java +++ /dev/null @@ -1,126 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; -import org.tensorflow.types.family.TType; - -/** - * The RiscTranspose operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscTranspose.OP_NAME, - inputsClass = RiscTranspose.Inputs.class -) -public final class RiscTranspose extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscTranspose"; - - private Output y; - - public RiscTranspose(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscTranspose operation. - * - * @param scope current scope - * @param x The x value - * @param perm The perm value - * @param data type for {@code RiscTranspose} output and operands - * @return a new instance of RiscTranspose - */ - @Endpoint( - describeByClass = true - ) - public static RiscTranspose create(Scope scope, Operand x, - Operand perm) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscTranspose"); - opBuilder.addInput(x.asOutput()); - opBuilder.addInput(perm.asOutput()); - return new RiscTranspose<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscTranspose.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The perm input - */ - public final Operand perm; - - /** - * The T attribute - */ - public final DataType T; - - /** - * The Tperm attribute - */ - public final DataType Tperm; - - public Inputs(GraphOperation op) { - super(new RiscTranspose<>(op), op, Arrays.asList("T", "Tperm")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - perm = (Operand) op.input(inputIndex++); - T = op.attributes().getAttrType("T"); - Tperm = op.attributes().getAttrType("Tperm"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTriangularSolve.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTriangularSolve.java deleted file mode 100644 index 8895cf7e1c1..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscTriangularSolve.java +++ /dev/null @@ -1,196 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscTriangularSolve operation - * - * @param data type for {@code output} output - */ -@OpMetadata( - opType = RiscTriangularSolve.OP_NAME, - inputsClass = RiscTriangularSolve.Inputs.class -) -public final class RiscTriangularSolve extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscTriangularSolve"; - - private Output output; - - public RiscTriangularSolve(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - output = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscTriangularSolve operation. - * - * @param scope current scope - * @param matrix The matrix value - * @param rhs The rhs value - * @param options carries optional attribute values - * @param data type for {@code RiscTriangularSolve} output and operands - * @return a new instance of RiscTriangularSolve - */ - @Endpoint( - describeByClass = true - ) - public static RiscTriangularSolve create(Scope scope, Operand matrix, - Operand rhs, Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscTriangularSolve"); - opBuilder.addInput(matrix.asOutput()); - opBuilder.addInput(rhs.asOutput()); - if (options != null) { - for (Options opts : options) { - if (opts.lower != null) { - opBuilder.setAttr("lower", opts.lower); - } - if (opts.adjoint != null) { - opBuilder.setAttr("adjoint", opts.adjoint); - } - } - } - return new RiscTriangularSolve<>(opBuilder.build()); - } - - /** - * Sets the lower option. - * - * @param lower the lower option - * @return this Options instance. - */ - public static Options lower(Boolean lower) { - return new Options().lower(lower); - } - - /** - * Sets the adjoint option. - * - * @param adjoint the adjoint option - * @return this Options instance. - */ - public static Options adjoint(Boolean adjoint) { - return new Options().adjoint(adjoint); - } - - /** - * Gets output. - * - * @return output. - */ - public Output output() { - return output; - } - - @Override - public Output asOutput() { - return output; - } - - /** - * Optional attributes for {@link org.tensorflow.op.risc.RiscTriangularSolve} - */ - public static class Options { - private Boolean lower; - - private Boolean adjoint; - - private Options() { - } - - /** - * Sets the lower option. - * - * @param lower the lower option - * @return this Options instance. - */ - public Options lower(Boolean lower) { - this.lower = lower; - return this; - } - - /** - * Sets the adjoint option. - * - * @param adjoint the adjoint option - * @return this Options instance. - */ - public Options adjoint(Boolean adjoint) { - this.adjoint = adjoint; - return this; - } - } - - @OpInputsMetadata( - outputsClass = RiscTriangularSolve.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The matrix input - */ - public final Operand matrix; - - /** - * The rhs input - */ - public final Operand rhs; - - /** - * The lower attribute - */ - public final boolean lower; - - /** - * The adjoint attribute - */ - public final boolean adjoint; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscTriangularSolve<>(op), op, Arrays.asList("lower", "adjoint", "T")); - int inputIndex = 0; - matrix = (Operand) op.input(inputIndex++); - rhs = (Operand) op.input(inputIndex++); - lower = op.attributes().getAttrBool("lower"); - adjoint = op.attributes().getAttrBool("adjoint"); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscUnary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscUnary.java deleted file mode 100644 index 1e9a8f38454..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscUnary.java +++ /dev/null @@ -1,118 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TNumber; - -/** - * The RiscUnary operation - * - * @param data type for {@code y} output - */ -@OpMetadata( - opType = RiscUnary.OP_NAME, - inputsClass = RiscUnary.Inputs.class -) -public final class RiscUnary extends RawOp implements Operand { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscUnary"; - - private Output y; - - public RiscUnary(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - y = operation.output(outputIdx++); - } - - /** - * Factory method to create a class wrapping a new RiscUnary operation. - * - * @param scope current scope - * @param x The x value - * @param opType The value of the opType attribute - * @param data type for {@code RiscUnary} output and operands - * @return a new instance of RiscUnary - */ - @Endpoint( - describeByClass = true - ) - public static RiscUnary create(Scope scope, Operand x, String opType) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscUnary"); - opBuilder.addInput(x.asOutput()); - opBuilder.setAttr("op_type", opType); - return new RiscUnary<>(opBuilder.build()); - } - - /** - * Gets y. - * - * @return y. - */ - public Output y() { - return y; - } - - @Override - public Output asOutput() { - return y; - } - - @OpInputsMetadata( - outputsClass = RiscUnary.class - ) - public static class Inputs extends RawOpInputs> { - /** - * The x input - */ - public final Operand x; - - /** - * The opType attribute - */ - public final String opType; - - /** - * The T attribute - */ - public final DataType T; - - public Inputs(GraphOperation op) { - super(new RiscUnary<>(op), op, Arrays.asList("op_type", "T")); - int inputIndex = 0; - x = (Operand) op.input(inputIndex++); - opType = op.attributes().getAttrString("op_type"); - T = op.attributes().getAttrType("T"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscWhile.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscWhile.java deleted file mode 100644 index 11fe3f85744..00000000000 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/risc/RiscWhile.java +++ /dev/null @@ -1,225 +0,0 @@ -/* Copyright 2018-2023 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -=======================================================================*/ - -// This class has been generated, DO NOT EDIT! - -package org.tensorflow.op.risc; - -import java.util.Arrays; -import java.util.Iterator; -import java.util.List; -import org.tensorflow.ConcreteFunction; -import org.tensorflow.GraphOperation; -import org.tensorflow.Operand; -import org.tensorflow.Operation; -import org.tensorflow.OperationBuilder; -import org.tensorflow.Output; -import org.tensorflow.ndarray.Shape; -import org.tensorflow.op.Operands; -import org.tensorflow.op.RawOp; -import org.tensorflow.op.RawOpInputs; -import org.tensorflow.op.Scope; -import org.tensorflow.op.annotation.Endpoint; -import org.tensorflow.op.annotation.OpInputsMetadata; -import org.tensorflow.op.annotation.OpMetadata; -import org.tensorflow.proto.DataType; -import org.tensorflow.types.family.TType; - -/** - * The RiscWhile operation - */ -@OpMetadata( - opType = RiscWhile.OP_NAME, - inputsClass = RiscWhile.Inputs.class -) -public final class RiscWhile extends RawOp implements Iterable> { - /** - * The name of this op, as known by TensorFlow core engine - */ - public static final String OP_NAME = "RiscWhile"; - - private List> output; - - @SuppressWarnings("unchecked") - public RiscWhile(Operation operation) { - super(operation, OP_NAME); - int outputIdx = 0; - int outputLength = operation.outputListLength("output"); - output = Arrays.asList(operation.outputList(outputIdx, outputLength)); - outputIdx += outputLength; - } - - /** - * Factory method to create a class wrapping a new RiscWhile operation. - * - * @param scope current scope - * @param input The input value - * @param cond The value of the cond attribute - * @param body The value of the body attribute - * @param options carries optional attribute values - * @return a new instance of RiscWhile - */ - @Endpoint( - describeByClass = true - ) - public static RiscWhile create(Scope scope, Iterable> input, ConcreteFunction cond, - ConcreteFunction body, Options... options) { - OperationBuilder opBuilder = scope.opBuilder(OP_NAME, "RiscWhile"); - opBuilder.addInputList(Operands.asOutputs(input)); - opBuilder.setAttr("cond", cond); - opBuilder.setAttr("body", body); - if (options != null) { - for (Options opts : options) { - if (opts.outputShapes != null) { - Shape[] outputShapesArray = new Shape[opts.outputShapes.size()]; - for (int i = 0 ; i < outputShapesArray.length ; i++) { - outputShapesArray[i] = opts.outputShapes.get(i); - } - opBuilder.setAttr("output_shapes", outputShapesArray); - } - if (opts.parallelIterations != null) { - opBuilder.setAttr("parallel_iterations", opts.parallelIterations); - } - } - } - return new RiscWhile(opBuilder.build()); - } - - /** - * Sets the outputShapes option. - * - * @param outputShapes the outputShapes option - * @return this Options instance. - */ - public static Options outputShapes(List outputShapes) { - return new Options().outputShapes(outputShapes); - } - - /** - * Sets the outputShapes option. - * - * @param outputShapes the outputShapes option - * @return this Options instance. - */ - public static Options outputShapes(Shape... outputShapes) { - return new Options().outputShapes(outputShapes); - } - - /** - * Sets the parallelIterations option. - * - * @param parallelIterations the parallelIterations option - * @return this Options instance. - */ - public static Options parallelIterations(Long parallelIterations) { - return new Options().parallelIterations(parallelIterations); - } - - /** - * Gets output. - * - * @return output. - */ - public List> output() { - return output; - } - - @Override - @SuppressWarnings({"rawtypes", "unchecked"}) - public Iterator> iterator() { - return (Iterator) output.iterator(); - } - - /** - * Optional attributes for {@link org.tensorflow.op.risc.RiscWhile} - */ - public static class Options { - private List outputShapes; - - private Long parallelIterations; - - private Options() { - } - - /** - * Sets the outputShapes option. - * - * @param outputShapes the outputShapes option - * @return this Options instance. - */ - public Options outputShapes(List outputShapes) { - this.outputShapes = outputShapes; - return this; - } - - /** - * Sets the outputShapes option. - * - * @param outputShapes the outputShapes option - * @return this Options instance. - */ - public Options outputShapes(Shape... outputShapes) { - this.outputShapes = Arrays.asList(outputShapes); - return this; - } - - /** - * Sets the parallelIterations option. - * - * @param parallelIterations the parallelIterations option - * @return this Options instance. - */ - public Options parallelIterations(Long parallelIterations) { - this.parallelIterations = parallelIterations; - return this; - } - } - - @OpInputsMetadata( - outputsClass = RiscWhile.class - ) - public static class Inputs extends RawOpInputs { - /** - * The input input - */ - public final Iterable> input; - - /** - * The T attribute - */ - public final DataType[] T; - - /** - * The outputShapes attribute - */ - public final Shape[] outputShapes; - - /** - * The parallelIterations attribute - */ - public final long parallelIterations; - - public Inputs(GraphOperation op) { - super(new RiscWhile(op), op, Arrays.asList("T", "output_shapes", "parallel_iterations")); - int inputIndex = 0; - int inputLength = op.inputListLength("input"); - input = Arrays.asList((Operand[]) op.inputList(inputIndex, inputLength)); - inputIndex += inputLength; - T = op.attributes().getAttrTypeList("T"); - outputShapes = op.attributes().getAttrShapeList("output_shapes"); - parallelIterations = op.attributes().getAttrInt("parallel_iterations"); - } - } -} diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseCountSparseOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseCountSparseOutput.java index 51c60674ed1..49d78c0517c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseCountSparseOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/DenseCountSparseOutput.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -43,6 +44,9 @@ opType = DenseCountSparseOutput.OP_NAME, inputsClass = DenseCountSparseOutput.Inputs.class ) +@Operator( + group = "sparse" +) public final class DenseCountSparseOutput extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCountSparseOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCountSparseOutput.java index 5e8ec528202..c3983444bd3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCountSparseOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/sparse/SparseCountSparseOutput.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TNumber; @@ -43,6 +44,9 @@ opType = SparseCountSparseOutput.OP_NAME, inputsClass = SparseCountSparseOutput.Inputs.class ) +@Operator( + group = "sparse" +) public final class SparseCountSparseOutput extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexFullMatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexFullMatch.java index c4692670796..8ba7591ee79 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexFullMatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexFullMatch.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBool; import org.tensorflow.types.TString; @@ -44,6 +45,9 @@ opType = StaticRegexFullMatch.OP_NAME, inputsClass = StaticRegexFullMatch.Inputs.class ) +@Operator( + group = "strings" +) public final class StaticRegexFullMatch extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexReplace.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexReplace.java index a43d580e2be..a7a156a731f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexReplace.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/StaticRegexReplace.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** @@ -39,6 +40,9 @@ opType = StaticRegexReplace.OP_NAME, inputsClass = StaticRegexReplace.Inputs.class ) +@Operator( + group = "strings" +) public final class StaticRegexReplace extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecode.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecode.java index b11541057cf..f3808a6b3d2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecode.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecode.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -58,6 +59,9 @@ opType = UnicodeDecode.OP_NAME, inputsClass = UnicodeDecode.Inputs.class ) +@Operator( + group = "strings" +) public final class UnicodeDecode extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecodeWithOffsets.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecodeWithOffsets.java index a90e55171de..9694f3cef53 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecodeWithOffsets.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeDecodeWithOffsets.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.TInt64; @@ -62,6 +63,9 @@ opType = UnicodeDecodeWithOffsets.OP_NAME, inputsClass = UnicodeDecodeWithOffsets.Inputs.class ) +@Operator( + group = "strings" +) public final class UnicodeDecodeWithOffsets extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeEncode.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeEncode.java index 7f30b6eb99a..d3dbb9589df 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeEncode.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnicodeEncode.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -53,6 +54,9 @@ opType = UnicodeEncode.OP_NAME, inputsClass = UnicodeEncode.Inputs.class ) +@Operator( + group = "strings" +) public final class UnicodeEncode extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnsortedSegmentJoin.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnsortedSegmentJoin.java index 9b4be85a8c6..b9d82efab59 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnsortedSegmentJoin.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/strings/UnsortedSegmentJoin.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TNumber; @@ -40,6 +41,9 @@ opType = UnsortedSegmentJoin.OP_NAME, inputsClass = UnsortedSegmentJoin.Inputs.class ) +@Operator( + group = "strings" +) public final class UnsortedSegmentJoin extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CloseSummaryWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CloseSummaryWriter.java index f8415b30828..6621499b9ad 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CloseSummaryWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CloseSummaryWriter.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -37,6 +38,9 @@ opType = CloseSummaryWriter.OP_NAME, inputsClass = CloseSummaryWriter.Inputs.class ) +@Operator( + group = "summary" +) public final class CloseSummaryWriter extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryDbWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryDbWriter.java index b3ea51287c2..5a24e79a5f3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryDbWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryDbWriter.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -38,6 +39,9 @@ opType = CreateSummaryDbWriter.OP_NAME, inputsClass = CreateSummaryDbWriter.Inputs.class ) +@Operator( + group = "summary" +) public final class CreateSummaryDbWriter extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryFileWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryFileWriter.java index 806e764f186..1e62bfe05c2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryFileWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/CreateSummaryFileWriter.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = CreateSummaryFileWriter.OP_NAME, inputsClass = CreateSummaryFileWriter.Inputs.class ) +@Operator( + group = "summary" +) public final class CreateSummaryFileWriter extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/FlushSummaryWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/FlushSummaryWriter.java index da07070af44..3faedb9a03b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/FlushSummaryWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/FlushSummaryWriter.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -37,6 +38,9 @@ opType = FlushSummaryWriter.OP_NAME, inputsClass = FlushSummaryWriter.Inputs.class ) +@Operator( + group = "summary" +) public final class FlushSummaryWriter extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImportEvent.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImportEvent.java index 38a842be645..a9723ef2fcb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImportEvent.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/ImportEvent.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -38,6 +39,9 @@ opType = ImportEvent.OP_NAME, inputsClass = ImportEvent.Inputs.class ) +@Operator( + group = "summary" +) public final class ImportEvent extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/StatsAggregatorSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/StatsAggregatorSummary.java index f60238e781c..b604a6b85ce 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/StatsAggregatorSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/StatsAggregatorSummary.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = StatsAggregatorSummary.OP_NAME, inputsClass = StatsAggregatorSummary.Inputs.class ) +@Operator( + group = "summary" +) public final class StatsAggregatorSummary extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/SummaryWriter.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/SummaryWriter.java index c5b7d4100c7..d891f9b35a1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/SummaryWriter.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/SummaryWriter.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -38,6 +39,9 @@ opType = SummaryWriter.OP_NAME, inputsClass = SummaryWriter.Inputs.class ) +@Operator( + group = "summary" +) public final class SummaryWriter extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteAudioSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteAudioSummary.java index 559b1cfb69b..9ba1858b59e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteAudioSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteAudioSummary.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -42,6 +43,9 @@ opType = WriteAudioSummary.OP_NAME, inputsClass = WriteAudioSummary.Inputs.class ) +@Operator( + group = "summary" +) public final class WriteAudioSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteGraphSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteGraphSummary.java index feef8cad613..565ef3940f3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteGraphSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteGraphSummary.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -40,6 +41,9 @@ opType = WriteGraphSummary.OP_NAME, inputsClass = WriteGraphSummary.Inputs.class ) +@Operator( + group = "summary" +) public final class WriteGraphSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteHistogramSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteHistogramSummary.java index 52f3deb5adf..ba431bc0d7e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteHistogramSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteHistogramSummary.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -41,6 +42,9 @@ opType = WriteHistogramSummary.OP_NAME, inputsClass = WriteHistogramSummary.Inputs.class ) +@Operator( + group = "summary" +) public final class WriteHistogramSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteImageSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteImageSummary.java index 1b6676414a0..12d4578a8b3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteImageSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteImageSummary.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -44,6 +45,9 @@ opType = WriteImageSummary.OP_NAME, inputsClass = WriteImageSummary.Inputs.class ) +@Operator( + group = "summary" +) public final class WriteImageSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteRawProtoSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteRawProtoSummary.java index fd559e1d210..d56f66e11c6 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteRawProtoSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteRawProtoSummary.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -40,6 +41,9 @@ opType = WriteRawProtoSummary.OP_NAME, inputsClass = WriteRawProtoSummary.Inputs.class ) +@Operator( + group = "summary" +) public final class WriteRawProtoSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteScalarSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteScalarSummary.java index 11c4f3978b0..d7055fb14dc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteScalarSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteScalarSummary.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -42,6 +43,9 @@ opType = WriteScalarSummary.OP_NAME, inputsClass = WriteScalarSummary.Inputs.class ) +@Operator( + group = "summary" +) public final class WriteScalarSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteSummary.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteSummary.java index a4d040a550d..31a5a470394 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteSummary.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/summary/WriteSummary.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.TString; @@ -41,6 +42,9 @@ opType = WriteSummary.OP_NAME, inputsClass = WriteSummary.Inputs.class ) +@Operator( + group = "summary" +) public final class WriteSummary extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/AllToAll.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/AllToAll.java index 4ff3db65f88..dfe6664886c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/AllToAll.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/AllToAll.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -55,6 +56,9 @@ opType = AllToAll.OP_NAME, inputsClass = AllToAll.Inputs.class ) +@Operator( + group = "tpu" +) public final class AllToAll extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompilationResult.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompilationResult.java index e4272d3223b..f6253a3f89f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompilationResult.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CompilationResult.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** @@ -41,6 +42,9 @@ opType = CompilationResult.OP_NAME, inputsClass = CompilationResult.Inputs.class ) +@Operator( + group = "tpu" +) public final class CompilationResult extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ComputeDedupDataTupleMask.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ComputeDedupDataTupleMask.java index 1eceb3fb43e..95078aebabc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ComputeDedupDataTupleMask.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ComputeDedupDataTupleMask.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; /** @@ -41,6 +42,9 @@ opType = ComputeDedupDataTupleMask.OP_NAME, inputsClass = ComputeDedupDataTupleMask.Inputs.class ) +@Operator( + group = "tpu" +) public final class ComputeDedupDataTupleMask extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureDistributedTPU.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureDistributedTPU.java index 7c2ce500a30..058a08bcba8 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureDistributedTPU.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureDistributedTPU.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** @@ -38,6 +39,9 @@ opType = ConfigureDistributedTPU.OP_NAME, inputsClass = ConfigureDistributedTPU.Inputs.class ) +@Operator( + group = "tpu" +) public final class ConfigureDistributedTPU extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureTPUEmbedding.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureTPUEmbedding.java index 1720e9bcc46..99f92e64fc1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureTPUEmbedding.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ConfigureTPUEmbedding.java @@ -27,6 +27,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; /** * Sets up TPUEmbedding in a distributed TPU system. @@ -35,6 +36,9 @@ opType = ConfigureTPUEmbedding.OP_NAME, inputsClass = ConfigureTPUEmbedding.Inputs.class ) +@Operator( + group = "tpu" +) public final class ConfigureTPUEmbedding extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CrossReplicaSum.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CrossReplicaSum.java index 0e573ddaa2b..c56e985eafb 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CrossReplicaSum.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/CrossReplicaSum.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TNumber; @@ -47,6 +48,9 @@ opType = CrossReplicaSum.OP_NAME, inputsClass = CrossReplicaSum.Inputs.class ) +@Operator( + group = "tpu" +) public final class CrossReplicaSum extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.java index a2df3c57f6c..8badd319ee1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -48,6 +49,9 @@ opType = DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.OP_NAME, inputsClass = DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.Inputs.class ) +@Operator( + group = "tpu" +) public final class DynamicEnqueueTPUEmbeddingArbitraryTensorBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/DynamicEnqueueTPUEmbeddingRaggedTensorBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/DynamicEnqueueTPUEmbeddingRaggedTensorBatch.java index 50490bb72ec..48cd749fe92 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/DynamicEnqueueTPUEmbeddingRaggedTensorBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/DynamicEnqueueTPUEmbeddingRaggedTensorBatch.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -42,6 +43,9 @@ opType = DynamicEnqueueTPUEmbeddingRaggedTensorBatch.OP_NAME, inputsClass = DynamicEnqueueTPUEmbeddingRaggedTensorBatch.Inputs.class ) +@Operator( + group = "tpu" +) public final class DynamicEnqueueTPUEmbeddingRaggedTensorBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EmbeddingActivations.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EmbeddingActivations.java index 9d8d9ea2947..b952f8edcae 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EmbeddingActivations.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EmbeddingActivations.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -43,6 +44,9 @@ opType = EmbeddingActivations.OP_NAME, inputsClass = EmbeddingActivations.Inputs.class ) +@Operator( + group = "tpu" +) public final class EmbeddingActivations extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingArbitraryTensorBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingArbitraryTensorBatch.java index ce3b1358616..eca45be182a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingArbitraryTensorBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingArbitraryTensorBatch.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TNumber; @@ -47,6 +48,9 @@ opType = EnqueueTPUEmbeddingArbitraryTensorBatch.OP_NAME, inputsClass = EnqueueTPUEmbeddingArbitraryTensorBatch.Inputs.class ) +@Operator( + group = "tpu" +) public final class EnqueueTPUEmbeddingArbitraryTensorBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingBatch.java index 7171cbe5da5..0697ab23c98 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingBatch.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** @@ -40,6 +41,9 @@ opType = EnqueueTPUEmbeddingBatch.OP_NAME, inputsClass = EnqueueTPUEmbeddingBatch.Inputs.class ) +@Operator( + group = "tpu" +) public final class EnqueueTPUEmbeddingBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.java index 4c23f2c972c..b8a39d7a685 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.TString; @@ -39,6 +40,9 @@ opType = EnqueueTPUEmbeddingIntegerBatch.OP_NAME, inputsClass = EnqueueTPUEmbeddingIntegerBatch.Inputs.class ) +@Operator( + group = "tpu" +) public final class EnqueueTPUEmbeddingIntegerBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingRaggedTensorBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingRaggedTensorBatch.java index 055da6fb91c..17ed314dab1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingRaggedTensorBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingRaggedTensorBatch.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TNumber; @@ -48,6 +49,9 @@ opType = EnqueueTPUEmbeddingRaggedTensorBatch.OP_NAME, inputsClass = EnqueueTPUEmbeddingRaggedTensorBatch.Inputs.class ) +@Operator( + group = "tpu" +) public final class EnqueueTPUEmbeddingRaggedTensorBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseBatch.java index efb6acb56fd..14f9f072088 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseBatch.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TNumber; @@ -49,6 +50,9 @@ opType = EnqueueTPUEmbeddingSparseBatch.OP_NAME, inputsClass = EnqueueTPUEmbeddingSparseBatch.Inputs.class ) +@Operator( + group = "tpu" +) public final class EnqueueTPUEmbeddingSparseBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseTensorBatch.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseTensorBatch.java index 1f95961867f..cec65b34619 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseTensorBatch.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingSparseTensorBatch.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TString; import org.tensorflow.types.family.TNumber; @@ -48,6 +49,9 @@ opType = EnqueueTPUEmbeddingSparseTensorBatch.OP_NAME, inputsClass = EnqueueTPUEmbeddingSparseTensorBatch.Inputs.class ) +@Operator( + group = "tpu" +) public final class EnqueueTPUEmbeddingSparseTensorBatch extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeue.java index 86eac21375e..20e200e26af 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeue.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = InfeedDequeue.OP_NAME, inputsClass = InfeedDequeue.Inputs.class ) +@Operator( + group = "tpu" +) public final class InfeedDequeue extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeueTuple.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeueTuple.java index 26165215fd9..aa59b48755f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeueTuple.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedDequeueTuple.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = InfeedDequeueTuple.OP_NAME, inputsClass = InfeedDequeueTuple.Inputs.class ) +@Operator( + group = "tpu" +) public final class InfeedDequeueTuple extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueue.java index 582a258d5a3..e9c5ebd146c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueue.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +41,9 @@ opType = InfeedEnqueue.OP_NAME, inputsClass = InfeedEnqueue.Inputs.class ) +@Operator( + group = "tpu" +) public final class InfeedEnqueue extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueuePrelinearizedBuffer.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueuePrelinearizedBuffer.java index 03e8ec9a5e3..0e2294decc2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueuePrelinearizedBuffer.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueuePrelinearizedBuffer.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -37,6 +38,9 @@ opType = InfeedEnqueuePrelinearizedBuffer.OP_NAME, inputsClass = InfeedEnqueuePrelinearizedBuffer.Inputs.class ) +@Operator( + group = "tpu" +) public final class InfeedEnqueuePrelinearizedBuffer extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueueTuple.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueueTuple.java index d6ca9a6a174..6d5b8ab5cb1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueueTuple.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/InfeedEnqueueTuple.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; /** @@ -40,6 +41,9 @@ opType = InfeedEnqueueTuple.OP_NAME, inputsClass = InfeedEnqueueTuple.Inputs.class ) +@Operator( + group = "tpu" +) public final class InfeedEnqueueTuple extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/IsTPUEmbeddingInitialized.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/IsTPUEmbeddingInitialized.java index e2e17414adc..a34b2fd2361 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/IsTPUEmbeddingInitialized.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/IsTPUEmbeddingInitialized.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TBool; /** @@ -38,6 +39,9 @@ opType = IsTPUEmbeddingInitialized.OP_NAME, inputsClass = IsTPUEmbeddingInitialized.Inputs.class ) +@Operator( + group = "tpu" +) public final class IsTPUEmbeddingInitialized extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadAllTPUEmbeddingParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadAllTPUEmbeddingParameters.java index 977173e1588..7729db3d126 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadAllTPUEmbeddingParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadAllTPUEmbeddingParameters.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -46,6 +47,9 @@ opType = LoadAllTPUEmbeddingParameters.OP_NAME, inputsClass = LoadAllTPUEmbeddingParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadAllTPUEmbeddingParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParameters.java index be276ade9ab..284da4dccac 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingADAMParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingADAMParameters.OP_NAME, inputsClass = LoadTPUEmbeddingADAMParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingADAMParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParameters.java index 1e5828f08de..f82fc61402b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdadeltaParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingAdadeltaParameters.OP_NAME, inputsClass = LoadTPUEmbeddingAdadeltaParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingAdadeltaParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradMomentumParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradMomentumParameters.java index 4475f85af31..54f09315582 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradMomentumParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradMomentumParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingAdagradMomentumParameters.OP_NAME, inputsClass = LoadTPUEmbeddingAdagradMomentumParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingAdagradMomentumParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParameters.java index 3568a65bd7b..fba43cc1e1e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingAdagradParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingAdagradParameters.OP_NAME, inputsClass = LoadTPUEmbeddingAdagradParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingAdagradParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingCenteredRMSPropParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingCenteredRMSPropParameters.java index 8a5eb8ac4d8..274e83ee16e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingCenteredRMSPropParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingCenteredRMSPropParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingCenteredRMSPropParameters.OP_NAME, inputsClass = LoadTPUEmbeddingCenteredRMSPropParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingCenteredRMSPropParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParameters.java index f8833e90843..d765e76ddf5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFTRLParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingFTRLParameters.OP_NAME, inputsClass = LoadTPUEmbeddingFTRLParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingFTRLParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFrequencyEstimatorParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFrequencyEstimatorParameters.java index e8d45f51acd..89988b64472 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFrequencyEstimatorParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingFrequencyEstimatorParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingFrequencyEstimatorParameters.OP_NAME, inputsClass = LoadTPUEmbeddingFrequencyEstimatorParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingFrequencyEstimatorParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMDLAdagradLightParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMDLAdagradLightParameters.java index 3c0bbd29f4d..1bb174085a5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMDLAdagradLightParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMDLAdagradLightParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingMDLAdagradLightParameters.OP_NAME, inputsClass = LoadTPUEmbeddingMDLAdagradLightParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingMDLAdagradLightParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParameters.java index a647db5ab54..a2588e51af9 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingMomentumParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingMomentumParameters.OP_NAME, inputsClass = LoadTPUEmbeddingMomentumParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingMomentumParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParameters.java index 737a5f081cd..f7f2ec6524b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalAdagradParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingProximalAdagradParameters.OP_NAME, inputsClass = LoadTPUEmbeddingProximalAdagradParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingProximalAdagradParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParameters.java index 90bf2fc4cf1..670993c8cbd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingProximalYogiParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -37,6 +38,9 @@ opType = LoadTPUEmbeddingProximalYogiParameters.OP_NAME, inputsClass = LoadTPUEmbeddingProximalYogiParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingProximalYogiParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParameters.java index f75e94c7795..02c378df53e 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingRMSPropParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingRMSPropParameters.OP_NAME, inputsClass = LoadTPUEmbeddingRMSPropParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingRMSPropParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParameters.java index a35574afbb4..cb0883872c2 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/LoadTPUEmbeddingStochasticGradientDescentParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = LoadTPUEmbeddingStochasticGradientDescentParameters.OP_NAME, inputsClass = LoadTPUEmbeddingStochasticGradientDescentParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class LoadTPUEmbeddingStochasticGradientDescentParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/MergeDedupData.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/MergeDedupData.java index da1f35645ec..e524db43c5f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/MergeDedupData.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/MergeDedupData.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -45,6 +46,9 @@ opType = MergeDedupData.OP_NAME, inputsClass = MergeDedupData.Inputs.class ) +@Operator( + group = "tpu" +) public final class MergeDedupData extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OrdinalSelector.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OrdinalSelector.java index 1f54ab4ed7c..5df0d72c590 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OrdinalSelector.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OrdinalSelector.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; /** @@ -41,6 +42,9 @@ opType = OrdinalSelector.OP_NAME, inputsClass = OrdinalSelector.Inputs.class ) +@Operator( + group = "tpu" +) public final class OrdinalSelector extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeue.java index 64123272bf8..fa6e47d2c7d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeue.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +45,9 @@ opType = OutfeedDequeue.OP_NAME, inputsClass = OutfeedDequeue.Inputs.class ) +@Operator( + group = "tpu" +) public final class OutfeedDequeue extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTuple.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTuple.java index 9ec3d005ea0..bef44e59c77 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTuple.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTuple.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -45,6 +46,9 @@ opType = OutfeedDequeueTuple.OP_NAME, inputsClass = OutfeedDequeueTuple.Inputs.class ) +@Operator( + group = "tpu" +) public final class OutfeedDequeueTuple extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTupleV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTupleV2.java index 491a13af3e9..23b7f3ae189 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTupleV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueTupleV2.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -47,6 +48,9 @@ opType = OutfeedDequeueTupleV2.OP_NAME, inputsClass = OutfeedDequeueTupleV2.Inputs.class ) +@Operator( + group = "tpu" +) public final class OutfeedDequeueTupleV2 extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueV2.java index 749f1811a27..481f916e86a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedDequeueV2.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -46,6 +47,9 @@ opType = OutfeedDequeueV2.OP_NAME, inputsClass = OutfeedDequeueV2.Inputs.class ) +@Operator( + group = "tpu" +) public final class OutfeedDequeueV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueue.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueue.java index e0192d4a325..f79f949acbe 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueue.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueue.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -38,6 +39,9 @@ opType = OutfeedEnqueue.OP_NAME, inputsClass = OutfeedEnqueue.Inputs.class ) +@Operator( + group = "tpu" +) public final class OutfeedEnqueue extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueueTuple.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueueTuple.java index 09e6d415fbe..df3eb3df0e1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueueTuple.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/OutfeedEnqueueTuple.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; /** @@ -38,6 +39,9 @@ opType = OutfeedEnqueueTuple.OP_NAME, inputsClass = OutfeedEnqueueTuple.Inputs.class ) +@Operator( + group = "tpu" +) public final class OutfeedEnqueueTuple extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedCall.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedCall.java index b1c5e7a3fbe..f20a2de4bda 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedCall.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedCall.java @@ -33,6 +33,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -44,6 +45,9 @@ opType = PartitionedCall.OP_NAME, inputsClass = PartitionedCall.Inputs.class ) +@Operator( + group = "tpu" +) public final class PartitionedCall extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedInput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedInput.java index d67f21f4430..be69029e573 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedInput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PartitionedInput.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = PartitionedInput.OP_NAME, inputsClass = PartitionedInput.Inputs.class ) +@Operator( + group = "tpu" +) public final class PartitionedInput extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Prelinearize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Prelinearize.java index 7a943b1dae0..d5417a9aa80 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Prelinearize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/Prelinearize.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +42,9 @@ opType = Prelinearize.OP_NAME, inputsClass = Prelinearize.Inputs.class ) +@Operator( + group = "tpu" +) public final class Prelinearize extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PrelinearizeTuple.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PrelinearizeTuple.java index ad07ce8acd3..cb67b7ad219 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PrelinearizeTuple.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/PrelinearizeTuple.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = PrelinearizeTuple.OP_NAME, inputsClass = PrelinearizeTuple.Inputs.class ) +@Operator( + group = "tpu" +) public final class PrelinearizeTuple extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RecvTPUEmbeddingActivations.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RecvTPUEmbeddingActivations.java index 4b763eff19e..68dcd11b979 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RecvTPUEmbeddingActivations.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RecvTPUEmbeddingActivations.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -46,6 +47,9 @@ opType = RecvTPUEmbeddingActivations.OP_NAME, inputsClass = RecvTPUEmbeddingActivations.Inputs.class ) +@Operator( + group = "tpu" +) public final class RecvTPUEmbeddingActivations extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicateMetadata.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicateMetadata.java index cf8d3571d67..67f49871a2c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicateMetadata.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicateMetadata.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; /** * Metadata indicating how the TPU computation should be replicated. @@ -37,6 +38,9 @@ opType = ReplicateMetadata.OP_NAME, inputsClass = ReplicateMetadata.Inputs.class ) +@Operator( + group = "tpu" +) public final class ReplicateMetadata extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedInput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedInput.java index 64d8aea2f9a..37c057fc375 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedInput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedInput.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -52,6 +53,9 @@ opType = ReplicatedInput.OP_NAME, inputsClass = ReplicatedInput.Inputs.class ) +@Operator( + group = "tpu" +) public final class ReplicatedInput extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedOutput.java index ccff2349b3b..fcc447fb932 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ReplicatedOutput.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -51,6 +52,9 @@ opType = ReplicatedOutput.OP_NAME, inputsClass = ReplicatedOutput.Inputs.class ) +@Operator( + group = "tpu" +) public final class ReplicatedOutput extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveAllTPUEmbeddingParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveAllTPUEmbeddingParameters.java index 9ccd84e69c1..723885b54fd 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveAllTPUEmbeddingParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveAllTPUEmbeddingParameters.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -46,6 +47,9 @@ opType = RetrieveAllTPUEmbeddingParameters.OP_NAME, inputsClass = RetrieveAllTPUEmbeddingParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveAllTPUEmbeddingParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParameters.java index b8242705b8e..3b2c5c54d62 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingADAMParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -41,6 +42,9 @@ opType = RetrieveTPUEmbeddingADAMParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingADAMParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingADAMParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParameters.java index b59880a136d..39fc4708c75 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdadeltaParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -41,6 +42,9 @@ opType = RetrieveTPUEmbeddingAdadeltaParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingAdadeltaParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingAdadeltaParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradMomentumParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradMomentumParameters.java index f13e816cb5a..29120ef3f9b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradMomentumParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradMomentumParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -41,6 +42,9 @@ opType = RetrieveTPUEmbeddingAdagradMomentumParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingAdagradMomentumParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingAdagradMomentumParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParameters.java index 4e28c761079..9ec5823b9b0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingAdagradParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -41,6 +42,9 @@ opType = RetrieveTPUEmbeddingAdagradParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingAdagradParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingAdagradParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingCenteredRMSPropParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingCenteredRMSPropParameters.java index 99d91c272b9..321d91d8acc 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingCenteredRMSPropParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingCenteredRMSPropParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -41,6 +42,9 @@ opType = RetrieveTPUEmbeddingCenteredRMSPropParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingCenteredRMSPropParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingCenteredRMSPropParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParameters.java index e5f0620bbe1..712be9b6d8f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFTRLParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -41,6 +42,9 @@ opType = RetrieveTPUEmbeddingFTRLParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingFTRLParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingFTRLParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFrequencyEstimatorParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFrequencyEstimatorParameters.java index 728e7f38dfe..1f8d062c7d3 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFrequencyEstimatorParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingFrequencyEstimatorParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -41,6 +42,9 @@ opType = RetrieveTPUEmbeddingFrequencyEstimatorParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingFrequencyEstimatorParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingFrequencyEstimatorParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMDLAdagradLightParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMDLAdagradLightParameters.java index f72bf50f310..ab167935ad5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMDLAdagradLightParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMDLAdagradLightParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -41,6 +42,9 @@ opType = RetrieveTPUEmbeddingMDLAdagradLightParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingMDLAdagradLightParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingMDLAdagradLightParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParameters.java index 53e683984ca..747f704c7a4 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingMomentumParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -41,6 +42,9 @@ opType = RetrieveTPUEmbeddingMomentumParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingMomentumParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingMomentumParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParameters.java index 6337bdae4b4..7bcac9a6f1d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalAdagradParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -41,6 +42,9 @@ opType = RetrieveTPUEmbeddingProximalAdagradParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingProximalAdagradParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingProximalAdagradParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParameters.java index 943b8002368..ab801fa3778 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingProximalYogiParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -37,6 +38,9 @@ opType = RetrieveTPUEmbeddingProximalYogiParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingProximalYogiParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingProximalYogiParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParameters.java index ddb134f6cc2..94904073f5b 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingRMSPropParameters.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -41,6 +42,9 @@ opType = RetrieveTPUEmbeddingRMSPropParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingRMSPropParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingRMSPropParameters extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParameters.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParameters.java index 209a907d72b..b64685d5b06 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParameters.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/RetrieveTPUEmbeddingStochasticGradientDescentParameters.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -42,6 +43,9 @@ opType = RetrieveTPUEmbeddingStochasticGradientDescentParameters.OP_NAME, inputsClass = RetrieveTPUEmbeddingStochasticGradientDescentParameters.Inputs.class ) +@Operator( + group = "tpu" +) public final class RetrieveTPUEmbeddingStochasticGradientDescentParameters extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients.java index 3fa2fcc1270..ec244c2d780 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -38,6 +39,9 @@ opType = SendTPUEmbeddingGradients.OP_NAME, inputsClass = SendTPUEmbeddingGradients.Inputs.class ) +@Operator( + group = "tpu" +) public final class SendTPUEmbeddingGradients extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ShutdownDistributedTPU.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ShutdownDistributedTPU.java index 39db94bffdc..d3a002317a0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ShutdownDistributedTPU.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/ShutdownDistributedTPU.java @@ -27,6 +27,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; /** * Shuts down a running distributed TPU system. @@ -36,6 +37,9 @@ opType = ShutdownDistributedTPU.OP_NAME, inputsClass = ShutdownDistributedTPU.Inputs.class ) +@Operator( + group = "tpu" +) public final class ShutdownDistributedTPU extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SplitDedupData.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SplitDedupData.java index 4e6fa0d8097..ad72b480077 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SplitDedupData.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/SplitDedupData.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -49,6 +50,9 @@ opType = SplitDedupData.OP_NAME, inputsClass = SplitDedupData.Inputs.class ) +@Operator( + group = "tpu" +) public final class SplitDedupData extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUCompilationResult.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUCompilationResult.java index 8afe4ae97e2..6e33eb5f4c0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUCompilationResult.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUCompilationResult.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** @@ -44,6 +45,9 @@ inputsClass = TPUCompilationResult.Inputs.class ) @Deprecated +@Operator( + group = "tpu" +) public final class TPUCompilationResult extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUEmbeddingActivations.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUEmbeddingActivations.java index de0a62dc4aa..29a1e429e53 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUEmbeddingActivations.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUEmbeddingActivations.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; /** @@ -46,6 +47,9 @@ inputsClass = TPUEmbeddingActivations.Inputs.class ) @Deprecated +@Operator( + group = "tpu" +) public final class TPUEmbeddingActivations extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicateMetadata.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicateMetadata.java index bf2065bdb0d..1e5fab82839 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicateMetadata.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicateMetadata.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; /** * Metadata indicating how the TPU computation should be replicated. @@ -40,6 +41,9 @@ inputsClass = TPUReplicateMetadata.Inputs.class ) @Deprecated +@Operator( + group = "tpu" +) public final class TPUReplicateMetadata extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedInput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedInput.java index 92b9ab4f117..1816bb842df 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedInput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedInput.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -55,6 +56,9 @@ inputsClass = TPUReplicatedInput.Inputs.class ) @Deprecated +@Operator( + group = "tpu" +) public final class TPUReplicatedInput extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedOutput.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedOutput.java index e10330b839d..ea53c36f109 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedOutput.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReplicatedOutput.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -54,6 +55,9 @@ inputsClass = TPUReplicatedOutput.Inputs.class ) @Deprecated +@Operator( + group = "tpu" +) public final class TPUReplicatedOutput extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReshardVariables.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReshardVariables.java index c513012f920..c1ddadbc8a5 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReshardVariables.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/TPUReshardVariables.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -44,6 +45,9 @@ opType = TPUReshardVariables.OP_NAME, inputsClass = TPUReshardVariables.Inputs.class ) +@Operator( + group = "tpu" +) public final class TPUReshardVariables extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/WorkerHeartbeat.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/WorkerHeartbeat.java index 690ac095ebc..46a9eaba027 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/WorkerHeartbeat.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/tpu/WorkerHeartbeat.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; /** @@ -40,6 +41,9 @@ opType = WorkerHeartbeat.OP_NAME, inputsClass = WorkerHeartbeat.Inputs.class ) +@Operator( + group = "tpu" +) public final class WorkerHeartbeat extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdaMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdaMax.java index b2b297a82d5..3745c1a0dc0 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdaMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdaMax.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -44,6 +45,9 @@ opType = ApplyAdaMax.OP_NAME, inputsClass = ApplyAdaMax.Inputs.class ) +@Operator( + group = "train" +) public final class ApplyAdaMax extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradV2.java index f13433b87e8..abb8cf13c08 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ApplyAdagradV2.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -43,6 +44,9 @@ opType = ApplyAdagradV2.OP_NAME, inputsClass = ApplyAdagradV2.Inputs.class ) +@Operator( + group = "train" +) public final class ApplyAdagradV2 extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ComputeBatchSize.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ComputeBatchSize.java index 45232b68afd..1e6930d5410 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ComputeBatchSize.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ComputeBatchSize.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = ComputeBatchSize.OP_NAME, inputsClass = ComputeBatchSize.Inputs.class ) +@Operator( + group = "train" +) public final class ComputeBatchSize extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/DistributedSave.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/DistributedSave.java index 7046d3980a8..5572cc4a03f 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/DistributedSave.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/DistributedSave.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TString; import org.tensorflow.types.family.TType; @@ -38,6 +39,9 @@ opType = DistributedSave.OP_NAME, inputsClass = DistributedSave.Inputs.class ) +@Operator( + group = "train" +) public final class DistributedSave extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorApplyGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorApplyGradient.java index ffaa98b7117..a13e18fe22a 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorApplyGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorApplyGradient.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -40,6 +41,9 @@ opType = ResourceAccumulatorApplyGradient.OP_NAME, inputsClass = ResourceAccumulatorApplyGradient.Inputs.class ) +@Operator( + group = "train" +) public final class ResourceAccumulatorApplyGradient extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorNumAccumulated.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorNumAccumulated.java index 040494fc7be..639f54c9b2d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorNumAccumulated.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorNumAccumulated.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -39,6 +40,9 @@ opType = ResourceAccumulatorNumAccumulated.OP_NAME, inputsClass = ResourceAccumulatorNumAccumulated.Inputs.class ) +@Operator( + group = "train" +) public final class ResourceAccumulatorNumAccumulated extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorSetGlobalStep.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorSetGlobalStep.java index a590fb51675..9b73fa3b03d 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorSetGlobalStep.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorSetGlobalStep.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TInt64; import org.tensorflow.types.family.TType; @@ -40,6 +41,9 @@ opType = ResourceAccumulatorSetGlobalStep.OP_NAME, inputsClass = ResourceAccumulatorSetGlobalStep.Inputs.class ) +@Operator( + group = "train" +) public final class ResourceAccumulatorSetGlobalStep extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorTakeGradient.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorTakeGradient.java index f494f6c7987..4b7a918f597 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorTakeGradient.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceAccumulatorTakeGradient.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.TInt32; import org.tensorflow.types.family.TType; @@ -48,6 +49,9 @@ opType = ResourceAccumulatorTakeGradient.OP_NAME, inputsClass = ResourceAccumulatorTakeGradient.Inputs.class ) +@Operator( + group = "train" +) public final class ResourceAccumulatorTakeGradient extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdaMax.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdaMax.java index 0f32bcb8f9d..b6cc57f3185 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdaMax.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdaMax.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -41,6 +42,9 @@ opType = ResourceApplyAdaMax.OP_NAME, inputsClass = ResourceApplyAdaMax.Inputs.class ) +@Operator( + group = "train" +) public final class ResourceApplyAdaMax extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagrad.java index c2ced603825..e0bc3b11020 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceApplyAdagrad.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -40,6 +41,9 @@ opType = ResourceApplyAdagrad.OP_NAME, inputsClass = ResourceApplyAdagrad.Inputs.class ) +@Operator( + group = "train" +) public final class ResourceApplyAdagrad extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceConditionalAccumulator.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceConditionalAccumulator.java index 251568751c4..93da410ed51 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceConditionalAccumulator.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceConditionalAccumulator.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -49,6 +50,9 @@ opType = ResourceConditionalAccumulator.OP_NAME, inputsClass = ResourceConditionalAccumulator.Inputs.class ) +@Operator( + group = "train" +) public final class ResourceConditionalAccumulator extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradV2.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradV2.java index b4fa0c1c251..7bc56093246 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradV2.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/ResourceSparseApplyAdagradV2.java @@ -28,6 +28,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -42,6 +43,9 @@ opType = ResourceSparseApplyAdagradV2.OP_NAME, inputsClass = ResourceSparseApplyAdagradV2.Inputs.class ) +@Operator( + group = "train" +) public final class ResourceSparseApplyAdagradV2 extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaOptimizer.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaOptimizer.java index f03ddc35c32..e06ddf1dae1 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaOptimizer.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SdcaOptimizer.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.TInt64; @@ -54,6 +55,9 @@ opType = SdcaOptimizer.OP_NAME, inputsClass = SdcaOptimizer.Inputs.class ) +@Operator( + group = "train" +) public final class SdcaOptimizer extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagrad.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagrad.java index 0742d6de727..b53c5b6f540 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagrad.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/train/SparseApplyAdagrad.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TNumber; import org.tensorflow.types.family.TType; @@ -45,6 +46,9 @@ opType = SparseApplyAdagrad.OP_NAME, inputsClass = SparseApplyAdagrad.Inputs.class ) +@Operator( + group = "train" +) public final class SparseApplyAdagrad extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/AssignVariableConcatND.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/AssignVariableConcatND.java index ef35e623500..51f000b2687 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/AssignVariableConcatND.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/AssignVariableConcatND.java @@ -30,6 +30,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -69,6 +70,9 @@ opType = AssignVariableConcatND.OP_NAME, inputsClass = AssignVariableConcatND.Inputs.class ) +@Operator( + group = "xla" +) public final class AssignVariableConcatND extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ConcatND.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ConcatND.java index 64bcaa1effb..7e55c95e679 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ConcatND.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ConcatND.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -72,6 +73,9 @@ opType = ConcatND.OP_NAME, inputsClass = ConcatND.Inputs.class ) +@Operator( + group = "xla" +) public final class ConcatND extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReadVariableSplitND.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReadVariableSplitND.java index 2839535884d..666103dd273 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReadVariableSplitND.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/ReadVariableSplitND.java @@ -32,6 +32,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -73,6 +74,9 @@ opType = ReadVariableSplitND.OP_NAME, inputsClass = ReadVariableSplitND.Inputs.class ) +@Operator( + group = "xla" +) public final class ReadVariableSplitND extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SplitND.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SplitND.java index ee8a3a6b29d..6bf5656f68c 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SplitND.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/SplitND.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.proto.DataType; import org.tensorflow.types.family.TType; @@ -72,6 +73,9 @@ opType = SplitND.OP_NAME, inputsClass = SplitND.Inputs.class ) +@Operator( + group = "xla" +) public final class SplitND extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvTPUEmbeddingActivations.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvTPUEmbeddingActivations.java index 7aa0bedf6a4..1af06a9de56 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvTPUEmbeddingActivations.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvTPUEmbeddingActivations.java @@ -31,6 +31,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TType; @@ -45,6 +46,9 @@ opType = XlaRecvTPUEmbeddingActivations.OP_NAME, inputsClass = XlaRecvTPUEmbeddingActivations.Inputs.class ) +@Operator( + group = "xla" +) public final class XlaRecvTPUEmbeddingActivations extends RawOp implements Iterable> { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvTPUEmbeddingDeduplicationData.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvTPUEmbeddingDeduplicationData.java index 8c0e285d222..abf76b9a0ad 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvTPUEmbeddingDeduplicationData.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaRecvTPUEmbeddingDeduplicationData.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.family.TType; /** @@ -44,6 +45,9 @@ opType = XlaRecvTPUEmbeddingDeduplicationData.OP_NAME, inputsClass = XlaRecvTPUEmbeddingDeduplicationData.Inputs.class ) +@Operator( + group = "xla" +) public final class XlaRecvTPUEmbeddingDeduplicationData extends RawOp implements Operand { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSendTPUEmbeddingGradients.java b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSendTPUEmbeddingGradients.java index 01376fa6683..3b4013ebc92 100644 --- a/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSendTPUEmbeddingGradients.java +++ b/tensorflow-core/tensorflow-core-api/src/gen/java/org/tensorflow/op/xla/XlaSendTPUEmbeddingGradients.java @@ -29,6 +29,7 @@ import org.tensorflow.op.annotation.Endpoint; import org.tensorflow.op.annotation.OpInputsMetadata; import org.tensorflow.op.annotation.OpMetadata; +import org.tensorflow.op.annotation.Operator; import org.tensorflow.types.TFloat32; import org.tensorflow.types.family.TType; @@ -44,6 +45,9 @@ opType = XlaSendTPUEmbeddingGradients.OP_NAME, inputsClass = XlaSendTPUEmbeddingGradients.Inputs.class ) +@Operator( + group = "xla" +) public final class XlaSendTPUEmbeddingGradients extends RawOp { /** * The name of this op, as known by TensorFlow core engine diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/module-info.java b/tensorflow-core/tensorflow-core-api/src/main/java/module-info.java index 9bd62485a8e..f3e4875193d 100644 --- a/tensorflow-core/tensorflow-core-api/src/main/java/module-info.java +++ b/tensorflow-core/tensorflow-core-api/src/main/java/module-info.java @@ -34,7 +34,6 @@ exports org.tensorflow.op.data.experimental; exports org.tensorflow.op.debugging; exports org.tensorflow.op.dtypes; - exports org.tensorflow.op.estimator; exports org.tensorflow.op.image; exports org.tensorflow.op.io; exports org.tensorflow.op.linalg; @@ -45,7 +44,6 @@ exports org.tensorflow.op.quantization; exports org.tensorflow.op.ragged; exports org.tensorflow.op.random; - exports org.tensorflow.op.risc; exports org.tensorflow.op.signal; exports org.tensorflow.op.sparse; exports org.tensorflow.op.strings;