Home
last modified time | relevance | path

Searched refs:output_lengths (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_array_ops.py378 output_lengths = input_lengths * multiples[axis + 1]
387 output_lengths = ragged_util.repeat_ranges(output_lengths, splits,
392 output_lengths = array_ops.tile(output_lengths, multiples[:1])
395 result_splits.append(ragged_util.lengths_to_splits(output_lengths))
/external/libtextclassifier/utils/tflite/
Dtoken_encoder.cc81 TfLiteTensor& output_lengths = in Prepare() local
87 context->ResizeTensor(context, &output_lengths, in Prepare()
154 TfLiteTensor& output_lengths = in Eval() local
156 output_lengths.data.i32[0] = encoded_positions.size() - num_skip; in Eval()
Dtext_encoder.cc168 TfLiteTensor& output_lengths = in Prepare() local
176 context->ResizeTensor(context, &output_lengths, in Prepare()
259 TfLiteTensor& output_lengths = in Eval() local
261 output_lengths.data.i32[0] = encoded_total.size() - num_skip; in Eval()