OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:sum_nnz
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
sparse_add_grad_op.cc
70
const int64
sum_nnz
= backprop_val_grad->NumElements();
variable
107
while (i < a_nnz && j < b_nnz && k <
sum_nnz
) {
117
while (i < a_nnz && k <
sum_nnz
) {
122
while (j < b_nnz && k <
sum_nnz
) {
sparse_sparse_binary_op_shared.cc
193
const int64
sum_nnz
= a_augmented_values.size();
variable
196
ctx->allocate_output(0, TensorShape({
sum_nnz
, num_dims}),
199
ctx, ctx->allocate_output(1, TensorShape({
sum_nnz
}), &output_values_t));
202
for (int64 i = 0; i <
sum_nnz
; ++i) {
220
UnalignedTensorMap(a_augmented_values.data(),
sum_nnz
);
222
UnalignedTensorMap(b_augmented_values.data(),
sum_nnz
);
sparse_add_op.cc
146
const int64
sum_nnz
= out_values.size();
variable
149
ctx->allocate_output(0, TensorShape({
sum_nnz
, num_dims}),
152
ctx, ctx->allocate_output(1, TensorShape({
sum_nnz
}), &out_values_t));
156
for (i = 0; i <
sum_nnz
; ++i) {
162
std::copy_n(out_values.begin(),
sum_nnz
, &out_values_flat(0));
sparse_conditional_accumulator.h
216
int64
sum_nnz
= 0;
variable
218
sum_nnz
++;
238
sum_nnz
++;
243
sum_nnz
++;
250
sum_indices_vec->reserve(
sum_nnz
);
253
sum_counts->reserve(
sum_nnz
);
259
sum_shape.set_dim(0,
sum_nnz
);
272
for (i = 0; i <
sum_nnz
; ++i) {
Completed in 760 milliseconds