OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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