Searched refs:output_lengths (Results 1 – 4 of 4) sorted by relevance
| /external/libtextclassifier/native/utils/tflite/ |
| D | token_encoder.cc | 81 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()
|
| D | text_encoder.cc | 169 TfLiteTensor& output_lengths = in Prepare() local
177 context->ResizeTensor(context, &output_lengths, in Prepare()
260 TfLiteTensor& output_lengths = in Eval() local
262 output_lengths.data.i32[0] = encoded_total.size() - num_skip; in Eval()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_array_ops.py | 357 output_lengths = input_lengths * multiples[axis + 1]
366 output_lengths = ragged_util.repeat_ranges(output_lengths, splits,
371 output_lengths = array_ops.tile(output_lengths, multiples[:1])
374 result_splits.append(ragged_util.lengths_to_splits(output_lengths))
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1187 const absl::Span<const int64> output_lengths, in CopyDataToOutput() argument
1202 for (int64 i = 0; i < output_lengths[axis]; i++) { in CopyDataToOutput()
1215 GenerateIndices(output_lengths, output_strides, fft_lengths, fft_strides, in CopyDataToOutput()
1244 const absl::Span<const int64> output_lengths, in CopyDataToOutput() argument
1250 fft_strides, output_lengths, output_strides, in CopyDataToOutput()
1254 fft_lengths, fft_strides, output_lengths, in CopyDataToOutput()
1409 const auto output_lengths = GetDimensionLengths(output_literal); in HandleFft() local
1413 const auto output_strides = ComputeStrides(output_lengths, output_literal); in HandleFft()
1434 fft_lengths, fft_strides, output_lengths, in HandleFft()
1440 GenerateIndices(output_lengths, output_strides, input_lengths, in HandleFft()
|