Searched defs:indices_dtype (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_tensor_dense_matmul_grad_test.py | 34 def _sparsify(self, x, indices_dtype=np.int64): argument
50 indices_dtype=np.int64): argument
63 indices_dtype): argument
85 def _testGradientsType(self, values_dtype, indices_dtype): argument
|
| D | batch_gather_op_test.py | 45 def testSimpleGather(self, indices_dtype): argument
61 def test2DArray(self, indices_dtype): argument
|
| D | sparse_tensor_dense_matmul_op_test.py | 55 indices_dtype=np.int64): argument
92 def _testBasic(self, value_dtype, indices_dtype=np.int64): argument
|
| D | sparse_ops_test.py | 115 def _SparseTensorValue_3x50(self, indices_dtype, values_dtype): argument
131 def _SparseTensor_3x50(self, indices_dtype, values_dtype): argument
229 def _SparseTensor_3x50(self, indices_dtype, values_dtype): argument
|