HomeSort by relevance Sort by last modified time
    Searched defs:a_indices (Results 1 - 3 of 3) sorted by null

  /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_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_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...]

Completed in 152 milliseconds