| /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 | 42 const Tensor* a_indices; in Compute() local
46 OP_REQUIRES_OK(ctx, ctx->input("a_indices", &a_indices)); in Compute()
65 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(a_indices->shape()), in Compute()
68 const int64 nnz = a_indices->shape().dim_size(0); in Compute()
74 ctx, a_indices->shape().dim_size(1) == a_shape->NumElements(), in Compute()
140 a_indices->matrix<Tindices>(), a_values->vec<T>(), \ in Compute()
185 TTypes<Tindices>::ConstMatrix a_indices, \
248 typename TTypes<Tindices>::ConstMatrix a_indices, in Compute()
268 const Tindices m = internal::SubtleMustCopy(a_indices(i, lhs_index_a)); in Compute()
269 const Tindices k = internal::SubtleMustCopy(a_indices(i, rhs_index_a)); in Compute()
[all …]
|
| D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 33 const Tindices* a_indices, in SparseTensorDenseMatMulKernel() argument
42 const int i = ldg(a_indices + 2 * a_ix + ((ADJ_A) ? 1 : 0)); in SparseTensorDenseMatMulKernel()
43 const int k = ldg(a_indices + 2 * a_ix + ((ADJ_A) ? 0 : 1)); in SparseTensorDenseMatMulKernel()
69 typename TTypes<Tindices>::ConstMatrix a_indices, in Compute()
87 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()
|
| D | sparse_tensor_dense_matmul_op.h | 33 typename TTypes<Tindices>::ConstMatrix a_indices,
|
| /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]
|
| D | sparse_ops.py | 2340 a_indices=sp_a.indices,
|
| /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_SparseAddGrad.pbtxt | 11 name: "a_indices"
|
| D | api_def_SparseTensorDenseMatMul.pbtxt | 4 name: "a_indices"
|
| D | api_def_SparseAdd.pbtxt | 4 name: "a_indices"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 53 ShapeHandle a_indices; in __anone6e195410202() local
55 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &a_indices)); in __anone6e195410202()
57 c->set_output(0, c->Vector(c->Dim(a_indices, 0))); in __anone6e195410202()
|
| D | ops.pbtxt | 35171 name: "a_indices"
35263 name: "a_indices"
37495 name: "a_indices"
37550 name: "a_indices"
37654 name: "a_indices"
37712 name: "a_indices"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3417 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3421 …argspec: "args=[\'backprop_val_grad\', \'a_indices\', \'b_indices\', \'sum_indices\', \'name\'], v…
3573 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3577 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3585 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b\', \'name\'], varargs=None, keywords…
3589 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b\', \'adjoint_a\', \'adjoint_b\', \'n…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3417 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3421 …argspec: "args=[\'backprop_val_grad\', \'a_indices\', \'b_indices\', \'sum_indices\', \'name\'], v…
3573 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3577 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3585 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b\', \'name\'], varargs=None, keywords…
3589 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b\', \'adjoint_a\', \'adjoint_b\', \'n…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 12996 func SparseTensorDenseAdd(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output,…
13003 a_indices, a_values, a_shape, b,
13947 func SparseTensorDenseMatMul(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Outp…
13958 a_indices, a_values, a_shape, b,
13991 func SparseAdd(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, b_indices …
13998 a_indices, a_values, a_shape, b_indices, b_values, b_shape, thresh,
14101 func SparseAddGrad(scope *Scope, backprop_val_grad tf.Output, a_indices tf.Output, b_indices tf.Out…
14108 backprop_val_grad, a_indices, b_indices, sum_indices,
15292 func SparseSparseMinimum(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, …
15299 a_indices, a_values, a_shape, b_indices, b_values, b_shape,
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 71783 name: "a_indices"
71865 name: "a_indices"
71951 name: "a_indices"
72039 name: "a_indices"
72131 name: "a_indices"
72180 name: "a_indices"
72231 name: "a_indices"
72283 name: "a_indices"
78644 name: "a_indices"
78696 name: "a_indices"
[all …]
|
| D | ops_history.v2.pbtxt | 71156 name: "a_indices"
71238 name: "a_indices"
71324 name: "a_indices"
71412 name: "a_indices"
71504 name: "a_indices"
71553 name: "a_indices"
71604 name: "a_indices"
71656 name: "a_indices"
78017 name: "a_indices"
78069 name: "a_indices"
[all …]
|
| D | ops_history.v0.pbtxt | 31934 name: "a_indices"
32014 name: "a_indices"
32100 name: "a_indices"
34674 name: "a_indices"
34726 name: "a_indices"
34827 name: "a_indices"
34882 name: "a_indices"
|