| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.cc | 26 DeviceBase::~DeviceBase() { gtl::STLDeleteElements(&eigen_cpu_devices_); } in ~DeviceBase()
28 const DeviceAttributes& DeviceBase::attributes() const { in attributes()
32 const string& DeviceBase::name() const { in name()
36 void DeviceBase::set_eigen_cpu_device(Eigen::ThreadPoolDevice* d) { in set_eigen_cpu_device()
48 const Eigen::ThreadPoolDevice* DeviceBase::eigen_cpu_device() { in eigen_cpu_device()
|
| D | device_base.h | 114 class DeviceBase {
116 explicit DeviceBase(Env* env) : env_(env) {} in DeviceBase() function
117 virtual ~DeviceBase();
220 virtual DeviceBase* UnderlyingDevice() { return this; } in UnderlyingDevice()
221 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 | 200 static int DeviceNumaNode(const DeviceBase* device);
266 OpKernelConstruction(DeviceType device_type, DeviceBase* device,
374 DeviceBase* device() const { return device_; } in device()
378 DeviceBase* const device_;
587 DeviceBase* device = nullptr;
1155 DeviceBase* device() const { return params_->device; } in device()
1315 DeviceBase* device,
1319 Status CreateOpKernel(DeviceType device_type, DeviceBase* device,
|
| D | device_base_test.cc | 30 DeviceBase dbase(Env::Default()); in TEST()
|
| /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;
|
| D | tensor_coding_test.cc | 33 class DummyDevice : public DeviceBase {
35 explicit DummyDevice(Env* env) : DeviceBase(env) { in DummyDevice()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.h | 35 class DeviceSimple : public DeviceBase {
49 DeviceBase::CpuWorkerThreads eigen_worker_threads_;
56 DeviceBase* cpu_device, ResourceMgr* resource_mgr,
|
| D | meta_optimizer.h | 37 MetaOptimizer(DeviceBase* cpu_device, const ConfigProto& cfg);
79 DeviceBase* const cpu_device_; // may be NULL
110 DeviceBase* cpu_device, Cluster* cluster,
|
| D | evaluation_utils.cc | 53 DeviceSimple::DeviceSimple() : DeviceBase(Env::Default()) { in DeviceSimple()
83 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode()
86 std::unique_ptr<DeviceBase> device; in EvaluateNode()
|
| D | constant_folding.h | 45 explicit ConstantFolding(DeviceBase* cpu_device);
46 ConstantFolding(RewriterConfig::Toggle opt_level, DeviceBase* cpu_device);
244 DeviceBase* cpu_device_;
245 std::unique_ptr<DeviceBase> owned_device_;
|
| D | loop_optimizer.h | 35 DeviceBase* cpu_device);
68 DeviceBase* cpu_device_;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 65 const DeviceBase::GpuDeviceInfo** dev_info, in PrepareCopy()
120 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in SetProtoFromGPU()
193 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in DeviceToDeviceCopy()
260 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyGPUTensorToCPU()
305 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyCPUTensorToGPU()
428 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyGPUTensorToSameGPU()
|
| D | gpu_util_platform_specific.cc | 50 const DeviceBase::GpuDeviceInfo* gpu_info = in ThenExecute()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | quantile_ops.cc | 203 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
264 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
312 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
365 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
424 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
501 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_cpu.cc | 52 DeviceBase* d, in ConcatCPU()
67 DeviceBase*, \
|
| D | scoped_allocator_ops.cc | 82 DeviceBase* device_;
163 DeviceBase* device_;
225 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 | collective_nccl_reducer_test.cc | 43 std::unique_ptr<OpKernel> GetKernel(const NodeDef& node, DeviceBase* device) { in GetKernel()
52 std::unique_ptr<OpKernel> GetAdd(DeviceBase* device) { in GetAdd()
62 std::unique_ptr<OpKernel> GetDiv(DeviceBase* device) { in GetDiv()
196 DeviceBase* device) { in GetCollectiveReduce()
|
| D | sdca_internal.cc | 371 const DeviceBase::CpuWorkerThreads& worker_threads = in Initialize()
387 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateSparseFeatureRepresentation()
467 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateDenseFeatureRepresentation()
499 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in ComputeSquaredNormPerExample()
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 25 class DummyDevice : public DeviceBase {
27 DummyDevice(Env* env, bool save) : DeviceBase(env), save_(save) {} in DummyDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 43 const DeviceBase* UnderlyingDevice() const override { in UnderlyingDevice()
46 DeviceBase* UnderlyingDevice() override { in UnderlyingDevice()
|
| D | collective_rma_local.cc | 104 const DeviceBase::GpuDeviceInfo* dev_info = in MemCpyAsync()
110 const DeviceBase::GpuDeviceInfo* dev_info = in MemCpyAsync()
|
| D | device.cc | 27 : DeviceBase(env), device_attributes_(device_attributes) { in Device()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 128 class DummyDevice : public DeviceBase {
130 DummyDevice(Env* env, bool save) : DeviceBase(env), save_(save) {} in DummyDevice()
|