Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for AllocateRaw (0.3 sec)

  1. tensorflow/c/c_api_test.cc

      *reinterpret_cast<bool*>(arg) = true;
    }
    
    TEST(CAPI, Tensor) {
      const int num_bytes = 6 * sizeof(float);
      float* values =
          reinterpret_cast<float*>(tensorflow::cpu_allocator()->AllocateRaw(
              TF_TensorDefaultAlignment(), num_bytes));
      int64_t dims[] = {2, 3};
      bool deallocator_called = false;
      TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
Back to top