HomeSort by relevance Sort by last modified time
    Searched refs:grad_idx (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
sparse_conditional_accumulator.h 161 const Tensor* grad_idx = std::get<0>(*grad); variable
164 const int64 nnz = grad_idx->dim_size(0);
171 accum_idx_vec_->push_back(grad_idx->vec<int64>()(i));
197 const Tensor* grad_idx = std::get<0>(*grad); variable
201 const int64 grad_nnz = grad_idx->dim_size(0);
219 switch (cmp(accum_idx_vec_, grad_idx, i, j)) {
297 sum_indices_vec->push_back(grad_idx->vec<int64>()(idx_b));
  /external/tensorflow/tensorflow/core/ops/
training_ops.cc 38 int grad_idx, ShapeHandle* s) {
39 ShapeHandle grad = ShapeOrHandleShape(c, grad_idx);
46 TF_RETURN_IF_ERROR(c->WithRank(c->input(grad_idx + 1), 1, &indices));
95 HandleGradAndIndicesInputs(c, sparse, 4 /* grad_idx */, &s));
166 HandleGradAndIndicesInputs(c, sparse, 6 /* grad_idx */, &s));
241 HandleGradAndIndicesInputs(c, sparse, 3 /* grad_idx */, &s));
279 HandleGradAndIndicesInputs(c, sparse, 5 /* grad_idx */, &s));
348 HandleGradAndIndicesInputs(c, sparse, 3 /* grad_idx */, &s));
463 HandleGradAndIndicesInputs(c, sparse, 3 /* grad_idx */, &s));
617 HandleGradAndIndicesInputs(c, sparse, 3 /* grad_idx */, &s))
    [all...]

Completed in 551 milliseconds