HomeSort by relevance Sort by last modified time
    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