| /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
35 OP_REQUIRES_OK(ctx, ctx->input("b_indices", &b_indices)); in Compute()
40 TensorShapeUtils::IsMatrix(b_indices->shape()) && in Compute()
45 b_indices->shape().DebugString(), " and ", in Compute()
54 a_indices->dim_size(1) == b_indices->dim_size(1) && in Compute()
55 b_indices->dim_size(1) == sum_indices->dim_size(1), in Compute()
58 a_indices->dim_size(1), b_indices->dim_size(1), in Compute()
69 const int64 b_nnz = b_indices->dim_size(0); in Compute()
73 const auto b_indices_mat = b_indices->matrix<int64>(); in Compute()
|
| D | sparse_add_op.cc | 32 const Tensor *a_indices, *b_indices, *a_values_t, *b_values_t, *a_shape, in Compute() local
36 OP_REQUIRES_OK(ctx, ctx->input("b_indices", &b_indices)); in Compute()
39 TensorShapeUtils::IsMatrix(b_indices->shape()), in Compute()
43 b_indices->shape().DebugString())); in Compute()
45 const int64 b_nnz = b_indices->dim_size(0); in Compute()
98 auto b_indices_mat = b_indices->matrix<int64>(); in Compute()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 85 b_indices = op.inputs[3]
92 val_grad, a_indices, b_indices, sum_indices)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseSparseMinimum.pbtxt | 23 name: "b_indices"
|
| D | api_def_SparseSparseMaximum.pbtxt | 23 name: "b_indices"
|
| D | api_def_SparseAddGrad.pbtxt | 17 name: "b_indices"
|
| D | api_def_SparseAdd.pbtxt | 22 name: "b_indices"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 54 ShapeHandle b_indices; in __anone6e195410202() local
56 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &b_indices)); in __anone6e195410202()
58 c->set_output(1, c->Vector(c->Dim(b_indices, 0))); in __anone6e195410202()
|
| D | ops.pbtxt | 35183 name: "b_indices"
35267 name: "b_indices"
37507 name: "b_indices"
37562 name: "b_indices"
|
| /external/tensorflow/tensorflow/contrib/crf/python/ops/ |
| D | crf.py | 510 b_indices = math_ops.range(batch_size) # [B]
511 indices = array_ops.stack([b_indices, state], axis=1) # [B, 2]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3417 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3421 …argspec: "args=[\'backprop_val_grad\', \'a_indices\', \'b_indices\', \'sum_indices\', \'name\'], v…
3573 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3577 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3417 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3421 …argspec: "args=[\'backprop_val_grad\', \'a_indices\', \'b_indices\', \'sum_indices\', \'name\'], v…
3573 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3577 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 13991 func SparseAdd(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, b_indices …
13998 a_indices, a_values, a_shape, b_indices, b_values, b_shape, thresh,
14101 func SparseAddGrad(scope *Scope, backprop_val_grad tf.Output, a_indices tf.Output, b_indices tf.Out…
14108 backprop_val_grad, a_indices, b_indices, sum_indices,
15292 …ope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, b_indices tf.Output, b_val…
15299 a_indices, a_values, a_shape, b_indices, b_values, b_shape,
28165 …ope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, b_indices tf.Output, b_val…
28172 a_indices, a_values, a_shape, b_indices, b_values, b_shape,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 71795 name: "b_indices"
71877 name: "b_indices"
71963 name: "b_indices"
72051 name: "b_indices"
72135 name: "b_indices"
72184 name: "b_indices"
72235 name: "b_indices"
72287 name: "b_indices"
78656 name: "b_indices"
78708 name: "b_indices"
[all …]
|
| D | ops_history.v2.pbtxt | 71168 name: "b_indices"
71250 name: "b_indices"
71336 name: "b_indices"
71424 name: "b_indices"
71508 name: "b_indices"
71557 name: "b_indices"
71608 name: "b_indices"
71660 name: "b_indices"
78029 name: "b_indices"
78081 name: "b_indices"
[all …]
|
| D | ops_history.v0.pbtxt | 31946 name: "b_indices"
32026 name: "b_indices"
32104 name: "b_indices"
34686 name: "b_indices"
34738 name: "b_indices"
|