Searched defs:indices_dtype (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_tensor_dense_matmul_grad_test.py | 30 def _sparsify(self, x, indices_dtype=np.int64): argument
46 indices_dtype=np.int64): argument
61 indices_dtype): argument
86 def _testGradientsType(self, values_dtype, indices_dtype): argument
|
| D | sparse_tensor_dense_matmul_op_test.py | 51 indices_dtype=np.int64): argument
90 def _testBasic(self, value_dtype, indices_dtype=np.int64): argument
|
| D | sparse_ops_test.py | 111 def _SparseTensorValue_3x50(self, indices_dtype, values_dtype): argument
127 def _SparseTensor_3x50(self, indices_dtype, values_dtype): argument
225 def _SparseTensor_3x50(self, indices_dtype, values_dtype): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | batch_gather_op_test.py | 42 def testSimpleGather(self, indices_dtype): argument
58 def test2DArray(self, indices_dtype): argument
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/include/ |
| D | pack_block_sparse.h | 82 enum pytorch_qnnp_sparse_matrix_indices_dtype indices_dtype; member
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | indexed_slices.py | 224 indices_dtype=dtypes.int64, dense_shape_dtype=None, argument
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| D | operator.h | 54 enum pytorch_qnnp_sparse_matrix_indices_dtype indices_dtype; member
|