Searched refs:labels_indices (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_loss_op.cc | 55 void DoHistogram(OpKernelContext* ctx, const Tensor* labels_indices, in DoHistogram() argument
58 const T* h_in = labels_indices->flat<T>().data(); in DoHistogram()
89 const Tensor* labels_indices; in Compute() local
93 OP_REQUIRES_OK(ctx, ctx->input("labels_indices", &labels_indices)); in Compute()
101 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(labels_indices->shape()), in Compute()
122 OP_REQUIRES(ctx, labels_indices->dim_size(0) == labels_values->dim_size(0), in Compute()
126 labels_indices->shape().DebugString(), " vs. ", in Compute()
133 auto labels_indices_t = labels_indices->matrix<int64>(); in Compute()
135 for (int i = 0; i < labels_indices->dim_size(0); i++) { in Compute()
143 ctx, sparse::SparseTensor::Create(*labels_indices, *labels_values, in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ctc_ops.cc | 40 ShapeHandle labels_indices; in __anon7dc6794d0102() local
45 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &labels_indices)); in __anon7dc6794d0102()
50 TF_RETURN_IF_ERROR(c->Merge(c->Dim(labels_indices, 0), in __anon7dc6794d0102()
77 ShapeHandle labels_indices; in __anon7dc6794d0202() local
82 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &labels_indices)); in __anon7dc6794d0202()
87 TF_RETURN_IF_ERROR(c->Merge(c->Dim(labels_indices, 0), in __anon7dc6794d0202()
|
| D | ops.pbtxt | 6894 name: "labels_indices"
6955 name: "labels_indices"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CTCLoss.pbtxt | 8 name: "labels_indices"
49 name: "labels_indices"
97 name: "labels_indices"
|
| D | CTCLossV2.pbtxt | 8 name: "labels_indices"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CTCLoss.pbtxt | 8 name: "labels_indices"
49 name: "labels_indices"
97 name: "labels_indices"
|
| D | CTCLossV2.pbtxt | 8 name: "labels_indices"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CTCLossV2.pbtxt | 12 name: "labels_indices"
15 `labels_indices(i, :) == [b, t]` means `labels_values(i)` stores the id for
|
| D | api_def_CTCLoss.pbtxt | 10 name: "labels_indices"
13 `labels_indices(i, :) == [b, t]` means `labels_values(i)` stores the id for
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 693 …argspec: "args=[\'inputs\', \'labels_indices\', \'labels_values\', \'sequence_length\', \'preproce…
697 …argspec: "args=[\'inputs\', \'labels_indices\', \'labels_values\', \'sequence_length\', \'preproce…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 693 …argspec: "args=[\'inputs\', \'labels_indices\', \'labels_values\', \'sequence_length\', \'preproce…
697 …argspec: "args=[\'inputs\', \'labels_indices\', \'labels_values\', \'sequence_length\', \'preproce…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 4990 func CTCLoss(scope *Scope, inputs tf.Output, labels_indices tf.Output, labels_values tf.Output, seq…
5001 inputs, labels_indices, labels_values, sequence_length,
48174 func CTCLossV2(scope *Scope, inputs tf.Output, labels_indices tf.Output, labels_values tf.Output, s…
48185 inputs, labels_indices, labels_values, sequence_length,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 7307 name: "labels_indices"
|
| D | ops_history.v2.pbtxt | 12343 name: "labels_indices"
12384 name: "labels_indices"
|