Home
last modified time | relevance | path

Searched refs:device_memory_allocator (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
DBUILD308 "device_memory_allocator.h",
347 "device_memory_allocator.h",
390 name = "device_memory_allocator",
391 hdrs = ["device_memory_allocator.h"],
393 "//tensorflow/compiler/xla/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
DBUILD27 "device_memory_allocator.h",
631 "device_memory_allocator.h",
679 name = "device_memory_allocator",
680 hdrs = ["device_memory_allocator.h"],
726 ":device_memory_allocator",
/external/tensorflow/tensorflow/compiler/xla/client/
DBUILD134 "//tensorflow/stream_executor:device_memory_allocator",
178 "//tensorflow/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
DBUILD57 "//tensorflow/compiler/xla/stream_executor:device_memory_allocator",
224 "//tensorflow/compiler/xla/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/xrt/
DBUILD105 "//tensorflow/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
DBUILD291 "//tensorflow/compiler/xla/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/xla/pjrt/
DBUILD93 "//tensorflow/stream_executor:device_memory_allocator",
113 "//tensorflow/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/core/tpu/
DBUILD360 "//tensorflow/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc85 TRTBaseAllocator* device_memory_allocator, in AllocateDeviceMemory() argument
88 device_memory_allocator_ = device_memory_allocator; in AllocateDeviceMemory()
/external/tensorflow/tensorflow/core/tpu/kernels/
DBUILD1076 "//tensorflow/stream_executor:device_memory_allocator",
1108 "//tensorflow/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
DBUILD417 "//tensorflow/stream_executor:device_memory_allocator",
680 "//tensorflow/stream_executor:device_memory_allocator",
932 "//tensorflow/stream_executor:device_memory_allocator",
1009 "//tensorflow/stream_executor:device_memory_allocator",
1118 "//tensorflow/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/xla/service/
DBUILD1121 "//tensorflow/stream_executor:device_memory_allocator",
1174 "//tensorflow/stream_executor:device_memory_allocator",
1210 "//tensorflow/stream_executor:device_memory_allocator",
1310 "//tensorflow/stream_executor:device_memory_allocator",
1333 "//tensorflow/stream_executor:device_memory_allocator",
1372 "//tensorflow/stream_executor:device_memory_allocator",
1451 "//tensorflow/stream_executor:device_memory_allocator",
5157 "//tensorflow/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/xla/tests/
DBUILD314 "//tensorflow/stream_executor:device_memory_allocator",
1517 "//tensorflow/stream_executor:device_memory_allocator",
2564 "//tensorflow/stream_executor:device_memory_allocator",
2688 "//tensorflow/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
DBUILD737 "//tensorflow/compiler/xla/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/jit/
DBUILD439 "//tensorflow/stream_executor:device_memory_allocator",
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
DBUILD483 "//tensorflow/stream_executor:device_memory_allocator",