Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 15 of 15 for arg (0.14 sec)

  1. tensorflow/c/eager/c_api.cc

        void (*deallocator)(void* data, size_t len, void* arg),
        void* deallocator_arg, TF_Status* status) {
      tensorflow::Device* device = nullptr;
      tensorflow::EagerContext* context =
          tensorflow::ContextFromInterface(tensorflow::unwrap(ctx));
      status->status = context->FindDeviceFromName(device_name, &device);
      if (!status->status.ok()) {
        deallocator(data, len, deallocator_arg);
        status->status =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  2. tensorflow/c/eager/c_api_test.cc

        ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
    
        auto cpu_arg =
            tensorflow::TensorHandleFromInterface(tensorflow::unwrap(hcpu));
        auto gpu_arg =
            tensorflow::TensorHandleFromInterface(tensorflow::unwrap(hgpu));
        auto gpu_device = gpu_arg->device();
        ASSERT_FALSE(cpu_arg->HasLocalMirror(gpu_device));
    
        TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu);
        if (cpu_op) {
    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)
  3. tensorflow/c/experimental/filesystem/modular_filesystem_test.cc

    //
    // Example invocation:
    //  bazel test //tensorflow/c/experimental/filesystem:modular_filesystem_test \\
    //  --test_arg=--dso=/path/to/one.so --test_arg=--dso=/path/to/another.so \\
    //  --test_arg=--scheme= --test_arg=--scheme=file
    //
    // Note that to test the local filesystem we use an empty value.
    
    namespace tensorflow {
    namespace {
    
    using ::tensorflow::error::Code;
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri May 27 20:25:58 GMT 2022
    - 71K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental_test.cc

      // Build two abstract input tensors as function arguments.
      std::vector<TF_AbstractTensor*> func_args;
      {
        TFE_Context* eager_ctx =
            TF_ExecutionContextGetTFEContext(eager_execution_ctx, s);
    
        // 1st Arg
        float vals1[] = {0.0f, 1.0f, 1.0f, 0.0f};
        int64_t dims[] = {2, 2};  // Matrices will be 2 x 2
        int num_dims = sizeof(dims) / sizeof(dims[0]);
    
        TFE_TensorHandle* input_eager =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri May 19 21:44:52 GMT 2023
    - 39.1K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_test.cc

      EXPECT_EQ(TF_CANCELLED, TF_GetCode(s));
      EXPECT_EQ(string("cancel"), TF_Message(s));
      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(
    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