@Namespace(value="tensorflow") @NoOffset @Properties(inherit=tensorflow.class) public class CollectiveContext extends Pointer
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
CollectiveContext(CollectiveExecutor col_exec,
DeviceMgr dev_mgr,
OpKernelContext ctx,
OpKernelContext.Params op_params,
CollectiveParams col_params,
BytePointer exec_key,
long step_id,
Tensor input,
Tensor output) |
CollectiveContext(CollectiveExecutor col_exec,
DeviceMgr dev_mgr,
OpKernelContext ctx,
OpKernelContext.Params op_params,
CollectiveParams col_params,
String exec_key,
long step_id,
Tensor input,
Tensor output) |
CollectiveContext(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
CollectiveExecutor |
col_exec() |
CollectiveContext |
col_exec(CollectiveExecutor setter) |
CollectiveParams |
col_params() |
DeviceMgr |
dev_mgr() |
CollectiveContext |
dev_mgr(DeviceMgr setter) |
DeviceLocality |
device_locality() |
CollectiveContext |
device_locality(DeviceLocality setter) |
BytePointer |
device_name() |
Device |
device() |
CollectiveContext |
device(Device setter) |
BytePointer |
exec_key() |
Tensor |
input() |
CollectiveContext |
input(Tensor setter) |
OpKernelContext |
op_ctx() |
CollectiveContext |
op_ctx(OpKernelContext setter) |
OpKernelContext.Params |
op_params() |
CollectiveContext |
op_params(OpKernelContext.Params setter) |
Tensor |
output() |
CollectiveContext |
output(Tensor setter) |
long |
step_id() |
address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero
public CollectiveContext(Pointer p)
Pointer(Pointer)
.public CollectiveContext(CollectiveExecutor col_exec, @Const DeviceMgr dev_mgr, OpKernelContext ctx, OpKernelContext.Params op_params, @Const @ByRef CollectiveParams col_params, @StdString BytePointer exec_key, @Cast(value="tensorflow::int64") long step_id, @Const Tensor input, Tensor output)
public CollectiveContext(CollectiveExecutor col_exec, @Const DeviceMgr dev_mgr, OpKernelContext ctx, OpKernelContext.Params op_params, @Const @ByRef CollectiveParams col_params, @StdString String exec_key, @Cast(value="tensorflow::int64") long step_id, @Const Tensor input, Tensor output)
public CollectiveExecutor col_exec()
public CollectiveContext col_exec(CollectiveExecutor setter)
public CollectiveContext dev_mgr(DeviceMgr setter)
public OpKernelContext op_ctx()
public CollectiveContext op_ctx(OpKernelContext setter)
public OpKernelContext.Params op_params()
public CollectiveContext op_params(OpKernelContext.Params setter)
@MemberGetter @Const @ByRef public CollectiveParams col_params()
@MemberGetter @StdString public BytePointer exec_key()
@MemberGetter @Cast(value="const tensorflow::int64") public long step_id()
public CollectiveContext input(Tensor setter)
public Tensor output()
public CollectiveContext output(Tensor setter)
public Device device()
public CollectiveContext device(Device setter)
@MemberGetter @StdString public BytePointer device_name()
@ByRef public DeviceLocality device_locality()
public CollectiveContext device_locality(DeviceLocality setter)
Copyright © 2022. All rights reserved.