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