| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_add_grad_op.cc | 32 const Tensor *backprop_val_grad, *a_indices, *b_indices, *sum_indices; variable
34 OP_REQUIRES_OK(ctx, ctx->input("a_indices", &a_indices));
39 TensorShapeUtils::IsMatrix(a_indices->shape()) &&
44 a_indices->shape().DebugString(), " and ",
54 a_indices->dim_size(1) == b_indices->dim_size(1) &&
58 a_indices->dim_size(1), b_indices->dim_size(1),
67 const int num_dims = a_indices->dim_size(1);
68 const int64 a_nnz = a_indices->dim_size(0);
72 const auto a_indices_mat = a_indices->matrix<int64>()
[all...] |
| sparse_tensor_dense_matmul_op_gpu.cu.cc | 33 const Tindices* a_indices,
42 const int i = ldg(a_indices + 2 * a_ix + ((ADJ_A) ? 1 : 0));
43 const int k = ldg(a_indices + 2 * a_ix + ((ADJ_A) ? 0 : 1));
69 typename TTypes<Tindices>::ConstMatrix a_indices,
86 nnz, m, b_rows, b_cols, p, a_indices.data(), a_values.data(),
|
| sparse_tensor_dense_matmul_op_test.cc | 26 Node* SparseTensorDenseMatMulNode(Graph* g, Node* a_indices, Node* a_values,
31 .Input(a_indices)
46 Tensor a_indices(DT_INT64, TensorShape({nnz, 2}));
52 auto a_indices_t = a_indices.matrix<int64>();
65 g, test::graph::Constant(g, a_indices),
|
| sparse_tensor_dense_matmul_op.cc | 42 const Tensor* a_indices; variable
46 OP_REQUIRES_OK(ctx, ctx->input("a_indices", &a_indices));
65 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(a_indices->shape()),
66 errors::InvalidArgument("Tensor 'a_indices' is not a matrix"));
68 const int64 nnz = a_indices->shape().dim_size(0);
70 errors::InvalidArgument("Number of rows of a_indices does not "
74 ctx, a_indices->shape().dim_size(1) == a_shape->NumElements(),
75 errors::InvalidArgument("Number of columns of a_indices does not match "
140 a_indices->matrix<Tindices>(), a_values->vec<T>(),
[all...] |
| sparse_add_op.cc | 32 const Tensor *a_indices, *b_indices, *a_values_t, *b_values_t, *a_shape, variable
35 OP_REQUIRES_OK(ctx, ctx->input("a_indices", &a_indices));
38 TensorShapeUtils::IsMatrix(a_indices->shape()) &&
42 a_indices->shape().DebugString(), " and ",
44 const int64 a_nnz = a_indices->dim_size(0);
97 auto a_indices_mat = a_indices->matrix<int64>();
|
| sparse_tensor_dense_add_op.cc | 35 Status ValidateInputs(const Tensor *a_indices, const Tensor *a_values,
37 if (!TensorShapeUtils::IsMatrix(a_indices->shape())) {
39 "Input a_indices should be a matrix but received shape: ",
40 a_indices->shape().DebugString());
76 OP_REQUIRES_OK(ctx, ctx->input("a_indices", &a_indices_t));
|
| sparse_tensor_dense_matmul_op.h | 33 typename TTypes<Tindices>::ConstMatrix a_indices,
|
| /external/tensorflow/tensorflow/python/ops/ |
| sparse_grad.py | 80 (a_indices, a_values, a_shape, b_indices, b_values, b_shape, thresh)
84 a_indices = op.inputs[0]
92 a_val_grad, b_val_grad = gen_sparse_ops._sparse_add_grad(val_grad, a_indices,
97 # (a_indices, a_values, a_shape, b_indices, b_values, b_shape, thresh)
139 a_indices, a_values, a_shape = op.inputs[:3]
155 a_indices, a_values, a_shape, grad, adjoint_a=not adj_a)
160 rows = a_indices[:, 0]
161 cols = a_indices[:, 1]
173 # gradients w.r.t. (a_indices, a_values, a_shape, b)
|
| sparse_ops.py | [all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| sparse_ops.cc | 30 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &unused)); // a_indices
46 .Input("a_indices: int64")
53 ShapeHandle a_indices;
55 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &a_indices));
57 c->set_output(0, c->Vector(c->Dim(a_indices, 0)));
63 .Input("a_indices: int64")
86 .Input("a_indices: Tindices")
101 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &unused)); // a_indices
370 .Input("a_indices: Tindices")
465 .Input("a_indices: int64"
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
