Searched refs:sum_indices (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_add_grad_op.cc | 32 const Tensor *backprop_val_grad, *a_indices, *b_indices, *sum_indices; in Compute() local
36 OP_REQUIRES_OK(ctx, ctx->input("sum_indices", &sum_indices)); in Compute()
41 TensorShapeUtils::IsMatrix(sum_indices->shape()), in Compute()
46 sum_indices->shape().DebugString())); in Compute()
55 b_indices->dim_size(1) == sum_indices->dim_size(1), in Compute()
59 sum_indices->dim_size(1))); in Compute()
61 ctx, backprop_val_grad->NumElements() == sum_indices->dim_size(0), in Compute()
65 sum_indices->dim_size(0))); in Compute()
74 const auto sum_indices_mat = sum_indices->matrix<int64>(); in Compute()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 86 sum_indices = op.outputs[0]
92 val_grad, a_indices, b_indices, sum_indices)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseAddGrad.pbtxt | 23 name: "sum_indices"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 13991 …s tf.Output, b_values tf.Output, b_shape tf.Output, thresh tf.Output) (sum_indices tf.Output, sum_…
14101 …backprop_val_grad tf.Output, a_indices tf.Output, b_indices tf.Output, sum_indices tf.Output) (a_v…
14108 backprop_val_grad, a_indices, b_indices, sum_indices,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 71811 name: "sum_indices"
71893 name: "sum_indices"
71979 name: "sum_indices"
72067 name: "sum_indices"
72139 name: "sum_indices"
72188 name: "sum_indices"
72239 name: "sum_indices"
72291 name: "sum_indices"
|
| D | ops_history.v2.pbtxt | 71184 name: "sum_indices"
71266 name: "sum_indices"
71352 name: "sum_indices"
71440 name: "sum_indices"
71512 name: "sum_indices"
71561 name: "sum_indices"
71612 name: "sum_indices"
71664 name: "sum_indices"
|
| D | ops_history.v0.pbtxt | 31962 name: "sum_indices"
32042 name: "sum_indices"
32108 name: "sum_indices"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3421 …argspec: "args=[\'backprop_val_grad\', \'a_indices\', \'b_indices\', \'sum_indices\', \'name\'], v…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3421 …argspec: "args=[\'backprop_val_grad\', \'a_indices\', \'b_indices\', \'sum_indices\', \'name\'], v…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 35199 name: "sum_indices"
35271 name: "sum_indices"
|