Home
last modified time | relevance | path

Searched defs:np_dtype (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_add_op_test.py53 def _randomTensor(self, size, np_dtype, sparse=True): argument
Dsparse_tensor_dense_matmul_op_test.py212 def _testLarge(self, np_dtype): argument
Dconfusion_matrix_test.py81 def _testConfMatrixOnTensors(self, tf_dtype, np_dtype): argument
Darray_ops_test.py304 def _reverse1DimAuto(self, np_dtype): argument
315 def _reverse2DimAuto(self, np_dtype): argument
461 def _compareDiffType(self, n, np_dtype, use_gpu): argument
/external/tensorflow/tensorflow/compiler/xla/python/
Dnumpy_bridge.cc132 PyArray_Descr* np_dtype = PyArray_DescrFromType(np_typenum); in PyShapeInfoFromXlaShape() local