| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.cc | 30 DeviceBase::~DeviceBase() { in ~DeviceBase()
66 const DeviceAttributes& DeviceBase::attributes() const { in attributes()
71 const string& DeviceBase::name() const { in name()
76 void DeviceBase::set_eigen_cpu_device(Eigen::ThreadPoolDevice* d) { in set_eigen_cpu_device()
88 const Eigen::ThreadPoolDevice* DeviceBase::eigen_cpu_device() { in eigen_cpu_device()
|
| D | device_base.h | 122 class DeviceBase {
124 explicit DeviceBase(Env* env) : env_(env) {} in DeviceBase() function
125 virtual ~DeviceBase();
214 virtual DeviceBase* UnderlyingDevice() { return this; } in UnderlyingDevice()
215 virtual const DeviceBase* UnderlyingDevice() const { return this; } in UnderlyingDevice()
|
| D | resource_op_kernel_test.cc | 37 class StubDevice : public DeviceBase {
39 StubDevice() : DeviceBase(nullptr) {} in StubDevice()
|
| D | op_kernel.h | 259 OpKernelConstruction(DeviceType device_type, DeviceBase* device,
373 DeviceBase* device() const { return device_; } in device()
377 DeviceBase* const device_;
605 DeviceBase* device = nullptr;
1192 DeviceBase* device() const { return params_->device; } in device()
1376 DeviceBase* device,
1382 DeviceType device_type, DeviceBase* device, Allocator* allocator,
1386 Status CreateOpKernel(DeviceType device_type, DeviceBase* device,
1391 Status CreateOpKernel(DeviceType device_type, DeviceBase* device,
|
| D | device_base_test.cc | 29 DeviceBase dbase(Env::Default()); in TEST()
|
| D | device.cc | 27 : DeviceBase(env), device_attributes_(device_attributes) { in Device()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | tensor_coding.h | 29 class DeviceBase; variable
47 void InitAlloc(DeviceBase* d, const AllocatorAttributes& aa);
92 DeviceBase* device() const { return device_; } in device()
101 DeviceBase* device_ = nullptr;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_platform_info.h | 83 Status BuildXlaCompilationCache(DeviceBase* dev,
88 XlaPlatformInfo XlaPlatformInfoFromDevice(DeviceBase* device);
99 DeviceBase* device, se::Stream* stream,
106 const FunctionLibraryRuntime& function_library, DeviceBase* device,
|
| D | xla_platform_info.cc | 22 Status BuildXlaCompilationCache(DeviceBase* device, in BuildXlaCompilationCache()
78 XlaPlatformInfo XlaPlatformInfoFromDevice(DeviceBase* device_base) { in XlaPlatformInfoFromDevice()
113 DeviceBase* device, se::Stream* stream, in GetAllocator()
132 const FunctionLibraryRuntime& function_library, DeviceBase* device, in GenerateCompilerOptions()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.h | 35 class DeviceSimple : public DeviceBase {
49 DeviceBase::CpuWorkerThreads eigen_worker_threads_;
55 DeviceBase* cpu_device, ResourceMgr* resource_mgr,
|
| D | evaluation_utils.cc | 34 DeviceSimple::DeviceSimple() : DeviceBase(Env::Default()) { in DeviceSimple()
62 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode()
65 std::unique_ptr<DeviceBase> device; in EvaluateNode()
|
| D | meta_optimizer.h | 37 MetaOptimizer(DeviceBase* cpu_device, const ConfigProto& cfg);
92 DeviceBase* const cpu_device_; // may be NULL
124 DeviceBase* cpu_device, Cluster* cluster,
|
| D | loop_optimizer.h | 37 DeviceBase* cpu_device);
72 DeviceBase* cpu_device_;
|
| D | constant_folding.h | 48 explicit ConstantFolding(DeviceBase* cpu_device,
51 ConstantFolding(RewriterConfig::Toggle opt_level, DeviceBase* cpu_device,
330 DeviceBase* cpu_device_;
331 std::unique_ptr<DeviceBase> owned_device_;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 39 const DeviceBase* UnderlyingDevice() const override { in UnderlyingDevice()
42 DeviceBase* UnderlyingDevice() override { in UnderlyingDevice()
167 DeviceBase::CpuWorkerThreads eigen_worker_threads_;
|
| D | collective_rma_local.cc | 139 const DeviceBase::GpuDeviceInfo* dev_info = in MemCpyAsync()
145 const DeviceBase::GpuDeviceInfo* dev_info = in MemCpyAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 63 const DeviceBase::GpuDeviceInfo** dev_info, in PrepareCopy()
118 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in SetProtoFromGPU()
192 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in DeviceToDeviceCopy()
259 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyGPUTensorToCPU()
304 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyCPUTensorToGPU()
428 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyGPUTensorToSameGPU()
|
| D | gpu_util_platform_specific.cc | 52 const DeviceBase::GpuDeviceInfo* gpu_info = in ThenExecute()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | quantile_ops.cc | 204 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
263 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
330 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
377 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
429 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
488 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
565 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_cpu.cc | 100 DeviceBase* d, in ConcatCPU()
110 DeviceBase*, \
|
| D | scoped_allocator_ops.cc | 83 DeviceBase* device_;
164 DeviceBase* device_;
226 DeviceBase* device_;
|
| D | sdca_internal.h | 352 const DeviceBase::CpuWorkerThreads& worker_threads, int num_examples,
363 const DeviceBase::CpuWorkerThreads& worker_threads, int num_examples,
371 const DeviceBase::CpuWorkerThreads& worker_threads, int num_examples,
|
| D | concat_lib.h | 46 DeviceBase* d,
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | summary_op_test.cc | 39 class DummyDevice : public DeviceBase {
41 explicit DummyDevice(Env* env) : DeviceBase(env) {} in DummyDevice()
|
| D | bitcast_op_test.cc | 28 class DummyDevice : public DeviceBase {
30 explicit DummyDevice(Env* env) : DeviceBase(env) {} in DummyDevice()
|