| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 103 sp_indices = op.inputs[0]
105 return (None, array_ops.gather_nd(out_grad, sp_indices), None, out_grad)
111 sp_indices = op.inputs[0]
117 return (None, array_ops.gather_nd(out_grad_reshaped, sp_indices // scale),
|
| D | math_ops.py | 995 def _sparse_dense_truediv(sp_indices, sp_values, sp_shape, y, name=None): argument
998 [sp_indices, sp_values, sp_shape, y]) as name:
1014 sp_indices, sp_values, sp_shape, y, name=name)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | string_split_op.cc | 105 auto sp_indices = sp_indices_t->matrix<int64>(); in Compute() local
113 sp_indices(c, 0) = i; in Compute()
114 sp_indices(c, 1) = j; in Compute()
|
| D | sparse_dense_binary_op_shared_test.cc | 211 static Graph* SparseMatCMulDenseMat(Graph* g, Node* sp_indices, Node* sp_vals, in SparseMatCMulDenseMat() argument
216 .Input(sp_indices) in SparseMatCMulDenseMat()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseDenseCwiseDiv.pbtxt | 4 name: "sp_indices"
13 1-D. `N` non-empty values corresponding to `sp_indices`.
|
| D | api_def_SparseDenseCwiseAdd.pbtxt | 4 name: "sp_indices"
13 1-D. `N` non-empty values corresponding to `sp_indices`.
|
| D | api_def_SparseDenseCwiseMul.pbtxt | 4 name: "sp_indices"
13 1-D. `N` non-empty values corresponding to `sp_indices`.
|
| D | api_def_SparseSoftmax.pbtxt | 4 name: "sp_indices"
13 1-D. `NNZ` non-empty values corresponding to `sp_indices`.
|
| /external/tensorflow/tensorflow/contrib/losses/python/metric_learning/ |
| D | metric_loss_ops_test.py | 203 sp_indices = []
206 sp_indices.append([j])
208 num_non_zeros = len(sp_indices)
210 indices=np.array(sp_indices),
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_test.py | 614 sp_indices = array_ops.identity(sp.indices)
617 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
620 [sp_indices, sp_values, sp_shape], {
633 [sp_indices, sp_values, sp_shape], {
660 sp_indices = array_ops.identity(sp.indices)
663 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
666 [sp_indices, sp_values, sp_shape], {
674 [sp_indices, sp_values, sp_shape], {
695 sp_indices = array_ops.identity(sp.indices)
698 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | lookup_ops_test.py | 161 sp_indices = [[0, 0], [0, 1], [1, 0]]
164 constant_op.constant(sp_indices, dtypes.int64),
172 self.assertAllEqual(sp_indices, out_indices)
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| D | lookup_ops_test.py | 167 sp_indices = [[0, 0], [0, 1], [1, 0]]
170 constant_op.constant(sp_indices, dtypes.int64),
178 self.assertAllEqual(sp_indices, out_indices)
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 5771 func SparseSoftmax(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Output) (ou…
5778 sp_indices, sp_values, sp_shape,
8171 func SparseDenseCwiseMul(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
8178 sp_indices, sp_values, sp_shape, dense,
11366 func SparseDenseCwiseDiv(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
11373 sp_indices, sp_values, sp_shape, dense,
11542 func SparseDenseCwiseAdd(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
11549 sp_indices, sp_values, sp_shape, dense,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 56884 name: "sp_indices"
56929 name: "sp_indices"
56976 name: "sp_indices"
57024 name: "sp_indices"
57072 name: "sp_indices"
57117 name: "sp_indices"
57164 name: "sp_indices"
57212 name: "sp_indices"
57260 name: "sp_indices"
57305 name: "sp_indices"
[all …]
|
| D | ops_history.v0.pbtxt | 33668 name: "sp_indices"
33713 name: "sp_indices"
33758 name: "sp_indices"
34531 name: "sp_indices"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 25857 name: "sp_indices"
25905 name: "sp_indices"
25953 name: "sp_indices"
26862 name: "sp_indices"
|