Home
last modified time | relevance | path

Searched refs:device_name_ (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/core/tfrt/eager/
Dop_cache.h84 device_name_(device_name), in CacheKey()
90 device_name_(other.device_name_), in CacheKey()
98 device_name_ = device_name_concrete_.data(); in CacheKey()
114 device_name_concrete_ = device_name_.str(); in MakeConcrete()
136 lhs.device_name_ == rhs.device_name_);
141 string_view DeviceName() { return device_name_; } in DeviceName()
149 string_view op_name_, device_name_; variable
Dc_api_tfrt.cc1445 op_name_, op_handler, device_name_, dtypes, this); in Initialize()
1449 " on device: ", device_name_, expected_op.takeError())); in Initialize()
1454 device_name_ = op_->DeviceName().str(); in Initialize()
1501 device_name_.empty() ? context_->GetEagerContext()->HostCPUName() in Initialize()
1502 : device_name_; in Initialize()
1541 op_name_, device_name_, dev_set, context_->GetEagerContext(), corert, in Initialize()
1559 if (op_ && name != device_name_) { in SetDeviceName()
1564 device_name_ = name ? name : ""; in SetDeviceName()
Dc_api_tfrt.h492 const std::string& DeviceName() const override { return device_name_; } in DeviceName()
599 std::string device_name_; variable
/external/nos/host/generic/libnos/
DNuggetClient.cpp25 : device_name_(name), open_(false) { in NuggetClient()
29 : device_name_(name ? name : ""), open_(false) { in NuggetClient()
40 device_name_.empty() ? nullptr : device_name_.c_str(), &device_) == 0; in Open()
113 return device_name_; in DeviceName()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dunion_find.cc129 : batch_size_(batch_size), device_name_(device_name) {} in ClusterProperty()
139 MergeIfCompatible(device_name_, other.device_name_); in Merge()
146 device_name_ = std::move(merged_device_name.value()); in Merge()
Dunion_find.h137 const DeviceNameUtils::ParsedName& DeviceName() const { return device_name_; } in DeviceName()
143 DeviceNameUtils::ParsedName device_name_; variable
/external/stressapptest/src/
Dworker.cc2610 device_name_ = device_name; in SetDevice()
2725 device_name_); in SetParameters()
2733 int fd = open(device_name_.c_str(), flags | O_DIRECT, 0); in OpenDevice()
2735 fd = open(device_name_.c_str(), flags, 0); // Try without O_DIRECT in OpenDevice()
2740 device_name_.c_str(), thread_num_); in OpenDevice()
2754 device_name_.c_str(), thread_num_); in GetDiskSize()
2765 device_name_.c_str(), thread_num_); in GetDiskSize()
2771 os_->ErrorReport(device_name_.c_str(), "device-size-zero", 1); in GetDiskSize()
2784 "device (thread %d).\n", device_name_.c_str(), in GetDiskSize()
2790 device_sectors_, device_name_.c_str(), thread_num_); in GetDiskSize()
[all …]
Ddisk_blocks.h150 const string& device_name() const { return device_name_; } in device_name()
164 string device_name_; // Device name variable
Ddisk_blocks.cc49 device_name_(""), device_sectors_(0), in DiskBlockTable()
191 device_name_ = device_name; in SetParameters()
201 "for sector data for disk %s.\n", device_name_.c_str()); in GetUnusedBlock()
/external/tensorflow/tensorflow/core/common_runtime/
Dscoped_allocator_mgr.h77 : device_name_(device_name) {} in ScopedAllocatorMgr()
101 const std::string& device_name() const { return device_name_; } in device_name()
104 std::string device_name_;
Dfunction.cc398 const string device_name_; member in tensorflow::FunctionLibraryRuntimeImpl
482 device_name_(device_ == nullptr in FunctionLibraryRuntimeImpl()
576 LocalHandle local_handle = parent_->GetHandleOnDevice(device_name_, h); in GetFunctionBody()
579 << " on device: " << device_name_; in GetFunctionBody()
594 LocalHandle local_handle = parent_->GetHandleOnDevice(device_name_, h); in GetRetTypes()
762 options_copy.target = device_name_; in Instantiate()
770 parent_->GetHandleOnDevice(device_name_, *handle); in Instantiate()
816 local_handle = parent_->GetHandleOnDevice(device_name_, *handle); in Instantiate()
819 *handle = parent_->AddHandle(key, device_name_, next_handle_); in Instantiate()
846 LocalHandle h = parent_->GetHandleOnDevice(device_name_, handle); in ReleaseHandle()
[all …]
/external/tensorflow/tensorflow/python/framework/
Dpython_tensor_converter.h49 device_name_(device_name) {} in PythonTensorConverter()
71 const char* device_name_; variable
Dpython_tensor_converter.cc84 tensorflow::ConvertToEagerTensor(ctx_, src, dtype, device_name_); in Convert()
/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
Dop_handler_selector_test.cc111 device_name_(device_name), in FakeTensorHandle()
135 return device_name_.c_str(); in DeviceName()
157 std::string device_name_; member in tfrt::tf::__anond52f36620111::FakeTensorHandle
178 device_name_ = raw_device_name; in Reset()
184 const std::string& DeviceName() const override { return device_name_; } in DeviceName()
186 device_name_ = name; in SetDeviceName()
318 std::string device_name_; member in tfrt::tf::__anond52f36620111::FakeOperation
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.h56 const string& DeviceName() const override { return device_name_; } in DeviceName()
74 device_name_ = absl::visit( in SetDevice()
77 DeviceNameUtils::ParseFullName(device_name_, &device_parsed_name_); in SetDevice()
261 string device_name_; variable
Deager_operation.cc459 device_name_ = DeviceNameUtils::ParsedNameToString(device_parsed_name_); in SetDeviceName()
497 strings::StrAppend(&out, "Device Name: [", device_name_, "]\n"); in DebugString()
/external/webrtc/modules/desktop_capture/win/
Ddxgi_output_duplicator.h77 const std::string& device_name() const { return device_name_; } in device_name()
123 const std::string device_name_; variable
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental_graph.cc105 device_name_ = raw_device_name; in Reset()
127 const string& DeviceName() const override { return device_name_; } in DeviceName()
131 device_name_ = name; in SetDeviceName()
312 string device_name_; member in tensorflow::tracing::graph::GraphOperation
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device.cc57 : device_name_(name), parallel_device_(std::move(parallel_device)) {} in NamedParallelDevice()
58 const std::string& name() const { return device_name_; } in name()
62 std::string device_name_; member in tensorflow::parallel_device::__anon7f9de75a0111::NamedParallelDevice
/external/nos/host/generic/libnos/include/nos/
DNuggetClient.h112 std::string device_name_;
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc253 device_name_(device_name), in DeviceInstance()
256 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(device_name_, &device_)) in DeviceInstance()
257 << "Could not find device " << device_name_ << " existing devices " in DeviceInstance()
423 string device_name_; member in tensorflow::NcclTestBase::DeviceInstance
Ddebug_ops.h398 device_name_(context->device()->name()), in DebugIdentityV2Op()
438 tfdbg_context_id_, device_name_, op_name_, in Compute()
447 string device_name_; variable
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc214 string device_name_; member in mlir::TF::__anon376bf20e0111::MlirAbstractOp
449 const string& MlirAbstractOp::DeviceName() const { return device_name_; } in DeviceName()
452 device_name_ = name; in SetDeviceName()
/external/tensorflow/tensorflow/c/
Dkernels_test.cc999 op_name, device_name_, nullptr, compute_func, nullptr); in SetupOp()
1007 DeviceFactory::NewDevice(device_name_, {}, "/job:a/replica:0/task:0")); in SetupOp()
1015 const char* device_name_ = tensorflow::DEVICE_GPU; member in tensorflow::DeviceKernelOpTest
1017 const char* device_name_ = tensorflow::DEVICE_CPU; member in tensorflow::DeviceKernelOpTest
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_kernel.h341 std::string device_name_; variable

12