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