Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 30 for device_mgr_ (0.26 sec)

  1. 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)
  2. tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc

    #include "tensorflow/compiler/tf2xla/functionalize_control_flow.h"
    #include "tensorflow/core/common_runtime/device.h"
    #include "tensorflow/core/common_runtime/device_factory.h"
    #include "tensorflow/core/common_runtime/device_mgr.h"
    #include "tensorflow/core/grappler/clusters/virtual_cluster.h"
    #include "tensorflow/core/grappler/grappler_item.h"
    #include "tensorflow/core/grappler/grappler_item_builder.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Feb 27 14:33:47 UTC 2024
    - 6.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla.cc

    #include "xla/client/xla_builder.h"
    #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h"
    #include "tensorflow/core/common_runtime/device.h"
    #include "tensorflow/core/common_runtime/device_factory.h"
    #include "tensorflow/core/common_runtime/device_mgr.h"
    #include "tensorflow/core/common_runtime/process_function_library_runtime.h"
    #include "tensorflow/core/framework/allocator.h"
    #include "tensorflow/core/framework/function.h"
    #include "tensorflow/core/framework/function.pb.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 9.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc

        return CreateSessionInitFunc(module);
      }
    }
    
    }  // namespace
    
    LogicalResult InitializeVariablesInSessionInitializer(
        ModuleOp module, tensorflow::Session* session) {
      const tensorflow::DeviceMgr* mgr = nullptr;
      auto status = session->LocalDeviceManager(&mgr);
      if (!status.ok()) {
        module->emitError(
            absl::StrCat("failed to fetch device manager: ", status.message()));
        return failure();
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 7K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_internal.h

    #include "tensorflow/core/platform/mutex.h"
    #include "tensorflow/core/platform/status.h"
    #include "tensorflow/core/platform/types.h"
    #include "tensorflow/core/public/session.h"
    
    namespace tensorflow {
    class Device;
    class DeviceMgr;
    class ServerInterface;
    }  // namespace tensorflow
    
    // Internal structures used by the C API. These are likely to change and should
    // not be depended on.
    
    struct TF_SessionOptions {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat May 13 00:49:12 UTC 2023
    - 7.6K bytes
    - Viewed (0)
  6. 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)
  7. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

      }
    
      std::unique_ptr<DeviceMgr> device_mgr =
          std::make_unique<StaticDeviceMgr>(std::move(devices));
      const auto* config = &options.session_options->config;
      std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(
          new ProcessFunctionLibraryRuntime(
              device_mgr.get(), options.session_options->env,
              /*config=*/config, TF_GRAPH_DEF_VERSION, library,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

      TF_CHECK_OK(DeviceFactory::AddDevices(
          session_options, "/job:localhost/replica:0/task:0", &devices));
      OptimizerOptions opts;
      auto device_mgr = std::make_unique<StaticDeviceMgr>(std::move(devices));
      auto pflr = std::make_unique<ProcessFunctionLibraryRuntime>(
          device_mgr.get(), Env::Default(), /*config=*/nullptr,
          TF_GRAPH_DEF_VERSION, lib_def.get(), opts,
          /*default_thread_pool=*/nullptr, /*cluster_flr=*/nullptr);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
  9. tensorflow/c/c_api.cc

    #include "tensorflow/c/c_api_internal.h"
    #include "tensorflow/c/tf_buffer_internal.h"
    #include "tensorflow/c/tf_status_internal.h"
    #include "tensorflow/c/tf_tensor.h"
    #include "tensorflow/core/common_runtime/device_mgr.h"
    #include "tensorflow/core/common_runtime/eval_const_tensor.h"
    #include "tensorflow/core/common_runtime/graph_constructor.h"
    #include "tensorflow/core/common_runtime/shape_refiner.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  10. RELEASE.md

        params after the creation of Estimator, leading to potentially undefined
        behavior.
    *   Added None check for save_path in `saver.restore`.
    *   Register devices under their legacy names in device_mgr to ease the
        transition to clusterspec-propagated configurations.
    *   VectorExponential added to distributions.
    *   Add a bitwise module with bitwise_and, bitwise_or, bitwise_xor, and invert
        functions.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 730.3K bytes
    - Viewed (0)
Back to top