Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for device_mgr_ (0.1 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc

                                            /*graph=*/nullptr);
      context_->Ref();
    
      device_mgr_ = CreateDeviceMgr(device_type_);
      if (!device_mgr_) return failure();
    
      // Type of params_.device is DeviceBase* so store it as Device* to access
      // derived class method.
      device_ = device_mgr_->ListDevices().front();
      params_.device = device_;
      params_.resource_manager = device_->resource_manager();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 18.9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

          &devices);
      if (!status->status.ok()) return nullptr;
      std::unique_ptr<tensorflow::DeviceMgr> device_mgr(
          new tensorflow::DynamicDeviceMgr(std::move(devices)));
    
      auto r = tsl::core::RefCountPtr<tensorflow::IntraProcessRendezvous>(
          new tensorflow::IntraProcessRendezvous(device_mgr.get()));
      tensorflow::EagerContext* eager_context = new tensorflow::EagerContext(
          opts->session_options.options,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc

    }
    
    // Returns ElementsAttr which has the value held by 'resource_tensor'.
    ElementsAttr GetTensorValueAsElementsAttr(
        TF::VarHandleOp var_handle_op, const tensorflow::Tensor& resource_tensor,
        const tensorflow::DeviceMgr* mgr, OpBuilder builder) {
      if (resource_tensor.dtype() != tensorflow::DT_RESOURCE) {
        return GetTensorValueAsElementsAttr(resource_tensor, builder);
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 09:56:53 UTC 2024
    - 19.4K bytes
    - Viewed (0)
Back to top