Searched refs:reverse_index_map (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseFillEmptyRowsGrad.pbtxt | 4 name: "reverse_index_map"
29 Takes vectors reverse_index_map, shaped `[N]`, and grad_values,
34 d_values[j] = grad_values[reverse_index_map[j]]
36 grad_values[k] * 1{k not in reverse_index_map})
|
| D | api_def_SparseFillEmptyRows.pbtxt | 43 name: "reverse_index_map"
85 reverse_index_map[j] = out_j s.t. indices[j, :] == output_indices[out_j, :]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_fill_empty_rows_op.cc | 87 auto reverse_index_map = reverse_index_map_t->vec<int64>(); in Compute() local
171 reverse_index_map(i) = output_i; in Compute()
217 const auto reverse_index_map = reverse_index_map_t->vec<int64>(); in Compute() local
244 d_values(i) = grad_values(reverse_index_map(i)); in Compute()
245 visited(reverse_index_map(i)) = true; in Compute()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 589 ShapeHandle reverse_index_map = c->Vector(N); in __anone6e195411602() local
593 c->set_output(3, reverse_index_map); in __anone6e195411602()
604 ShapeHandle reverse_index_map = c->input(0); in __anone6e195411702() local
605 TF_RETURN_IF_ERROR(c->WithRank(reverse_index_map, 1, &reverse_index_map)); in __anone6e195411702()
608 c->set_output(0, reverse_index_map); in __anone6e195411702()
|
| D | ops.pbtxt | 36541 name: "reverse_index_map"
36552 name: "reverse_index_map"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 306 reverse_index_map = op.outputs[3]
309 reverse_index_map=reverse_index_map, grad_values=output_grad_values)
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 9588 …s tf.Output, output_values tf.Output, empty_row_indicator tf.Output, reverse_index_map tf.Output) {
26954 func SparseFillEmptyRowsGrad(scope *Scope, reverse_index_map tf.Output, grad_values tf.Output) (d_v…
26961 reverse_index_map, grad_values,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3493 …argspec: "args=[\'reverse_index_map\', \'grad_values\', \'name\'], varargs=None, keywords=None, de…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3493 …argspec: "args=[\'reverse_index_map\', \'grad_values\', \'name\'], varargs=None, keywords=None, de…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 76764 name: "reverse_index_map"
76775 name: "reverse_index_map"
|
| D | ops_history.v2.pbtxt | 76137 name: "reverse_index_map"
76148 name: "reverse_index_map"
|