OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:entries_to_copy
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
sparse_sparse_binary_op_shared.cc
56
// "
entries_to_copy
" with "(from_a?, index)" pairs. All three vectors have the
68
std::vector<std::pair<bool, int64>> *
entries_to_copy
) {
69
entries_to_copy
->reserve(a_nnz + b_nnz);
79
entries_to_copy
->emplace_back(true, i);
85
entries_to_copy
->emplace_back(true, i);
92
entries_to_copy
->emplace_back(false, j);
101
entries_to_copy
->emplace_back(/* is_a */ true, i);
106
entries_to_copy
->emplace_back(/* is_a */ false, j);
187
std::vector<std::pair<bool, int64>>
entries_to_copy
; // from_a?, idx
variable
190
&b_augmented_values, &
entries_to_copy
);
[
all
...]
sparse_add_op.cc
99
std::vector<std::pair<bool, int64>>
entries_to_copy
; // from_a?, idx
variable
100
entries_to_copy
.reserve(a_nnz + b_nnz);
112
entries_to_copy
.emplace_back(true, i);
119
entries_to_copy
.emplace_back(true, i);
126
entries_to_copy
.emplace_back(false, j);
135
entries_to_copy
.emplace_back(IS_A, IDX); \
157
const bool from_a =
entries_to_copy
[i].first;
158
const int64 idx =
entries_to_copy
[i].second;
sparse_conditional_accumulator.h
208
std::vector<std::tuple<Source, int64, int64>>
entries_to_copy
;
variable
209
entries_to_copy
.reserve(accum_nnz + grad_nnz);
221
entries_to_copy
.emplace_back(from_accum, i, -1);
225
entries_to_copy
.emplace_back(from_accum_and_grad, i, j);
230
entries_to_copy
.emplace_back(from_grad, -1, j);
239
entries_to_copy
.emplace_back(from_accum, i, -1);
244
entries_to_copy
.emplace_back(from_grad, -1, j);
273
const Source src = std::get<0>(
entries_to_copy
[i]);
274
const int64 idx_a = std::get<1>(
entries_to_copy
[i]);
275
const int64 idx_b = std::get<2>(
entries_to_copy
[i])
[
all
...]
Completed in 67 milliseconds