Package | Description |
---|---|
org.bytedeco.tensorflow | |
org.bytedeco.tensorflow.global |
Modifier and Type | Class and Description |
---|---|
class |
AllocatorWrapper |
class |
TrackingAllocator |
Modifier and Type | Method and Description |
---|---|
Allocator |
WrappedAllocator.first() |
Allocator |
WrappedAllocator.first() |
Allocator |
OpKernelContext.get_allocator(AllocatorAttributes attr) |
Allocator |
OpKernelContext.get_allocator(AllocatorAttributes attr) |
Allocator |
DeviceBase.GetAllocator(AllocatorAttributes arg0) |
Allocator |
DeviceBase.GetAllocator(AllocatorAttributes arg0) |
Allocator |
DeviceBase.GetScopedAllocator(AllocatorAttributes attr,
long step_id) |
Allocator |
DeviceBase.GetScopedAllocator(AllocatorAttributes attr,
long step_id) |
Allocator |
DeviceBase.GetStepAllocator(AllocatorAttributes attr,
ResourceMgr arg1) |
Allocator |
DeviceBase.GetStepAllocator(AllocatorAttributes attr,
ResourceMgr arg1) |
Allocator |
AllocatorWrapper.wrapped() |
Allocator |
AllocatorWrapper.wrapped() |
Modifier and Type | Method and Description |
---|---|
WrappedAllocator |
WrappedAllocator.first(Allocator first) |
WrappedAllocator |
WrappedAllocator.first(Allocator first) |
boolean |
Tensor.FromProto(Allocator a,
TensorProto other) |
boolean |
Tensor.FromProto(Allocator a,
TensorProto other) |
WrappedAllocator |
WrappedAllocator.put(Allocator firstValue,
TrackingAllocator secondValue) |
WrappedAllocator |
WrappedAllocator.put(Allocator firstValue,
TrackingAllocator secondValue) |
Status |
DeviceBase.ReinitializeGpuDevice(OpKernelContext arg0,
PerOpGpuDevice arg1,
DeviceContext arg2,
Allocator arg3) |
Status |
DeviceBase.ReinitializeGpuDevice(OpKernelContext arg0,
PerOpGpuDevice arg1,
DeviceContext arg2,
Allocator arg3) |
Constructor and Description |
---|
AllocatorWrapper(Allocator wrapped) |
AllocatorWrapper(Allocator wrapped) |
OpKernelConstruction(DeviceType device_type,
DeviceBase device,
Allocator allocator,
NodeDef node_def,
OpDef op_def,
FunctionLibraryRuntime flib,
DataTypeVector input_types,
StringAttrPairVector input_memory_types,
DataTypeVector output_types,
StringAttrPairVector output_memory_types,
int graph_def_version,
Status status) |
OpKernelConstruction(DeviceType device_type,
DeviceBase device,
Allocator allocator,
NodeDef node_def,
OpDef op_def,
FunctionLibraryRuntime flib,
DataTypeVector input_types,
StringAttrPairVector input_memory_types,
DataTypeVector output_types,
StringAttrPairVector output_memory_types,
int graph_def_version,
Status status) |
Tensor(Allocator a,
int type,
TensorShape shape)
\brief Creates a tensor with the input
type and shape , using
the allocator a to allocate the underlying buffer. |
Tensor(Allocator a,
int type,
TensorShape shape)
\brief Creates a tensor with the input
type and shape , using
the allocator a to allocate the underlying buffer. |
Tensor(Allocator a,
int type,
TensorShape shape,
AllocationAttributes allocation_attr)
\brief Creates a tensor with the input
type and shape , using
the allocator a and the specified "allocation_attr" to
allocate the underlying buffer. |
Tensor(Allocator a,
int type,
TensorShape shape,
AllocationAttributes allocation_attr)
\brief Creates a tensor with the input
type and shape , using
the allocator a and the specified "allocation_attr" to
allocate the underlying buffer. |
WrappedAllocator(Allocator firstValue,
TrackingAllocator secondValue) |
WrappedAllocator(Allocator firstValue,
TrackingAllocator secondValue) |
Modifier and Type | Method and Description |
---|---|
static Allocator |
tensorflow.cpu_allocator_base() |
static Allocator |
tensorflow.cpu_allocator_base() |
static Allocator |
tensorflow.cpu_allocator() |
static Allocator |
tensorflow.cpu_allocator() |
static Allocator |
tensorflow.cpu_allocator(int numa_node) |
static Allocator |
tensorflow.cpu_allocator(int numa_node) |
Modifier and Type | Method and Description |
---|---|
static Pointer |
tensorflow.allocate_tensor(BytePointer operation,
long len,
Allocator allocator) |
static Pointer |
tensorflow.allocate_tensor(BytePointer operation,
long len,
Allocator allocator) |
static Pointer |
tensorflow.allocate_tensor(String operation,
long len,
Allocator allocator) |
static Pointer |
tensorflow.allocate_tensor(String operation,
long len,
Allocator allocator) |
static Status |
tensorflow.CreateOpKernel(DeviceType device_type,
DeviceBase device,
Allocator allocator,
FunctionLibraryRuntime flib,
NodeDef def,
int graph_def_version,
OpKernel kernel) |
static Status |
tensorflow.CreateOpKernel(DeviceType device_type,
DeviceBase device,
Allocator allocator,
FunctionLibraryRuntime flib,
NodeDef def,
int graph_def_version,
OpKernel kernel) |
static Status |
tensorflow.CreateOpKernel(DeviceType device_type,
DeviceBase device,
Allocator allocator,
FunctionLibraryRuntime flib,
NodeDef def,
int graph_def_version,
PointerPointer kernel) |
static Status |
tensorflow.CreateOpKernel(DeviceType device_type,
DeviceBase device,
Allocator allocator,
FunctionLibraryRuntime flib,
NodeDef def,
int graph_def_version,
PointerPointer kernel) |
static OpKernel |
tensorflow.CreateOpKernel(DeviceType device_type,
DeviceBase device,
Allocator allocator,
NodeDef def,
int graph_def_version,
Status status) |
static OpKernel |
tensorflow.CreateOpKernel(DeviceType device_type,
DeviceBase device,
Allocator allocator,
NodeDef def,
int graph_def_version,
Status status) |
Copyright © 2022. All rights reserved.