Searched defs:np_type (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | numpy_bridge.cc | 70 PrimitiveType NumpyTypeToPrimitiveType(int np_type) { in NumpyTypeToPrimitiveType()
107 bool NumpyTypeIsValid(int np_type) { in NumpyTypeIsValid()
223 PyObject* np_type; in XlaShapeFromPyShape() local
389 int np_type = PrimitiveTypeToNumpyType(literal.shape().element_type()); in PyObjectFromXlaLiteral() local
416 int np_type = PyArray_TYPE(py_array); in XlaLiteralFromPyObject() local
427 Status CopyNumpyArrayToLiteral(int np_type, PyArrayObject* py_array, in CopyNumpyArrayToLiteral()
479 Status CopyLiteralToNumpyArray(int np_type, const LiteralSlice& literal, in CopyLiteralToNumpyArray()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | matrix_square_root_op_test.py | 33 def _verifySquareRoot(self, matrix, np_type): argument
|
| D | matrix_logarithm_op_test.py | 41 def _verifyLogarithm(self, x, np_type): argument
|
| D | matrix_inverse_op_test.py | 38 def _verifyInverse(self, x, np_type): argument
|
| D | matrix_exponential_op_test.py | 52 def _verifyExponential(self, x, np_type): argument
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_func.cc | 372 const int np_type = PyArray_TYPE(input); in ConvertNdarrayToTensor() local
|