| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/ |
| H A D | BUILD | 308 "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",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | BUILD | 27 "device_memory_allocator.h",
631 "device_memory_allocator.h",
679 name = "device_memory_allocator",
680 hdrs = ["device_memory_allocator.h"],
726 ":device_memory_allocator",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | BUILD | 134 "//tensorflow/stream_executor:device_memory_allocator",
178 "//tensorflow/stream_executor:device_memory_allocator",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | BUILD | 57 "//tensorflow/compiler/xla/stream_executor:device_memory_allocator",
224 "//tensorflow/compiler/xla/stream_executor:device_memory_allocator",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/ |
| H A D | BUILD | 105 "//tensorflow/stream_executor:device_memory_allocator",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | BUILD | 93 "//tensorflow/stream_executor:device_memory_allocator",
113 "//tensorflow/stream_executor:device_memory_allocator",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
| H A D | BUILD | 291 "//tensorflow/compiler/xla/stream_executor:device_memory_allocator",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_op.cc | 85 TRTBaseAllocator* device_memory_allocator, in AllocateDeviceMemory() argument
88 device_memory_allocator_ = device_memory_allocator; in AllocateDeviceMemory()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | BUILD | 1076 "//tensorflow/stream_executor:device_memory_allocator",
1108 "//tensorflow/stream_executor:device_memory_allocator",
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | BUILD | 360 "//tensorflow/stream_executor:device_memory_allocator",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | BUILD | 417 "//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",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | BUILD | 1121 "//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",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | BUILD | 314 "//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",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | BUILD | 737 "//tensorflow/compiler/xla/stream_executor:device_memory_allocator",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | BUILD | 439 "//tensorflow/stream_executor:device_memory_allocator",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | BUILD | 483 "//tensorflow/stream_executor:device_memory_allocator",
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr.o.d | 1339 external/tensorflow/tensorflow/compiler/xla/stream_executor/device_memory_allocator.h \
|