Searched refs:indices_dtype (Results 1 – 5 of 5) 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
38 x_indices = np.vstack(non_zero).astype(indices_dtype).T
50 indices_dtype=np.int64): argument
58 return self._sparsify(x, indices_dtype=indices_dtype)
63 indices_dtype): argument
70 indices_dtype=indices_dtype)
85 def _testGradientsType(self, values_dtype, indices_dtype): argument
89 adjoint_a, adjoint_b, values_dtype.__name__, indices_dtype.__name__)
91 indices_dtype)
|
| D | batch_gather_op_test.py | 45 def testSimpleGather(self, indices_dtype): argument
52 indices_tf = constant_op.constant(indices, dtype=indices_dtype)
61 def test2DArray(self, indices_dtype): argument
68 indices_tf = constant_op.constant(indices, dtype=indices_dtype)
|
| D | sparse_tensor_dense_matmul_op_test.py | 55 indices_dtype=np.int64): argument
65 x_indices = np.vstack(np.where(x)).astype(indices_dtype).T
92 def _testBasic(self, value_dtype, indices_dtype=np.int64): argument
98 self._testMatmul(x, y, indices_dtype=indices_dtype)
108 self._testBasic(np.int32, indices_dtype=np.int32)
109 self._testBasic(np.float32, indices_dtype=np.int32)
|
| D | sparse_ops_test.py | 115 def _SparseTensorValue_3x50(self, indices_dtype, values_dtype): argument
125 np.array(indices, indices_dtype), np.array(shape, np.int64))
131 def _SparseTensor_3x50(self, indices_dtype, values_dtype): argument
132 indices, values = self._SparseTensorValue_3x50(indices_dtype, values_dtype)
229 def _SparseTensor_3x50(self, indices_dtype, values_dtype): argument
240 np.array(indices0, indices_dtype), np.array(shape, np.int64))
243 np.array(indices1, indices_dtype), np.array(shape, np.int64))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3495 indices_dtype = indices.dtype.base_dtype
3496 accum_dim_value = ones((), dtype=indices_dtype)
3498 casted_params_shape = gen_math_ops.cast(params_shape, indices_dtype)
3502 start = zeros((), dtype=indices_dtype)
3503 step = ones((), dtype=indices_dtype)
|