Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for SetDeviceName (0.32 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
Back to top