OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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