MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / PyTensorObject_data_ptr

Function PyTensorObject_data_ptr

oneflow/api/python/framework/tensor.cpp:400–408  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

398}
399
400static PyObject* PyTensorObject_data_ptr(PyObject* self, PyObject* unused) {
401 HANDLE_ERRORS
402 const auto& t = PyTensor_Unpack(self);
403 const std::shared_ptr<LocalTensor> local_tensor =
404 t->is_local() ? ASSERT_PTR(t->AsLocalTensor()) : ASSERT_PTR(t->cur_rank_phy_tensor());
405 return functional::CastToPyObject(
406 reinterpret_cast<int64_t>(ASSERT(GetTensorDataPtr(local_tensor))));
407 END_HANDLE_ERRORS
408}
409
410static PyObject* PyTensorObject_to_numpy(PyObject* self, PyObject* unused) {
411 HANDLE_ERRORS

Callers

nothing calls this directly

Calls 5

CastToPyObjectFunction · 0.85
GetTensorDataPtrFunction · 0.85
is_localMethod · 0.45
AsLocalTensorMethod · 0.45
cur_rank_phy_tensorMethod · 0.45

Tested by

no test coverage detected