Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TensorHandleFromInterface (0.14 sec)

  1. tensorflow/c/eager/c_api_remote_test_util.cc

        TFE_OpSetDevice(matmul, cpu_device_name.c_str(), status);
        EXPECT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
        auto remote_arg =
            tensorflow::TensorHandleFromInterface(tensorflow::unwrap(h1_task2));
        // The input handles should never change since they have been mirrored.
        ASSERT_FALSE(remote_arg->HasLocalMirror(nullptr));
      }
    
      TFE_TensorHandle* retvals[1];
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Dec 11 22:56:03 UTC 2020
    - 9.1K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_debug.cc

    }
    
    }  // namespace
    
    extern "C" {
    
    TF_CAPI_EXPORT extern TFE_TensorDebugInfo* TFE_TensorHandleTensorDebugInfo(
        TFE_TensorHandle* h, TF_Status* status) {
      tensorflow::TensorHandle* handle =
          TensorHandleFromInterface(tensorflow::unwrap(h));
      const tensorflow::Tensor* tensor;
      status->status = handle->Tensor(&tensor);
      if (!status->status.ok()) {
        return nullptr;
      }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 2.5K bytes
    - Viewed (0)
Back to top