| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compiler.h | 86 se::DeviceMemoryAllocator* device_allocator() const { in device_allocator() function
89 void set_device_allocator(se::DeviceMemoryAllocator* device_allocator) { in set_device_allocator() argument
90 device_allocator_ = device_allocator; in set_device_allocator()
169 se::DeviceMemoryAllocator* device_allocator = nullptr; member
187 se::DeviceMemoryAllocator* device_allocator) { in RunHloPasses() argument
189 CompileOptions{device_allocator}); in RunHloPasses()
216 se::DeviceMemoryAllocator* device_allocator) { in RunBackend() argument
218 CompileOptions{device_allocator}); in RunBackend()
234 se::DeviceMemoryAllocator* device_allocator) { in Compile() argument
236 CompileOptions{device_allocator}); in Compile()
|
| D | llvm_compiler.cc | 47 options.device_allocator)); in Compile()
50 options.device_allocator)); in Compile()
|
| D | local_service.cc | 167 {build_options.device_allocator(), in CompileExecutables()
184 Compiler::CompileOptions{build_options.device_allocator(), in CompileExecutables()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_platform_info.h | 35 se::DeviceMemoryAllocator* device_allocator) in XlaPlatformInfo() argument
39 device_allocator_(device_allocator) {} in XlaPlatformInfo()
|
| D | xla_platform_info.cc | 145 options.device_allocator = in GenerateCompilerOptions()
|
| D | xla_compilation_cache.cc | 173 build_options.set_device_allocator(options.device_allocator); in BuildExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.h | 68 se::DeviceMemoryAllocator* device_allocator);
72 se::DeviceMemoryAllocator* device_allocator) = 0;
76 se::DeviceMemoryAllocator* device_allocator);
|
| D | nvptx_compiler.cc | 109 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization() argument
162 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment() argument
173 hlo_module, stream_exec, device_allocator)); in OptimizeHloPostLayoutAssignment()
178 post_pipeline.AddPass<GemmAlgorithmPicker>(stream_exec, device_allocator); in OptimizeHloPostLayoutAssignment()
|
| D | nvptx_compiler.h | 43 se::DeviceMemoryAllocator* device_allocator) override;
47 se::DeviceMemoryAllocator* device_allocator) override;
|
| D | amdgpu_compiler.h | 38 se::DeviceMemoryAllocator* device_allocator) override;
|
| D | gpu_compiler.cc | 151 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloModule() argument
286 hlo_module, stream_exec, device_allocator)); in OptimizeHloModule()
307 device_allocator)); in OptimizeHloModule()
404 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment() argument
478 pipeline.AddPass<GpuConvAlgorithmPicker>(stream_exec, device_allocator); in OptimizeHloPostLayoutAssignment()
498 OptimizeHloModule(module.get(), stream_exec, options.device_allocator)); in RunHloPasses()
|
| D | amdgpu_compiler.cc | 75 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization() argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 46 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization() argument
52 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment() argument
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 229 auto allocator = ApiConverter::ToC(options.device_allocator); in RunHloPasses()
258 auto allocator = ApiConverter::ToC(options.device_allocator); in RunBackend()
310 ApiConverter::ToC(options.device_allocator); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.h | 72 se::DeviceMemoryAllocator* device_allocator() const;
|
| D | executable_build_options.cc | 31 se::DeviceMemoryAllocator* ExecutableBuildOptions::device_allocator() const { in device_allocator() function in xla::ExecutableBuildOptions
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 434 std::pair<int, Allocator*> device_allocator = in CreateTRTNode() local
438 if (device_allocator.first >= 0) { in CreateTRTNode()
439 cuda_device_id = device_allocator.first; in CreateTRTNode()
440 trt_allocator.reset(new TRTDeviceAllocator(device_allocator.second)); in CreateTRTNode()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 195 se::DeviceMemoryAllocator* device_allocator = nullptr; member
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 2123 if (!build_options.device_allocator()) { in Compile()
|