Skip to content

Metrics Add init(Ops) method #337

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 7 commits into from
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
1 change: 1 addition & 0 deletions tensorflow-core/tensorflow-core-api/.bazelversion
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
3.7.2
Original file line number Diff line number Diff line change
@@ -84,7 +84,7 @@ private AudioSpectrogram(Operation operation) {
)
public static AudioSpectrogram create(Scope scope, Operand<TFloat32> input, Long windowSize,
Long stride, Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("AudioSpectrogram", scope.makeOpName("AudioSpectrogram"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("AudioSpectrogram"));
opBuilder.addInput(input.asOutput());
opBuilder = scope.apply(opBuilder);
opBuilder.setAttr("window_size", windowSize);
Original file line number Diff line number Diff line change
@@ -75,7 +75,7 @@ private DecodeWav(Operation operation) {
describeByClass = true
)
public static DecodeWav create(Scope scope, Operand<TString> contents, Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("DecodeWav", scope.makeOpName("DecodeWav"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("DecodeWav"));
opBuilder.addInput(contents.asOutput());
opBuilder = scope.apply(opBuilder);
if (options != null) {
Original file line number Diff line number Diff line change
@@ -67,7 +67,7 @@ private EncodeWav(Operation operation) {
describeByClass = true
)
public static EncodeWav create(Scope scope, Operand<TFloat32> audio, Operand<TInt32> sampleRate) {
OperationBuilder opBuilder = scope.env().opBuilder("EncodeWav", scope.makeOpName("EncodeWav"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("EncodeWav"));
opBuilder.addInput(audio.asOutput());
opBuilder.addInput(sampleRate.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -69,7 +69,7 @@ private Mfcc(Operation operation) {
)
public static Mfcc create(Scope scope, Operand<TFloat32> spectrogram, Operand<TInt32> sampleRate,
Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("Mfcc", scope.makeOpName("Mfcc"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("Mfcc"));
opBuilder.addInput(spectrogram.asOutput());
opBuilder.addInput(sampleRate.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -79,7 +79,7 @@ private BitwiseAnd(Operation operation) {
describeByClass = true
)
public static <T extends TNumber> BitwiseAnd<T> create(Scope scope, Operand<T> x, Operand<T> y) {
OperationBuilder opBuilder = scope.env().opBuilder("BitwiseAnd", scope.makeOpName("BitwiseAnd"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("BitwiseAnd"));
opBuilder.addInput(x.asOutput());
opBuilder.addInput(y.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -79,7 +79,7 @@ private BitwiseOr(Operation operation) {
describeByClass = true
)
public static <T extends TNumber> BitwiseOr<T> create(Scope scope, Operand<T> x, Operand<T> y) {
OperationBuilder opBuilder = scope.env().opBuilder("BitwiseOr", scope.makeOpName("BitwiseOr"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("BitwiseOr"));
opBuilder.addInput(x.asOutput());
opBuilder.addInput(y.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -79,7 +79,7 @@ private BitwiseXor(Operation operation) {
describeByClass = true
)
public static <T extends TNumber> BitwiseXor<T> create(Scope scope, Operand<T> x, Operand<T> y) {
OperationBuilder opBuilder = scope.env().opBuilder("BitwiseXor", scope.makeOpName("BitwiseXor"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("BitwiseXor"));
opBuilder.addInput(x.asOutput());
opBuilder.addInput(y.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -99,7 +99,7 @@ private Invert(Operation operation) {
describeByClass = true
)
public static <T extends TNumber> Invert<T> create(Scope scope, Operand<T> x) {
OperationBuilder opBuilder = scope.env().opBuilder("Invert", scope.makeOpName("Invert"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("Invert"));
opBuilder.addInput(x.asOutput());
opBuilder = scope.apply(opBuilder);
return new Invert<>(opBuilder.build());
Original file line number Diff line number Diff line change
@@ -90,7 +90,7 @@ private LeftShift(Operation operation) {
describeByClass = true
)
public static <T extends TNumber> LeftShift<T> create(Scope scope, Operand<T> x, Operand<T> y) {
OperationBuilder opBuilder = scope.env().opBuilder("LeftShift", scope.makeOpName("LeftShift"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("LeftShift"));
opBuilder.addInput(x.asOutput());
opBuilder.addInput(y.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -92,7 +92,7 @@ private RightShift(Operation operation) {
describeByClass = true
)
public static <T extends TNumber> RightShift<T> create(Scope scope, Operand<T> x, Operand<T> y) {
OperationBuilder opBuilder = scope.env().opBuilder("RightShift", scope.makeOpName("RightShift"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("RightShift"));
opBuilder.addInput(x.asOutput());
opBuilder.addInput(y.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -62,7 +62,7 @@ private KMC2ChainInitialization(Operation operation) {
)
public static KMC2ChainInitialization create(Scope scope, Operand<TFloat32> distances,
Operand<TInt64> seed) {
OperationBuilder opBuilder = scope.env().opBuilder("KMC2ChainInitialization", scope.makeOpName("KMC2ChainInitialization"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("KMC2ChainInitialization"));
opBuilder.addInput(distances.asOutput());
opBuilder.addInput(seed.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -66,7 +66,7 @@ private KmeansPlusPlusInitialization(Operation operation) {
)
public static KmeansPlusPlusInitialization create(Scope scope, Operand<TFloat32> points,
Operand<TInt64> numToSample, Operand<TInt64> seed, Operand<TInt64> numRetriesPerSample) {
OperationBuilder opBuilder = scope.env().opBuilder("KmeansPlusPlusInitialization", scope.makeOpName("KmeansPlusPlusInitialization"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("KmeansPlusPlusInitialization"));
opBuilder.addInput(points.asOutput());
opBuilder.addInput(numToSample.asOutput());
opBuilder.addInput(seed.asOutput());
Original file line number Diff line number Diff line change
@@ -71,7 +71,7 @@ private AllReduce(Operation operation) {
public static <T extends TNumber> AllReduce<T> create(Scope scope, Operand<T> input,
Long groupSize, Long groupKey, Long instanceKey, String mergeOp, String finalOp,
List<Long> subdivOffsets, Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("CollectiveReduce", scope.makeOpName("AllReduce"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("AllReduce"));
opBuilder.addInput(input.asOutput());
opBuilder = scope.apply(opBuilder);
opBuilder.setAttr("group_size", groupSize);
Original file line number Diff line number Diff line change
@@ -65,7 +65,7 @@ private BroadcastRecv(Operation operation) {
)
public static <T extends TType> BroadcastRecv<T> create(Scope scope, Class<T> T, Long groupSize,
Long groupKey, Long instanceKey, Shape shape, Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("CollectiveBcastRecv", scope.makeOpName("BroadcastRecv"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("BroadcastRecv"));
opBuilder = scope.apply(opBuilder);
opBuilder.setAttr("T", Operands.toDataType(T));
opBuilder.setAttr("group_size", groupSize);
Original file line number Diff line number Diff line change
@@ -64,7 +64,7 @@ private BroadcastSend(Operation operation) {
)
public static <T extends TType> BroadcastSend<T> create(Scope scope, Operand<T> input,
Long groupSize, Long groupKey, Long instanceKey, Shape shape, Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("CollectiveBcastSend", scope.makeOpName("BroadcastSend"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("BroadcastSend"));
opBuilder.addInput(input.asOutput());
opBuilder = scope.apply(opBuilder);
opBuilder.setAttr("group_size", groupSize);
Original file line number Diff line number Diff line change
@@ -64,7 +64,7 @@ private Gather(Operation operation) {
)
public static <T extends TNumber> Gather<T> create(Scope scope, Operand<T> input, Long groupSize,
Long groupKey, Long instanceKey, Shape shape, Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("CollectiveGather", scope.makeOpName("Gather"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("Gather"));
opBuilder.addInput(input.asOutput());
opBuilder = scope.apply(opBuilder);
opBuilder.setAttr("group_size", groupSize);
Original file line number Diff line number Diff line change
@@ -67,7 +67,7 @@ private GatherV2(Operation operation) {
public static <T extends TNumber> GatherV2<T> create(Scope scope, Operand<T> input,
Operand<TInt32> groupSize, Operand<TInt32> groupKey, Operand<TInt32> instanceKey,
Iterable<Operand<? extends TType>> orderingToken, Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("CollectiveGatherV2", scope.makeOpName("GatherV2"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("GatherV2"));
opBuilder.addInput(input.asOutput());
opBuilder.addInput(groupSize.asOutput());
opBuilder.addInput(groupKey.asOutput());
Original file line number Diff line number Diff line change
@@ -68,7 +68,7 @@ private Reduce(Operation operation) {
public static <T extends TNumber> Reduce<T> create(Scope scope, Operand<T> input, Long groupSize,
Long groupKey, Long instanceKey, String mergeOp, String finalOp, List<Long> subdivOffsets,
Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("CollectiveReduce", scope.makeOpName("Reduce"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("Reduce"));
opBuilder.addInput(input.asOutput());
opBuilder = scope.apply(opBuilder);
opBuilder.setAttr("group_size", groupSize);
Original file line number Diff line number Diff line change
@@ -70,7 +70,7 @@ public static <T extends TNumber> ReduceV2<T> create(Scope scope, Operand<T> inp
Operand<TInt32> groupSize, Operand<TInt32> groupKey, Operand<TInt32> instanceKey,
Iterable<Operand<? extends TType>> orderingToken, String mergeOp, String finalOp,
Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("CollectiveReduceV2", scope.makeOpName("ReduceV2"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("ReduceV2"));
opBuilder.addInput(input.asOutput());
opBuilder.addInput(groupSize.asOutput());
opBuilder.addInput(groupKey.asOutput());
Original file line number Diff line number Diff line change
@@ -52,7 +52,7 @@ private Abort(Operation operation) {
describeByClass = true
)
public static Abort create(Scope scope, Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("Abort", scope.makeOpName("Abort"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("Abort"));
opBuilder = scope.apply(opBuilder);
if (options != null) {
for (Options opts : options) {
Original file line number Diff line number Diff line change
@@ -65,7 +65,7 @@ private All(Operation operation) {
)
public static All create(Scope scope, Operand<TBool> input, Operand<? extends TNumber> axis,
Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("All", scope.makeOpName("All"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("All"));
opBuilder.addInput(input.asOutput());
opBuilder.addInput(axis.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -65,7 +65,7 @@ private Any(Operation operation) {
)
public static Any create(Scope scope, Operand<TBool> input, Operand<? extends TNumber> axis,
Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("Any", scope.makeOpName("Any"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("Any"));
opBuilder.addInput(input.asOutput());
opBuilder.addInput(axis.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -57,7 +57,7 @@ private AssertThat(Operation operation) {
)
public static AssertThat create(Scope scope, Operand<TBool> condition, Iterable<Operand<?>> data,
Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("Assert", scope.makeOpName("AssertThat"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("AssertThat"));
opBuilder.addInput(condition.asOutput());
opBuilder.addInputList(Operands.asOutputs(data));
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -64,7 +64,7 @@ private Assign(Operation operation) {
)
public static <T extends TType> Assign<T> create(Scope scope, Operand<T> ref, Operand<T> value,
Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("Assign", scope.makeOpName("Assign"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("Assign"));
opBuilder.addInput(ref.asOutput());
opBuilder.addInput(value.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -64,7 +64,7 @@ private AssignAdd(Operation operation) {
)
public static <T extends TType> AssignAdd<T> create(Scope scope, Operand<T> ref, Operand<T> value,
Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("AssignAdd", scope.makeOpName("AssignAdd"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("AssignAdd"));
opBuilder.addInput(ref.asOutput());
opBuilder.addInput(value.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -55,7 +55,7 @@ private AssignAddVariableOp(Operation operation) {
)
public static AssignAddVariableOp create(Scope scope, Operand<? extends TType> resource,
Operand<? extends TType> value) {
OperationBuilder opBuilder = scope.env().opBuilder("AssignAddVariableOp", scope.makeOpName("AssignAddVariableOp"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("AssignAddVariableOp"));
opBuilder.addInput(resource.asOutput());
opBuilder.addInput(value.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -64,7 +64,7 @@ private AssignSub(Operation operation) {
)
public static <T extends TType> AssignSub<T> create(Scope scope, Operand<T> ref, Operand<T> value,
Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("AssignSub", scope.makeOpName("AssignSub"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("AssignSub"));
opBuilder.addInput(ref.asOutput());
opBuilder.addInput(value.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -55,7 +55,7 @@ private AssignSubVariableOp(Operation operation) {
)
public static AssignSubVariableOp create(Scope scope, Operand<? extends TType> resource,
Operand<? extends TType> value) {
OperationBuilder opBuilder = scope.env().opBuilder("AssignSubVariableOp", scope.makeOpName("AssignSubVariableOp"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("AssignSubVariableOp"));
opBuilder.addInput(resource.asOutput());
opBuilder.addInput(value.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -55,7 +55,7 @@ private AssignVariableOp(Operation operation) {
)
public static AssignVariableOp create(Scope scope, Operand<? extends TType> resource,
Operand<? extends TType> value) {
OperationBuilder opBuilder = scope.env().opBuilder("AssignVariableOp", scope.makeOpName("AssignVariableOp"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("AssignVariableOp"));
opBuilder.addInput(resource.asOutput());
opBuilder.addInput(value.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -70,7 +70,7 @@ private Barrier(Operation operation) {
)
public static Barrier create(Scope scope, List<Class<? extends TType>> componentTypes,
Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("Barrier", scope.makeOpName("Barrier"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("Barrier"));
opBuilder = scope.apply(opBuilder);
opBuilder.setAttr("component_types", Operands.toDataTypes(componentTypes));
if (options != null) {
Original file line number Diff line number Diff line change
@@ -58,7 +58,7 @@ private BarrierClose(Operation operation) {
describeByClass = true
)
public static BarrierClose create(Scope scope, Operand<TString> handle, Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("BarrierClose", scope.makeOpName("BarrierClose"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("BarrierClose"));
opBuilder.addInput(handle.asOutput());
opBuilder = scope.apply(opBuilder);
if (options != null) {
Original file line number Diff line number Diff line change
@@ -57,7 +57,7 @@ private BarrierIncompleteSize(Operation operation) {
describeByClass = true
)
public static BarrierIncompleteSize create(Scope scope, Operand<TString> handle) {
OperationBuilder opBuilder = scope.env().opBuilder("BarrierIncompleteSize", scope.makeOpName("BarrierIncompleteSize"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("BarrierIncompleteSize"));
opBuilder.addInput(handle.asOutput());
opBuilder = scope.apply(opBuilder);
return new BarrierIncompleteSize(opBuilder.build());
Original file line number Diff line number Diff line change
@@ -61,7 +61,7 @@ private BarrierInsertMany(Operation operation) {
)
public static BarrierInsertMany create(Scope scope, Operand<TString> handle,
Operand<TString> keys, Operand<? extends TType> values, Long componentIndex) {
OperationBuilder opBuilder = scope.env().opBuilder("BarrierInsertMany", scope.makeOpName("BarrierInsertMany"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("BarrierInsertMany"));
opBuilder.addInput(handle.asOutput());
opBuilder.addInput(keys.asOutput());
opBuilder.addInput(values.asOutput());
Original file line number Diff line number Diff line change
@@ -57,7 +57,7 @@ private BarrierReadySize(Operation operation) {
describeByClass = true
)
public static BarrierReadySize create(Scope scope, Operand<TString> handle) {
OperationBuilder opBuilder = scope.env().opBuilder("BarrierReadySize", scope.makeOpName("BarrierReadySize"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("BarrierReadySize"));
opBuilder.addInput(handle.asOutput());
opBuilder = scope.apply(opBuilder);
return new BarrierReadySize(opBuilder.build());
Original file line number Diff line number Diff line change
@@ -83,7 +83,7 @@ private BarrierTakeMany(Operation operation) {
public static BarrierTakeMany create(Scope scope, Operand<TString> handle,
Operand<TInt32> numElements, List<Class<? extends TType>> componentTypes,
Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("BarrierTakeMany", scope.makeOpName("BarrierTakeMany"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("BarrierTakeMany"));
opBuilder.addInput(handle.asOutput());
opBuilder.addInput(numElements.asOutput());
opBuilder = scope.apply(opBuilder);
Original file line number Diff line number Diff line change
@@ -105,7 +105,7 @@ private Batch(Operation operation) {
)
public static Batch create(Scope scope, Iterable<Operand<?>> inTensors, Long numBatchThreads,
Long maxBatchSize, Long batchTimeoutMicros, Long gradTimeoutMicros, Options... options) {
OperationBuilder opBuilder = scope.env().opBuilder("Batch", scope.makeOpName("Batch"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("Batch"));
opBuilder.addInputList(Operands.asOutputs(inTensors));
opBuilder = scope.apply(opBuilder);
opBuilder.setAttr("num_batch_threads", numBatchThreads);
Original file line number Diff line number Diff line change
@@ -76,7 +76,7 @@ private BatchToSpace(Operation operation) {
)
public static <T extends TType> BatchToSpace<T> create(Scope scope, Operand<T> input,
Operand<? extends TNumber> crops, Long blockSize) {
OperationBuilder opBuilder = scope.env().opBuilder("BatchToSpace", scope.makeOpName("BatchToSpace"));
OperationBuilder opBuilder = scope.env().opBuilder(OP_NAME, scope.makeOpName("BatchToSpace"));
opBuilder.addInput(input.asOutput());
opBuilder.addInput(crops.asOutput());
opBuilder = scope.apply(opBuilder);
Loading