Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for CPU (0.16 sec)

  1. tensorflow/c/eager/c_api_test.cc

      TFE_DeleteTensorHandle(retvals[0]);
      TFE_DeleteContext(ctx);
      TF_DeleteStatus(status);
    }
    TEST(CAPI, Execute_MatMul_CPU_Runtime_Error) {
      Execute_MatMul_CPU_Runtime_Error(false);
    }
    TEST(CAPI, Execute_MatMul_CPU_Runtime_ErrorAsync) {
      Execute_MatMul_CPU_Runtime_Error(true);
    }
    
    void Execute_MatMul_CPU_Type_Error(bool async) {
      TF_Status* status = TF_NewStatus();
      TFE_ContextOptions* opts = TFE_NewContextOptions();
    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/c_api_test.cc

      TF_DeleteStatus(s);
    }
    
    void Deallocator(void* data, size_t, void* arg) {
      tensorflow::cpu_allocator()->DeallocateRaw(data);
      *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};
    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)
  3. tensorflow/c/c_api_function_test.cc

            TF_NewOperation(host_graph_, func_name_, func_node_name_);
        for (auto input : inputs) {
          TF_AddInput(desc, input);
        }
        // Set device to CPU because some ops inside the function might not be
        // available on GPU.
        TF_SetDevice(desc, "/cpu:0");
        *op = TF_FinishOperation(desc, s_);
        ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
        ASSERT_NE(*op, nullptr);
      }
    
      FunctionDef fdef() {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
Back to top