Searched defs:gpu_allocator_ (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | ||
| H A D | light_outside_compilation.cc | 361 Allocator* gpu_allocator_; member in tensorflow::__anonbb96d3be0311::TfCallbackDevice |
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/ | ||
| H A D | device_event_mgr_test.cc | 169 Allocator* gpu_allocator_; member in tensorflow::GPUDeviceTestHelper |