Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for CreateTestingDeviceMgr (0.26 sec)

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

    #include "tensorflow/core/platform/logging.h"
    #include "tensorflow/core/platform/test.h"
    #include "tensorflow/core/platform/types.h"
    
    namespace tensorflow {
    namespace testing {
    
    std::unique_ptr<StaticDeviceMgr> CreateTestingDeviceMgr() {
      return std::make_unique<StaticDeviceMgr>(
          DeviceFactory::NewDevice("CPU", {}, "/job:localhost/replica:0/task:0"));
    }
    
    EagerContextPtr CreateTestingEagerContext(DeviceMgr* device_mgr) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Sep 08 20:13:32 UTC 2023
    - 5.8K bytes
    - Viewed (0)
Back to top