Searched refs:EagerTensor_Handle (Results 1 – 10 of 10) sorted by relevance
12 TFE_TensorHandle* thandle = EagerTensor_Handle(eager_tensor_pyobject_ptr);27 TFE_TensorHandle* EagerTensor_Handle(const PyObject* o) {
231 input_vector[i] = EagerTensor_Handle(py_eager_tensor_handles[i].get()); in PYBIND11_MODULE()270 TFE_TensorHandle* input_handle = EagerTensor_Handle(dtensor_handle.ptr()); in PYBIND11_MODULE()296 EagerTensor_Handle(dtensor_handle.ptr()), device_info, in PYBIND11_MODULE()309 TFE_TensorHandle* input_handle = EagerTensor_Handle(dtensor_handle.ptr()); in PYBIND11_MODULE()
135 (input_tensor_handles)[i] = EagerTensor_Handle(elem); in InputTFE_InputTensorHandles()150 (input_tensor_handles)[i] = EagerTensor_Handle(value_attr.get()); in InputTFE_InputTensorHandles()1586 TFE_TensorHandle* thandle = EagerTensor_Handle(eager_tensor_pyobject_ptr); in PYBIND11_MODULE()
27 TFE_TensorHandle* EagerTensor_Handle(const PyObject* o);
142 TFE_TensorHandle* EagerTensor_Handle(const PyObject* o);
956 TFE_TensorHandle* EagerTensor_Handle(const PyObject* o) { in EagerTensor_Handle() function1229 TFE_TensorHandle* handle = EagerTensor_Handle(tensor); in TFE_Py_TensorShapeOnDevice()
2162 tensorflow::unwrap(EagerTensor_Handle(tensor));3415 TFE_OpAddInput(op, EagerTensor_Handle(handle.get()), status);3522 TFE_TensorHandle* input_handle = EagerTensor_Handle(py_eager_tensor.get());3878 EagerTensor_Handle(py_eager_tensor.get());
254 TFE_TensorHandle* eager_tensor = EagerTensor_Handle(handle.ptr()); in PYBIND11_MODULE()
156 tensorflow::unwrap(EagerTensor_Handle(eager_tensor)))); in ExtractTensorFromEagerTensor()
156 EagerTensor_Handle