| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compiler.h | 78 DeviceMemoryAllocator* device_allocator() const { return device_allocator_; } in device_allocator()
79 void set_device_allocator(DeviceMemoryAllocator* device_allocator) { in set_device_allocator()
101 DeviceMemoryAllocator* device_allocator_ = nullptr;
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 | device_memory_allocator.h | 33 class DeviceMemoryAllocator {
37 explicit DeviceMemoryAllocator(const se::Platform* platform) in DeviceMemoryAllocator() function
39 virtual ~DeviceMemoryAllocator() {} in ~DeviceMemoryAllocator()
79 class StreamExecutorMemoryAllocator : public DeviceMemoryAllocator {
89 using DeviceMemoryAllocator::Allocate;
|
| D | owning_device_memory.h | 27 class DeviceMemoryAllocator; variable
47 DeviceMemoryAllocator* allocator) in OwningDeviceMemory()
88 DeviceMemoryAllocator* allocator() const { return allocator_; } in allocator()
126 DeviceMemoryAllocator* allocator_; // Null if this object is inactive.
|
| D | shaped_buffer.h | 138 DeviceMemoryAllocator* allocator,
144 DeviceMemoryAllocator* allocator);
157 DeviceMemoryAllocator* memory_allocator() const { return allocator_; } in memory_allocator()
187 DeviceMemoryAllocator* allocator_;
|
| D | llvm_compiler.h | 75 DeviceMemoryAllocator* device_allocator) override;
80 DeviceMemoryAllocator* device_allocator) override;
85 DeviceMemoryAllocator* device_allocator) override;
|
| D | llvm_compiler.cc | 27 DeviceMemoryAllocator* device_allocator) { in RunHloPassesOnModuleGroup()
36 DeviceMemoryAllocator* device_allocator) { in RunBackendOnModuleGroup()
44 DeviceMemoryAllocator* device_allocator) { in Compile()
|
| D | shaped_buffer_test.cc | 46 class TestAllocator : public DeviceMemoryAllocator {
49 : DeviceMemoryAllocator(PlatformUtil::GetDefaultPlatform().ValueOrDie()) { in TestAllocator()
59 using DeviceMemoryAllocator::Allocate;
|
| D | shaped_buffer.cc | 108 DeviceMemoryAllocator* allocator, in ScopedShapedBuffer()
115 DeviceMemoryAllocator* allocator) in ScopedShapedBuffer()
|
| /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 | 99 DeviceMemoryAllocator* /*device_allocator*/) { in RunHloPasses() argument
108 DeviceMemoryAllocator* device_allocator) { in RunHloPassesOnModuleGroup()
114 DeviceMemoryAllocator* /*device_allocator*/) { in RunBackend() argument
140 DeviceMemoryAllocator* device_allocator) { in RunBackendOnModuleGroup()
148 DeviceMemoryAllocator* device_allocator) { in Compile()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.h | 49 xla::DeviceMemoryAllocator* allocator);
72 xla::DeviceMemoryAllocator* allocator_;
196 XRTTupleAllocation(int device_ordinal, xla::DeviceMemoryAllocator* allocator,
203 xla::DeviceMemoryAllocator* allocator,
213 xla::DeviceMemoryAllocator* allocator, xla::Shape* host_shape,
218 xla::DeviceMemoryAllocator* allocator_;
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.h | 34 class DeviceMemoryAllocator; variable
42 ExecutableRunOptions& set_allocator(DeviceMemoryAllocator* allocator);
43 DeviceMemoryAllocator* allocator() const;
84 DeviceMemoryAllocator* allocator_ = nullptr;
|
| D | executable_run_options.cc | 29 DeviceMemoryAllocator* allocator) { in set_allocator()
34 DeviceMemoryAllocator* ExecutableRunOptions::allocator() const { in allocator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_allocations.h | 53 DeviceMemoryAllocator* memory_allocator);
65 DeviceMemoryAllocator* memory_allocator() const { return memory_allocator_; } in memory_allocator()
87 DeviceMemoryAllocator* memory_allocator, in BufferAllocations()
107 DeviceMemoryAllocator* memory_allocator_;
|
| D | buffer_comparator.h | 38 DeviceMemoryAllocator* allocator, se::Stream* stream);
52 F16BufferComparator(se::Stream* stream, DeviceMemoryAllocator* allocator, in F16BufferComparator()
63 DeviceMemoryAllocator* allocator_;
|
| D | cusolver_rewriter.h | 33 DeviceMemoryAllocator* allocator);
42 DeviceMemoryAllocator* allocator_; // may be null
|
| D | cudnn_conv_algorithm_picker.h | 41 DeviceMemoryAllocator* allocator, Compiler* compiler) in CudnnConvAlgorithmPicker()
57 DeviceMemoryAllocator* allocator_; // may be null
|
| D | scratch_allocator.h | 32 ScratchAllocator(int device_ordinal, DeviceMemoryAllocator* memory_allocator) in ScratchAllocator()
53 DeviceMemoryAllocator* memory_allocator_;
|
| D | fft_thunk.h | 41 DeviceMemoryAllocator* memory_allocator);
52 DeviceMemoryAllocator* memory_allocator_;
|
| D | nvptx_compiler.h | 55 DeviceMemoryAllocator* device_allocator) override;
59 DeviceMemoryAllocator* device_allocator) override;
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.h | 63 DeviceMemoryAllocator* allocator);
64 DeviceMemoryAllocator* device_allocator() const;
80 DeviceMemoryAllocator* device_allocator_ = nullptr;
|
| D | executable_build_options.cc | 25 DeviceMemoryAllocator* allocator) { in set_device_allocator()
30 DeviceMemoryAllocator* ExecutableBuildOptions::device_allocator() const { in device_allocator()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.h | 43 xla::DeviceMemoryAllocator* device_allocator) in XlaPlatformInfo()
58 xla::DeviceMemoryAllocator* allocator() const { in allocator()
89 xla::DeviceMemoryAllocator* device_allocator_;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.h | 111 class XlaAllocator : public xla::DeviceMemoryAllocator {
145 xla::DeviceMemoryAllocator* xla_allocator,
189 xla::DeviceMemoryAllocator* xla_allocator_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.h | 138 DeviceMemoryAllocator* device_allocator) override;
142 DeviceMemoryAllocator* device_allocator) override;
|