HomeSort by relevance Sort by last modified time
    Searched refs:indices_dtype (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/python/kernel_tests/
sparse_tensor_dense_matmul_grad_test.py 33 def _sparsify(self, x, indices_dtype=np.int64):
37 x_indices = np.vstack(non_zero).astype(indices_dtype).T
49 indices_dtype=np.int64):
57 return self._sparsify(x, indices_dtype=indices_dtype)
62 indices_dtype):
69 indices_dtype=indices_dtype)
84 def _testGradientsType(self, values_dtype, indices_dtype):
88 adjoint_a, adjoint_b, values_dtype.__name__, indices_dtype.__name__
    [all...]
sparse_tensor_dense_matmul_op_test.py 54 indices_dtype=np.int64):
64 x_indices = np.vstack(np.where(x)).astype(indices_dtype).T
91 def _testBasic(self, value_dtype, indices_dtype=np.int64):
97 self._testMatmul(x, y, indices_dtype=indices_dtype)
106 self._testBasic(np.int32, indices_dtype=np.int32)
107 self._testBasic(np.float32, indices_dtype=np.int32)
sparse_ops_test.py 113 def _SparseTensorValue_3x50(self, indices_dtype, values_dtype):
123 np.array(indices, indices_dtype), np.array(shape, np.int64))
129 def _SparseTensor_3x50(self, indices_dtype, values_dtype):
130 indices, values = self._SparseTensorValue_3x50(indices_dtype, values_dtype)
211 def _SparseTensor_3x50(self, indices_dtype, values_dtype):
222 np.array(indices0, indices_dtype), np.array(shape, np.int64))
225 np.array(indices1, indices_dtype), np.array(shape, np.int64))
    [all...]

Completed in 67 milliseconds