Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for TestVariable (0.2 sec)

  1. tensorflow/c/eager/c_api_cluster_test.cc

      const char dev1_name[] = "/job:localhost/replica:0/task:1/device:CPU:0";
    
      TFE_TensorHandle* var_handle0 = TestVariable(ctx, 1.0, dev0_name);
      EXPECT_NE(var_handle0, nullptr);
      TFE_TensorHandle* var_handle1 = TestVariable(ctx, 2.0, dev1_name);
      EXPECT_NE(var_handle1, nullptr);
    
      TFE_TensorHandle* value_handle = nullptr;
      ReadVariable(ctx, var_handle1, &value_handle);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_distributed_test.cc

      const char task2_name[] = "/job:localhost/replica:0/task:2/device:CPU:0";
    
      // Create one variable per task.
      TFE_TensorHandle* h0 = TestVariable(ctx, 1.0, task1_name);
      TFE_TensorHandle* h1 = TestVariable(ctx, 2.0, task2_name);
      TFE_TensorHandle* h2 = TestVariable(ctx, 3.0, task0_name);
    
      // Add a sync point to make sure that variables have been initialized
      // before the function execution starts.
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_test_util.h

    TFE_TensorHandle* TestMatrixTensorHandle3X2(TFE_Context* ctx);
    
    // Return a variable handle referring to a variable with the given initial value
    // on the given device.
    TFE_TensorHandle* TestVariable(TFE_Context* ctx, float value,
                                   const tensorflow::string& device_name = "");
    
    // Return an add op multiplying `a` by `b`.
    TFE_Op* AddOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Jul 17 23:43:59 GMT 2023
    - 7.7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_test_util.cc

      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TF_DeleteTensor(t);
      TF_DeleteStatus(status);
      return th;
    }
    
    TFE_TensorHandle* TestVariable(TFE_Context* ctx, float value,
                                   const tensorflow::string& device_name) {
      TF_Status* status = TF_NewStatus();
      // Create the variable handle.
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  5. tensorflow/c/eager/c_api_test.cc

      TFE_Context* ctx = TFE_NewContext(opts, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_DeleteContextOptions(opts);
    
      TFE_TensorHandle* var_handle = TestVariable(ctx, 12.0);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
Back to top