Searched refs:num_classes_raw (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_decoder_ops.cc | 205 const int64 num_classes_raw = inputs_shape.dim_size(2); in Compute() local
207 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
209 const int num_classes = static_cast<const int>(num_classes_raw); in Compute()
306 const int64 num_classes_raw = inputs_shape.dim_size(2); in Compute() local
308 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
310 const int num_classes = static_cast<const int>(num_classes_raw); in Compute()
|
| D | ctc_loss_op.cc | 109 const int64 num_classes_raw = inputs_shape.dim_size(2); in Compute() local
111 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
113 const int num_classes = static_cast<const int>(num_classes_raw); in Compute()
277 const int64 num_classes_raw = inputs_shape.dim_size(2); in Compute() local
285 ctx, FastBoundsCheck(num_classes_raw, std::numeric_limits<int>::max()), in Compute()
289 const int num_classes = static_cast<const int>(num_classes_raw); in Compute()
|