/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() 86 nnz, m, b_rows, b_cols, p, a_indices.data(), a_values.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 a_val_grad, b_val_grad = gen_sparse_ops._sparse_add_grad(val_grad, a_indices, 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]
|
D | sparse_ops.py | 1832 a_indices=sp_a.indices,
|
/external/tensorflow/tensorflow/core/api_def/base_api/ |
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_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_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 __anon4f27251e0202() local 55 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &a_indices)); in __anon4f27251e0202() 57 c->set_output(0, c->Vector(c->Dim(a_indices, 0))); in __anon4f27251e0202()
|
D | ops.pbtxt | 24679 name: "a_indices" 24771 name: "a_indices" 26935 name: "a_indices" 26990 name: "a_indices" 27094 name: "a_indices" 27152 name: "a_indices"
|
/external/tensorflow/tensorflow/go/op/ |
D | wrappers.go | 6968 func SparseAddGrad(scope *Scope, backprop_val_grad tf.Output, a_indices tf.Output, b_indices tf.Out… 6975 backprop_val_grad, a_indices, b_indices, sum_indices, 12463 func SparseTensorDenseMatMul(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Outp… 12474 a_indices, a_values, a_shape, b, 12857 func SparseTensorDenseAdd(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output,… 12864 a_indices, a_values, a_shape, b, 13017 func SparseAdd(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, b_indices … 13024 a_indices, a_values, a_shape, b_indices, b_values, b_shape, thresh, 13305 func SparseSparseMinimum(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, … 13312 a_indices, a_values, a_shape, b_indices, b_values, b_shape, [all …]
|
/external/tensorflow/tensorflow/core/ops/compat/ |
D | ops_history.v1.pbtxt | 52639 name: "a_indices" 52721 name: "a_indices" 52807 name: "a_indices" 52895 name: "a_indices" 52987 name: "a_indices" 53036 name: "a_indices" 53087 name: "a_indices" 53139 name: "a_indices" 59308 name: "a_indices" 59360 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"
|