Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for BuildDeviceManager (0.16 sec)

  1. tensorflow/compiler/mlir/tensorflow/utils/fake_session.cc

      // global initialization fiasco between tests and code in TF.
    }
    
    void FakeSession::Initialize() {
      if (initialized_) return;
      BuildDeviceManager();
      InitVariables();
      initialized_ = true;
    }
    
    void FakeSession::BuildDeviceManager() {
      auto device =
          tensorflow::DeviceFactory::NewDevice("CPU", {}, kDeviceNamePrefix);
      device_mgr_ =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 03:47:51 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/fake_session.h

          tensorflow::RunMetadata* run_metadata,
          const tensorflow::thread::ThreadPoolOptions& thread_pool_options)
          override;
    
     private:
      void InitVariables();
      void BuildDeviceManager();
      void Initialize();
    
      std::unique_ptr<tensorflow::DeviceMgr> device_mgr_;
      bool initialized_ = false;
    };
    
    }  // namespace test_util
    }  // namespace TF
    }  // namespace mlir
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 06 04:50:13 UTC 2023
    - 3K bytes
    - Viewed (0)
Back to top