- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for SetDeviceName (0.32 sec)
-
tensorflow/c/eager/abstract_operation.h
// Returns the operation's device name. // // The value returned may be different from the one set by SetDeviceName, but // it will be compatible with it: the name will be updated by device placement // logic to refer to the specific device chosen. // // Example: If one calls `op->SetDeviceName("/device:GPU")`, the value // returned by DeviceName should be "/device:GPU:*" until a particular GPU is
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Jul 14 16:20:41 GMT 2021 - 6.8K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.h
void Release() override; Status Reset(const char* op, const char* raw_device_name) override; const string& Name() const override; const string& DeviceName() const override; Status SetDeviceName(const char* name) override; Status AddInput(AbstractTensorHandle* input) override; Status AddInputList(absl::Span<AbstractTensorHandle* const> inputs) override; Status Execute(absl::Span<AbstractTensorHandle*> retvals,
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 3.7K bytes - Viewed (1) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
} const string& TapeOperation::Name() const { return parent_op_->Name(); } const string& TapeOperation::DeviceName() const { return parent_op_->DeviceName(); } Status TapeOperation::SetDeviceName(const char* name) { return parent_op_->SetDeviceName(name); } Status TapeOperation::AddInput(AbstractTensorHandle* input) { TF_RETURN_IF_ERROR(parent_op_->AddInput(input)); forward_op_.inputs.push_back(input); return OkStatus(); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
return absl::OkStatus(); } const string& Name() const override { return op_type_; } const string& DeviceName() const override { return device_name_; } Status SetDeviceName(const char* name) override { // TODO(srbs): Implement this. device_name_ = name; return absl::OkStatus(); } Status AddInput(AbstractTensorHandle* input) override {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/eager/c_api.cc
return tensorflow::wrap(tensorflow::unwrap(op)->GetContext()); } void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) { status->status = tensorflow::unwrap(op)->SetDeviceName(device_name); } const char* TFE_OpGetDevice(const TFE_Op* op, TF_Status* status) { return tensorflow::unwrap(op)->DeviceName().c_str(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2)