| /external/tensorflow/tensorflow/core/kernels/ |
| string_split_op.cc | 222 auto sp_indices = sp_indices_t->matrix<int64>(); variable
230 sp_indices(c, 0) = i;
231 sp_indices(c, 1) = j;
292 auto sp_indices = sp_indices_t->matrix<int64>(); variable
300 sp_indices(c, 0) = i;
301 sp_indices(c, 1) = j;
|
| sparse_dense_binary_op_shared_test.cc | 212 static Graph* SparseMatCMulDenseMat(Graph* g, Node* sp_indices, Node* sp_vals,
217 .Input(sp_indices)
|
| /external/tensorflow/tensorflow/python/ops/ |
| sparse_grad.py | 101 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),
240 # (sp_indices, sp_vals, sp_shape, dense)
271 Gradients w.r.t. the input (sp_indices, sp_values, sp_shape).
|
| math_ops.py | 851 If `clazz_object` is `SparseTensor`, assumes `func` takes `(sp_indices,
934 def _sparse_dense_truediv(sp_indices, sp_values, sp_shape, y, name=None):
937 [sp_indices, sp_values, sp_shape, y]) as name:
953 sp_indices, sp_values, sp_shape, y, name=name)
[all...] |
| /external/tensorflow/tensorflow/contrib/losses/python/metric_learning/ |
| 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/ |
| session_test.py | 699 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/ |
| lookup_ops_test.py | 171 sp_indices = [[0, 0], [0, 1], [1, 0]]
174 constant_op.constant(sp_indices, dtypes.int64),
182 self.assertAllEqual(sp_indices, out_indices)
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| lookup_ops_test.py | 191 sp_indices = [[0, 0], [0, 1], [1, 0]]
194 constant_op.constant(sp_indices, dtypes.int64),
202 self.assertAllEqual(sp_indices, out_indices)
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
