Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 27 for index (0.43 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc

          TFE_NewContext(opts.get(), status.get()), TFE_DeleteContext);
      tensorflow::ServerDef server_def = GetServerDef("worker", 3);
    
      // This server def has the task index set to 0.
      std::string serialized = server_def.SerializeAsString();
    
      server_def.set_task_index(1);
      std::unique_ptr<tensorflow::GrpcServer> worker_server1;
      ASSERT_TRUE(tensorflow::GrpcServer::Create(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 22:09:57 GMT 2023
    - 6.7K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.cc

      if (ic->num_outputs() <= new_src.index) {
        status->status = tensorflow::errors::OutOfRange(
            "Cannot update edge. Output index [", new_src.index,
            "] is greater than the number of total outputs [", ic->num_outputs(),
            "].");
        return;
      }
      tensorflow::shape_inference::ShapeHandle shape = ic->output(new_src.index);
    
      tensorflow::shape_inference::InferenceContext* ic_dst =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_cluster_test.cc

    namespace {
    
    using ::tensorflow::string;
    
    void ReplaceTaskInServerDef(tensorflow::ServerDef* server_def, int task_index) {
      tensorflow::JobDef* job_def = server_def->mutable_cluster()->mutable_job(0);
      int port = tensorflow::testing::PickUnusedPortOrDie();
      job_def->mutable_tasks()->at(task_index) =
          tensorflow::strings::StrCat("localhost:", port);
    }
    
    void CheckTFE_TensorHandleHasFloats(TFE_TensorHandle* handle,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_experimental.cc

                                            int index) {
      DCHECK(index >= 0 && index < shape_list->num_items);
      TF_ShapeAndType& shape = shape_list->items[index];
      DCHECK(shape.dims == nullptr) << "Shape at " << index << " is already set!";
      shape.num_dims = -1;
      shape.dims = nullptr;
    }
    
    void TF_ShapeAndTypeListSetDtype(TF_ShapeAndTypeList* shape_list, int index,
                                     TF_DataType dtype) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_remote_test.cc

    namespace {
    
    using ::tensorflow::string;
    
    void TestRemoteExecute(bool async) {
      tensorflow::ServerDef server_def = GetServerDef(2);
    
      // This server def has the task index set to 0.
      string serialized = server_def.SerializeAsString();
    
      server_def.set_task_index(1);
    
      std::unique_ptr<tensorflow::GrpcServer> worker_server;
      ASSERT_TRUE(tensorflow::GrpcServer::Create(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Aug 12 00:14:22 GMT 2020
    - 5.4K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/gradients/grad_test_helper.cc

        if (!outputs[i]) continue;
    
        AbstractTensorHandlePtr numerical_grad;
        {
          AbstractTensorHandle* numerical_grad_raw;
          s = CalcNumericalGrad(ctx, model, inputs,
                                /*input_index=*/i, use_function,
                                &numerical_grad_raw);
          ASSERT_EQ(errors::OK, s.code()) << s.message();
          numerical_grad.reset(numerical_grad_raw);
        }
    
        TF_Tensor* numerical_tensor;
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5K bytes
    - Viewed (0)
  7. tensorflow/c/experimental/grappler/grappler.cc

      const int len = std::min(num_values, static_cast<int>(nodes.size()));
      char* p = static_cast<char*>(storage);
      for (int index = 0; index < len; ++index) {
        const std::string& s = nodes[index];
        values[index] = p;
        lengths[index] = s.size();
        if ((p + s.size()) > (static_cast<char*>(storage) + storage_size)) {
          tsl::Set_TF_Status_from_Status(
              status,
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 15K bytes
    - Viewed (1)
  8. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      for (int device_index = 0; device_index < underlying_devices_.size();
           ++device_index) {
        DeviceThread* device_thread = device_threads_[device_index].get();
        std::vector<TFE_TensorHandle*> device_inputs;
        device_inputs.reserve(inputs.size());
        for (int input_index = 0; input_index < inputs.size(); ++input_index) {
          // Parallel tensors are divided between operations by device.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  9. tensorflow/c/eager/c_api_experimental_test.cc

      tensorflow::ServerDef server_def_1 =
          ReplaceTaskInServerDef(server_def_0, /*task_index=*/0);
    
      // These server defs have task index set to 0.
      string serialized_server_def_0 = server_def_0.SerializeAsString();
      string serialized_server_def_1 = server_def_1.SerializeAsString();
    
      // Create two worker tasks.
      server_def_0.set_task_index(1);
      std::unique_ptr<tensorflow::GrpcServer> worker_server1;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 03:14:26 GMT 2023
    - 31.5K bytes
    - Viewed (1)
  10. tensorflow/c/eager/c_api_debug.cc

        TFE_TensorDebugInfo* debug_info) {
      return debug_info->dev_dims.size();
    }
    
    TF_CAPI_EXPORT extern int64_t TFE_TensorDebugInfoOnDeviceDim(
        TFE_TensorDebugInfo* debug_info, int dim_index) {
      return debug_info->dev_dims[dim_index];
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Aug 11 01:20:50 GMT 2021
    - 2.5K bytes
    - Viewed (0)
Back to top