| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_add_grad_op.cc | 32 const Tensor *backprop_val_grad, *a_indices, *b_indices, *sum_indices; in Compute() local
34 OP_REQUIRES_OK(ctx, ctx->input("a_indices", &a_indices)); in Compute()
39 TensorShapeUtils::IsMatrix(a_indices->shape()) && in Compute()
44 a_indices->shape().DebugString(), " and ", in Compute()
54 a_indices->dim_size(1) == b_indices->dim_size(1) && in Compute()
58 a_indices->dim_size(1), b_indices->dim_size(1), in Compute()
67 const int num_dims = a_indices->dim_size(1); in Compute()
68 const int64 a_nnz = a_indices->dim_size(0); in Compute()
72 const auto a_indices_mat = a_indices->matrix<int64>(); in Compute()
|
| D | sparse_tensor_dense_matmul_op.cc | 43 const Tensor* a_indices; in Compute() local
47 OP_REQUIRES_OK(ctx, ctx->input("a_indices", &a_indices)); in Compute()
66 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(a_indices->shape()), in Compute()
69 const int64 nnz = a_indices->shape().dim_size(0); in Compute()
75 ctx, a_indices->shape().dim_size(1) == a_shape->NumElements(), in Compute()
141 a_indices->matrix<Tindices>(), a_values->vec<T>(), \ in Compute()
187 TTypes<Tindices>::ConstMatrix a_indices, \
250 typename TTypes<Tindices>::ConstMatrix a_indices, in Compute()
270 const Tindices m = internal::SubtleMustCopy(a_indices(i, lhs_index_a)); in Compute()
271 const Tindices k = internal::SubtleMustCopy(a_indices(i, rhs_index_a)); in Compute()
[all …]
|
| D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 32 const Tindices* __restrict__ a_indices, const T* __restrict__ a_values, in SparseTensorDenseMatMulKernel() argument
40 const int i = ldg(a_indices + 2 * a_ix + ((ADJ_A) ? 1 : 0)); in SparseTensorDenseMatMulKernel()
41 const int k = ldg(a_indices + 2 * a_ix + ((ADJ_A) ? 0 : 1)); in SparseTensorDenseMatMulKernel()
67 typename TTypes<Tindices>::ConstMatrix a_indices, in Compute()
85 b_rows, b_cols, p, a_indices.data(), a_values.data(), b.data(), in Compute()
|
| D | sparse_tensor_dense_matmul_op_test.cc | 26 Node* SparseTensorDenseMatMulNode(Graph* g, Node* a_indices, Node* a_values, in SparseTensorDenseMatMulNode() argument
31 .Input(a_indices) in SparseTensorDenseMatMulNode()
46 Tensor a_indices(DT_INT64, TensorShape({nnz, 2})); in SparseTensorDenseMatmul() local
52 auto a_indices_t = a_indices.matrix<int64>(); in SparseTensorDenseMatmul()
65 g, test::graph::Constant(g, a_indices), in SparseTensorDenseMatmul()
|
| D | sparse_add_op.cc | 32 const Tensor *a_indices, *b_indices, *a_values_t, *b_values_t, *a_shape, in Compute() local
35 OP_REQUIRES_OK(ctx, ctx->input("a_indices", &a_indices)); in Compute()
38 TensorShapeUtils::IsMatrix(a_indices->shape()) && in Compute()
42 a_indices->shape().DebugString(), " and ", in Compute()
44 const int64 a_nnz = a_indices->dim_size(0); in Compute()
97 auto a_indices_mat = a_indices->matrix<int64>(); in Compute()
|
| D | sparse_tensor_dense_add_op.cc | 35 Status ValidateInputs(const Tensor *a_indices, const Tensor *a_values, in ValidateInputs() argument
37 if (!TensorShapeUtils::IsMatrix(a_indices->shape())) { in ValidateInputs()
40 a_indices->shape().DebugString()); in ValidateInputs()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 84 a_indices = op.inputs[0]
92 val_grad, a_indices, b_indices, sum_indices)
166 a_indices, a_values, a_shape = op.inputs[:3]
182 a_indices, a_values, a_shape, grad, adjoint_a=not adj_a)
187 rows = a_indices[:, 0]
188 cols = a_indices[:, 1]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseSparseMinimum.pbtxt | 4 name: "a_indices"
13 1-D. `N` non-empty values corresponding to `a_indices`.
25 counterpart to `a_indices` for the other operand.
|
| D | api_def_SparseSparseMaximum.pbtxt | 4 name: "a_indices"
13 1-D. `N` non-empty values corresponding to `a_indices`.
25 counterpart to `a_indices` for the other operand.
|
| D | api_def_SparseTensorDenseAdd.pbtxt | 4 name: "a_indices"
29 This Op does not require `a_indices` be sorted in standard lexicographic order.
|
| D | api_def_SparseMatrixOrderingAMD.pbtxt | 42 a_indices = np.array([[0, 0], [1, 1], [2, 1], [2, 2], [3, 3]])
48 a_st = tf.sparse.SparseTensor(a_indices, a_values, a_dense_shape)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SparseSparseMaximum.pbtxt | 4 name: "a_indices"
56 name: "a_indices"
110 name: "a_indices"
165 name: "a_indices"
|
| D | SparseAddGrad.pbtxt | 8 name: "a_indices"
57 name: "a_indices"
108 name: "a_indices"
160 name: "a_indices"
|
| D | SparseTensorDenseAdd.pbtxt | 4 name: "a_indices"
59 name: "a_indices"
116 name: "a_indices"
174 name: "a_indices"
|
| D | SparseSparseMinimum.pbtxt | 4 name: "a_indices"
61 name: "a_indices"
120 name: "a_indices"
180 name: "a_indices"
|
| D | SparseTensorDenseMatMul.pbtxt | 4 name: "a_indices"
45 name: "a_indices"
|
| D | SparseAdd.pbtxt | 4 name: "a_indices"
86 name: "a_indices"
172 name: "a_indices"
260 name: "a_indices"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseSparseMaximum.pbtxt | 4 name: "a_indices"
56 name: "a_indices"
110 name: "a_indices"
165 name: "a_indices"
|
| D | SparseAddGrad.pbtxt | 8 name: "a_indices"
57 name: "a_indices"
108 name: "a_indices"
160 name: "a_indices"
|
| D | SparseTensorDenseAdd.pbtxt | 4 name: "a_indices"
59 name: "a_indices"
116 name: "a_indices"
174 name: "a_indices"
|
| D | SparseSparseMinimum.pbtxt | 4 name: "a_indices"
61 name: "a_indices"
120 name: "a_indices"
180 name: "a_indices"
|
| D | SparseTensorDenseMatMul.pbtxt | 4 name: "a_indices"
45 name: "a_indices"
|
| D | SparseAdd.pbtxt | 4 name: "a_indices"
86 name: "a_indices"
172 name: "a_indices"
260 name: "a_indices"
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_ops_test.py | 107 a_indices = np.array([[0, 0], [2, 3]])
111 (a_indices[:, 0], a_indices[:, 1])),
142 a_indices = np.array([[0, 0], [2, 3], [2, 4], [3, 0]])
146 (a_indices[:, 0], a_indices[:, 1])),
151 a_st = sparse_tensor.SparseTensor(a_indices, a_values, a_dense_shape)
173 self.assertAllEqual(a_indices, a_st_rt_value.indices)
207 a_indices = np.array([[0, 0, 0], [0, 2, 3], [2, 0, 1]])
219 a_st = sparse_tensor.SparseTensor(a_indices, a_values, a_dense_shape)
243 self.assertAllEqual(a_indices, a_st_rt_value.indices)
309 a_indices = np.array([[0, 0], [2, 3], [2, 4], [3, 0]])
[all …]
|
| D | csr_sparse_matrix_test.py | 44 a_indices = np.array([[0, 0], [2, 3], [2, 4], [3, 0]])
48 a_st = sparse_tensor.SparseTensor(a_indices, a_values, a_dense_shape)
56 self.assertAllEqual(a_indices, a_st_rt.indices)
|