| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compiler.h | 78 DeviceMemoryAllocator* device_allocator() const { return device_allocator_; } in device_allocator() function
79 void set_device_allocator(DeviceMemoryAllocator* device_allocator) { in set_device_allocator() argument
80 device_allocator_ = device_allocator; in set_device_allocator()
150 DeviceMemoryAllocator* device_allocator) = 0;
157 DeviceMemoryAllocator* device_allocator) = 0;
171 DeviceMemoryAllocator* device_allocator) = 0;
179 DeviceMemoryAllocator* device_allocator) = 0;
192 DeviceMemoryAllocator* device_allocator) = 0;
|
| D | llvm_compiler.cc | 27 DeviceMemoryAllocator* device_allocator) { in RunHloPassesOnModuleGroup() argument
36 DeviceMemoryAllocator* device_allocator) { in RunBackendOnModuleGroup() argument
44 DeviceMemoryAllocator* device_allocator) { in Compile() argument
68 device_allocator)); in Compile()
71 device_allocator)); in Compile()
|
| D | llvm_compiler.h | 75 DeviceMemoryAllocator* device_allocator) override;
80 DeviceMemoryAllocator* device_allocator) override;
85 DeviceMemoryAllocator* device_allocator) override;
|
| D | service.h | 237 DeviceMemoryAllocator* device_allocator = nullptr);
245 DeviceMemoryAllocator* device_allocator);
|
| D | service.cc | 350 DeviceMemoryAllocator* device_allocator) { in BuildExecutables() argument
380 std::move(executors), device_allocator)); in BuildExecutables()
786 se::StreamExecutor* executor, DeviceMemoryAllocator* device_allocator) { in BuildExecutable() argument
797 device_allocator)); in BuildExecutable()
801 std::move(module), executor, device_allocator)); in BuildExecutable()
|
| D | local_service.cc | 180 build_options.device_allocator()); in CompileExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.h | 48 DeviceMemoryAllocator* device_allocator) override;
52 DeviceMemoryAllocator* device_allocator) override;
56 DeviceMemoryAllocator* device_allocator) override;
60 DeviceMemoryAllocator* device_allocator) override;
65 DeviceMemoryAllocator* device_allocator) override;
|
| D | compiler.cc | 108 DeviceMemoryAllocator* device_allocator) { in RunHloPassesOnModuleGroup() argument
140 DeviceMemoryAllocator* device_allocator) { in RunBackendOnModuleGroup() argument
148 DeviceMemoryAllocator* device_allocator) { in Compile() argument
163 device_allocator)); in Compile()
166 RunBackend(std::move(module), stream_exec[0][0], device_allocator)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.h | 55 DeviceMemoryAllocator* device_allocator) override;
59 DeviceMemoryAllocator* device_allocator) override;
|
| D | nvptx_compiler.cc | 180 DeviceMemoryAllocator* device_allocator, in OptimizeHloModule() argument
273 pipeline.AddPass<CusolverRewriter>(stream_exec, device_allocator); in OptimizeHloModule()
344 pipeline.AddPass<CudnnConvAlgorithmPicker>(stream_exec, device_allocator, in OptimizeHloModule()
620 DeviceMemoryAllocator* device_allocator) { in RunHloPasses() argument
626 OptimizeHloModule(module.get(), stream_exec, device_allocator, this)); in RunHloPasses()
635 DeviceMemoryAllocator* device_allocator) { in RunBackend() argument
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.h | 43 xla::DeviceMemoryAllocator* device_allocator) in XlaPlatformInfo() argument
48 device_allocator_(device_allocator) { in XlaPlatformInfo()
|
| D | xla_ops.cc | 65 xla::DeviceMemoryAllocator* device_allocator = nullptr; in PlatformInfoFromContext() local
87 device_allocator = in PlatformInfoFromContext()
91 if (!device_allocator) { in PlatformInfoFromContext()
101 std::move(xla_allocator), device_allocator); in PlatformInfoFromContext()
293 options.device_allocator = platform_info.allocator(); in CompileToLocalExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.h | 138 DeviceMemoryAllocator* device_allocator) override;
142 DeviceMemoryAllocator* device_allocator) override;
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.h | 64 DeviceMemoryAllocator* device_allocator() const;
|
| D | executable_build_options.cc | 30 DeviceMemoryAllocator* ExecutableBuildOptions::device_allocator() const { in device_allocator() function in xla::ExecutableBuildOptions
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 327 xla::DeviceMemoryAllocator* device_allocator = nullptr; member
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 156 build_options.set_device_allocator(options.device_allocator); in BuildExecutable()
|