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
Show file tree
Hide file tree
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
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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());
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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());
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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());
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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());
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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());
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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());
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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());
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down
Loading