Searched refs:reverse_index_map (Results 1 – 8 of 8) 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/python/ops/ |
| D | sparse_grad.py | 279 reverse_index_map = op.outputs[3]
283 reverse_index_map=reverse_index_map, grad_values=output_grad_values)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 576 ShapeHandle reverse_index_map = c->Vector(N); in __anon4f27251e1502() local
580 c->set_output(3, reverse_index_map); in __anon4f27251e1502()
591 ShapeHandle reverse_index_map = c->input(0); in __anon4f27251e1602() local
592 TF_RETURN_IF_ERROR(c->WithRank(reverse_index_map, 1, &reverse_index_map)); in __anon4f27251e1602()
595 c->set_output(0, reverse_index_map); in __anon4f27251e1602()
|
| D | ops.pbtxt | 26029 name: "reverse_index_map"
26040 name: "reverse_index_map"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 9091 …s tf.Output, output_values tf.Output, empty_row_indicator tf.Output, reverse_index_map tf.Output) {
13590 func SparseFillEmptyRowsGrad(scope *Scope, reverse_index_map tf.Output, grad_values tf.Output) (d_v…
13597 reverse_index_map, grad_values,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 57476 name: "reverse_index_map"
57487 name: "reverse_index_map"
|