Searched refs:PyTapeTensor (Results 1 – 1 of 1) sorted by relevance
1185 class PyTapeTensor { class1187 PyTapeTensor(int64_t id, tensorflow::DataType dtype, in PyTapeTensor() function in PyTapeTensor1190 PyTapeTensor(int64_t id, tensorflow::DataType dtype, PyObject* shape) in PyTapeTensor() function in PyTapeTensor1194 PyTapeTensor(const PyTapeTensor& other) { in PyTapeTensor() function in PyTapeTensor1203 ~PyTapeTensor() { in ~PyTapeTensor()1227 static PyTapeTensor TapeTensorFromTensor(PyObject* tensor);1230 PyTapeTensor> {1339 Status BuildOnesLike(const PyTapeTensor& t, in BuildOnesLike()1420 PyTapeTensor TapeTensorFromGradient(PyObject* tensor) const final { in TapeTensorFromGradient()1463 PyObject* PyTapeTensor::GetShape() const { in GetShape()[all …]