Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dnumpy_bridge.cc258 PyObject* py_dimensions; in XlaShapeFromPyShape() local
260 TF_ASSIGN_OR_RETURN(py_dimensions, call_method("dimensions")); in XlaShapeFromPyShape()
262 if (!PyTuple_Check(py_dimensions)) { in XlaShapeFromPyShape()
270 const int length = PyTuple_Size(py_dimensions); in XlaShapeFromPyShape()
280 dimensions[i] = PyIntOrPyLongToLong(PyTuple_GetItem(py_dimensions, i)); in XlaShapeFromPyShape()
294 Py_DECREF(py_dimensions); in XlaShapeFromPyShape()