Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for experimental_distributed_variable_components (0.5 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/c/experimental/saved_model/core/saved_model_utils.cc 

      tensorflow::TensorShape shape(variable.shape());
  tensorflow::DataType dtype = variable.dtype();
  std::vector<std::string> component_devices;

  for (const auto& component :
       variable.experimental_distributed_variable_components()) {
    component_devices.push_back(component.device());
  }

  TF_RETURN_IF_ERROR(Variable::CreateUninitialized(
      ctx, dtype, shape, name,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jan 12 19:17:46 UTC 2023
    - 24K bytes
    - Viewed (0)
Back to top