Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for addMap (0.27 sec)

  1. tensorflow/c/eager/c_api_test.cc

      // Store pointer to raw buffer for validation of forwarding behaviour.
      TF_Tensor* orig = TFE_TensorHandleResolve(n, status);
      void* orig_ptr = TF_TensorData(orig);
      TF_DeleteTensor(orig);
    
      TFE_Op* add_op = AddOp(ctx, n, m);
      std::string cpu_device_name;
      ASSERT_TRUE(GetDeviceName(ctx, &cpu_device_name, "CPU"));
      TFE_OpSetDevice(add_op, cpu_device_name.c_str(), status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  2. tensorflow/c/eager/c_api_test_util.cc

      TFE_DeleteOp(op);
      if (TF_GetCode(status) != TF_OK) return nullptr;
      CHECK_EQ(0, num_retvals);
    
      TF_DeleteStatus(status);
    
      return var_handle;
    }
    
    TFE_Op* AddOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b) {
      TF_Status* status = TF_NewStatus();
    
      TFE_Op* op = TFE_NewOp(ctx, "AddV2", status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
Back to top