Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for device_mgr_ (0.14 sec)

  1. tensorflow/c/experimental/saved_model/core/tf_concrete_function_loading_test.cc

     public:
      SavedConcreteFunctionLoadingTest()
          : device_mgr_(testing::CreateTestingDeviceMgr()),
            ctx_(testing::CreateTestingEagerContext(device_mgr_.get())) {}
    
      EagerContext* context() { return ctx_.get(); }
    
     private:
      std::unique_ptr<StaticDeviceMgr> device_mgr_;
      EagerContextPtr ctx_;
    };
    
    class DummyCapture : public TensorHandleConvertible {
     public:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 14 19:16:58 UTC 2023
    - 10.6K bytes
    - Viewed (0)
  2. 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)
  3. tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc

            func_name_attrs, new_func_name, host_graph_func_name, host_compute_core,
            flr, fld, shape_inference_graphs, has_outside_compilation);
      }
    
     private:
      std::unique_ptr<DeviceMgr> device_mgr_;
      std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_;
    };
    
    TEST_F(ExtractOutsideCompilationForFunctionTest, Basic) {
      // Build the XLA computation func.
      // "const0"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 41K bytes
    - Viewed (0)
  4. 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)
  5. 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