Home
last modified time | relevance | path

Searched defs:memory_limit (Results 1 – 25 of 37) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device_factory.cc32 Bytes memory_limit, const DeviceLocality& locality, in GPUDevice()
65 const SessionOptions& options, const string& name, Bytes memory_limit, in CreateGPUDevice()
84 Bytes memory_limit, const DeviceLocality& locality, in GPUCompatibleCPUDevice()
H A Dgpu_device.cc388 Bytes memory_limit, const DeviceLocality& locality, in BaseGPUDevice()
1087 int64_t* memory_limit) { in SingleVirtualDeviceMemoryLimit()
1561 TfDeviceId tf_device_id, int64_t memory_limit, in CreateGPUDevice()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_factory.cc81 int64_t* memory_limit) { in SingleVirtualDeviceMemoryLimit()
227 TfDeviceId tf_device_id, int64_t memory_limit, in CreatePluggableDevice()
H A Dpluggable_device.cc142 Bytes memory_limit, const DeviceLocality& locality, TfDeviceId tf_device_id, in PluggableDevice()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Ddata_flow_ops.py1618 memory_limit=0): argument
1676 def memory_limit(self): member in BaseStagingArea
1860 memory_limit=0): argument
2113 memory_limit=0): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstage_op.cc37 explicit Buffer(std::size_t capacity, std::size_t memory_limit) in Buffer()
186 int64_t memory_limit; in GetBuffer() local
H A Dmap_stage_op.cc347 std::size_t memory_limit) in StagingMap()
504 int64_t memory_limit; in GetStagingMap() local
H A Dconv_ops_gpu.h50 DnnScratchAllocator(int64_t memory_limit, OpKernelContext* context) in DnnScratchAllocator()
H A Dfft_ops.cc376 CufftScratchAllocator(int64_t memory_limit, OpKernelContext* context) in CufftScratchAllocator()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddevice.cc44 const string& name, DeviceType device, Bytes memory_limit, in BuildDeviceAttributes()
H A Ddevice_attributes.proto41 int64 memory_limit = 4; field
H A Ddevice.h170 const std::string& name, DeviceType device, Bytes memory_limit, in BuildDeviceAttributes()
/aosp_15_r20/external/libchrome/base/
H A Dsys_info.cc39 size_t memory_limit = kLowMemoryDeviceThresholdMB * 1024 * 1024; in AmountOfAvailablePhysicalMemory() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dstat_summarizer_options.h37 int memory_limit; member
H A Dtest_log.proto133 int64 memory_limit = 3; // Memory capacity in bytes. field
/aosp_15_r20/external/iproute2/tc/
H A Dq_fq_codel.c165 unsigned int memory_limit; in fq_codel_print_opt() local
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/
H A Dprofile.proto43 uint64 memory_limit = 3; field
/aosp_15_r20/external/perfetto/protos/third_party/pprof/
H A Dprofile.proto156 uint64 memory_limit = 3; field
/aosp_15_r20/external/cronet/base/system/
H A Dsys_info.cc70 uint64_t memory_limit = kLowMemoryDeviceThresholdMB * 1024 * 1024; in AmountOfAvailablePhysicalMemory() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_platform.cc168 Status TpuPlatform::TpuMemoryLimit(int64_t* memory_limit) { in TpuMemoryLimit()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dthreadpool_device.cc57 const string& name, Bytes memory_limit, in ThreadPoolDevice()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dredzone_allocator.cc53 int64_t memory_limit, int64_t redzone_size, in RedzoneAllocator()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_tpu_device.cc442 int64_t memory_limit; in CreateDevices() local
/aosp_15_r20/external/perfetto/src/trace_processor/util/
H A Dprofile_builder.h229 const uint64_t memory_limit; member
/aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/
H A Dbenchmark_model.cc394 int memory_limit = 10; in Main() local

12