| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | compiler.h | 107 se::DeviceMemoryAllocator* device_allocator() const { in device_allocator() function
110 void set_device_allocator(se::DeviceMemoryAllocator* device_allocator) { in set_device_allocator()
230 se::DeviceMemoryAllocator* device_allocator = nullptr; member
248 se::DeviceMemoryAllocator* device_allocator) { in RunHloPasses()
275 se::DeviceMemoryAllocator* device_allocator) { in RunBackend()
300 se::DeviceMemoryAllocator* device_allocator) { in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | amdgpu_compiler.cc | 78 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()
110 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
|
| H A D | nvptx_compiler.cc | 73 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()
121 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
|
| H A D | gpu_compiler.cc | 338 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloModule()
724 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | llvm_compiler_test.cc | 48 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()
54 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_platform_info.h | 35 std::shared_ptr<se::DeviceMemoryAllocator> device_allocator) in XlaPlatformInfo()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_process_state.h | 108 Allocator* device_allocator; member
|
| H A D | pluggable_device_process_state.cc | 119 Allocator* device_allocator = nullptr; in GetPluggableDeviceAllocator() local
|
| H A D | pluggable_device_factory.cc | 249 Allocator* device_allocator = process_state->GetPluggableDeviceAllocator( in CreatePluggableDevice() local
|
| H A D | pluggable_device.cc | 143 const std::string& physical_device_desc, Allocator* device_allocator, in PluggableDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | executable_build_options.cc | 31 se::DeviceMemoryAllocator* ExecutableBuildOptions::device_allocator() const { in device_allocator() function in xla::ExecutableBuildOptions
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebraLib.cpp | 739 auto& device_allocator = *at::cuda::getCUDADeviceAllocator(); in apply_cholesky_cusolver_potrf_looped() local
1014 auto& device_allocator = *at::cuda::getCUDADeviceAllocator(); in apply_geqrf() local
1264 auto& device_allocator = *at::cuda::getCUDADeviceAllocator(); in apply_syevd() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_compiler.h | 194 std::shared_ptr<se::DeviceMemoryAllocator> device_allocator; member
|
| /aosp_15_r20/external/pytorch/c10/cuda/ |
| H A D | CUDAMallocAsyncAllocator.cpp | 895 CudaMallocAsyncAllocator device_allocator; variable
|
| H A D | CUDACachingAllocator.cpp | 3183 std::vector<std::unique_ptr<DeviceCachingAllocator>> device_allocator; member in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_graph.cc | 697 std::pair<int, Allocator*> device_allocator = in CreateStaticEngine() local
|