Home
last modified time | relevance | path

Searched refs:Safe_TF_TensorPtr (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/lib/core/
Dsafe_ptr.cc24 Safe_TF_TensorPtr make_safe(TF_Tensor* tensor) { in make_safe()
25 return Safe_TF_TensorPtr(tensor); in make_safe()
Dndarray_tensor.h29 Status TF_TensorToPyArray(Safe_TF_TensorPtr tensor, PyObject** out_ndarray);
37 Status PyArrayToTF_Tensor(PyObject* ndarray, Safe_TF_TensorPtr* out_tensor);
Dsafe_ptr.h54 using Safe_TF_TensorPtr = std::unique_ptr<TF_Tensor, detail::TFTensorDeleter>; variable
55 Safe_TF_TensorPtr make_safe(TF_Tensor* tensor);
Dndarray_tensor.cc360 Status TF_TensorToPyArray(Safe_TF_TensorPtr tensor, PyObject** out_ndarray) { in TF_TensorToPyArray()
416 Status PyArrayToTF_Tensor(PyObject* ndarray, Safe_TF_TensorPtr* out_tensor) { in PyArrayToTF_Tensor()
471 Safe_TF_TensorPtr tf_tensor = make_safe(static_cast<TF_Tensor*>(nullptr)); in NdarrayToTensor()
481 Safe_TF_TensorPtr tf_tensor = make_safe(TF_TensorFromTensor(t, status)); in TensorToNdarray()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc73 std::vector<Safe_TF_TensorPtr> inputs_safe; // Used to delete tensors. in TF_Run_wrapper_helper()
135 std::vector<Safe_TF_TensorPtr> tf_outputs_safe; in TF_Run_wrapper_helper()
387 std::vector<Safe_TF_TensorPtr> input_vals_safe; in TF_SessionRun_wrapper_helper()
419 std::vector<Safe_TF_TensorPtr> output_vals_safe; in TF_SessionRun_wrapper_helper()
677 Safe_TF_TensorPtr safe_result_tensor(result_tensor); in TF_TryEvaluateConstant_wrapper()