public interface MemoryLogTensorAllocationOrBuilder
extends org.nd4j.shade.protobuf.MessageOrBuilder
Modifier and Type | Method and Description |
---|---|
String |
getKernelName()
Name of the kernel making the allocation as set in GraphDef,
e.g., "affine2/weights/Assign".
|
org.nd4j.shade.protobuf.ByteString |
getKernelNameBytes()
Name of the kernel making the allocation as set in GraphDef,
e.g., "affine2/weights/Assign".
|
long |
getStepId()
Process-unique step id.
|
TensorDescription |
getTensor()
Allocated tensor details.
|
TensorDescriptionOrBuilder |
getTensorOrBuilder()
Allocated tensor details.
|
boolean |
hasTensor()
Allocated tensor details.
|
findInitializationErrors, getAllFields, getDefaultInstanceForType, getDescriptorForType, getField, getInitializationErrorString, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, getUnknownFields, hasField, hasOneof
long getStepId()
Process-unique step id.
int64 step_id = 1;
String getKernelName()
Name of the kernel making the allocation as set in GraphDef, e.g., "affine2/weights/Assign".
string kernel_name = 2;
org.nd4j.shade.protobuf.ByteString getKernelNameBytes()
Name of the kernel making the allocation as set in GraphDef, e.g., "affine2/weights/Assign".
string kernel_name = 2;
boolean hasTensor()
Allocated tensor details.
.tensorflow.TensorDescription tensor = 3;
TensorDescription getTensor()
Allocated tensor details.
.tensorflow.TensorDescription tensor = 3;
TensorDescriptionOrBuilder getTensorOrBuilder()
Allocated tensor details.
.tensorflow.TensorDescription tensor = 3;
Copyright © 2019. All rights reserved.