Home
last modified time | relevance | path

Searched refs:sp_indices (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstring_split_op.cc222 auto sp_indices = sp_indices_t->matrix<int64>(); in Compute() local
230 sp_indices(c, 0) = i; in Compute()
231 sp_indices(c, 1) = j; in Compute()
292 auto sp_indices = sp_indices_t->matrix<int64>(); in Compute() local
300 sp_indices(c, 0) = i; in Compute()
301 sp_indices(c, 1) = j; in Compute()
Dsparse_dense_binary_op_shared_test.cc212 static Graph* SparseMatCMulDenseMat(Graph* g, Node* sp_indices, Node* sp_vals, in SparseMatCMulDenseMat() argument
217 .Input(sp_indices) in SparseMatCMulDenseMat()
/external/tensorflow/tensorflow/python/ops/
Dsparse_grad.py101 sp_indices = op.inputs[0]
103 return (None, array_ops.gather_nd(out_grad, sp_indices), None, out_grad)
109 sp_indices = op.inputs[0]
115 return (None, array_ops.gather_nd(out_grad_reshaped, sp_indices // scale),
Dmath_ops.py934 def _sparse_dense_truediv(sp_indices, sp_values, sp_shape, y, name=None): argument
937 [sp_indices, sp_values, sp_shape, y]) as name:
953 sp_indices, sp_values, sp_shape, y, name=name)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseDenseCwiseDiv.pbtxt4 name: "sp_indices"
13 1-D. `N` non-empty values corresponding to `sp_indices`.
Dapi_def_SparseDenseCwiseAdd.pbtxt4 name: "sp_indices"
13 1-D. `N` non-empty values corresponding to `sp_indices`.
Dapi_def_SparseDenseCwiseMul.pbtxt4 name: "sp_indices"
13 1-D. `N` non-empty values corresponding to `sp_indices`.
Dapi_def_SparseSoftmax.pbtxt4 name: "sp_indices"
13 1-D. `NNZ` non-empty values corresponding to `sp_indices`.
/external/tensorflow/tensorflow/contrib/losses/python/metric_learning/
Dmetric_loss_ops_test.py203 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/
Dsession_test.py699 sp_indices = array_ops.identity(sp.indices)
702 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
705 [sp_indices, sp_values, sp_shape], {
718 [sp_indices, sp_values, sp_shape], {
745 sp_indices = array_ops.identity(sp.indices)
748 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
751 [sp_indices, sp_values, sp_shape], {
759 [sp_indices, sp_values, sp_shape], {
780 sp_indices = array_ops.identity(sp.indices)
783 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
[all …]
/external/tensorflow/tensorflow/contrib/lookup/
Dlookup_ops_test.py171 sp_indices = [[0, 0], [0, 1], [1, 0]]
174 constant_op.constant(sp_indices, dtypes.int64),
182 self.assertAllEqual(sp_indices, out_indices)
/external/tensorflow/tensorflow/python/kernel_tests/
Dlookup_ops_test.py191 sp_indices = [[0, 0], [0, 1], [1, 0]]
194 constant_op.constant(sp_indices, dtypes.int64),
202 self.assertAllEqual(sp_indices, out_indices)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt3477 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3481 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3485 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3565 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'name\'], varargs=None, keywords=Non…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt3477 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3481 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3485 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3565 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'name\'], varargs=None, keywords=Non…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go12839 func SparseDenseCwiseMul(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
12846 sp_indices, sp_values, sp_shape, dense,
14286 func SparseDenseCwiseAdd(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
14293 sp_indices, sp_values, sp_shape, dense,
17269 func SparseSoftmax(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Output) (ou…
17276 sp_indices, sp_values, sp_shape,
23931 func SparseDenseCwiseDiv(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
23938 sp_indices, sp_values, sp_shape, dense,
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt75545 name: "sp_indices"
75590 name: "sp_indices"
75637 name: "sp_indices"
75685 name: "sp_indices"
75733 name: "sp_indices"
75778 name: "sp_indices"
75825 name: "sp_indices"
75873 name: "sp_indices"
75921 name: "sp_indices"
75966 name: "sp_indices"
[all …]
Dops_history.v1.pbtxt76172 name: "sp_indices"
76217 name: "sp_indices"
76264 name: "sp_indices"
76312 name: "sp_indices"
76360 name: "sp_indices"
76405 name: "sp_indices"
76452 name: "sp_indices"
76500 name: "sp_indices"
76548 name: "sp_indices"
76593 name: "sp_indices"
[all …]
Dops_history.v0.pbtxt33668 name: "sp_indices"
33713 name: "sp_indices"
33758 name: "sp_indices"
34531 name: "sp_indices"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt36369 name: "sp_indices"
36417 name: "sp_indices"
36465 name: "sp_indices"
37422 name: "sp_indices"