Searched refs:input_lengths (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1134 const absl::Span<const int64> input_lengths, in CopyDataFromInput() argument
1156 if (within_src_bounds && i < input_lengths[axis]) { in CopyDataFromInput()
1167 GenerateIndices(fft_lengths, fft_strides, input_lengths, input_strides, in CopyDataFromInput()
1224 const absl::Span<const int64> input_lengths, in CopyDataFromInput() argument
1231 fft_strides, input_lengths, input_strides, data); in CopyDataFromInput()
1235 fft_strides, input_lengths, input_strides, data); in CopyDataFromInput()
1408 const auto input_lengths = GetDimensionLengths(input_literal); in HandleFft() local
1412 const auto input_strides = ComputeStrides(input_lengths, input_literal); in HandleFft()
1426 fft_size, fft_lengths, fft_strides, input_lengths, in HandleFft()
1440 GenerateIndices(output_lengths, output_strides, input_lengths, in HandleFft()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_array_ops.py | 353 input_lengths = nested_splits[axis][1:] - nested_splits[axis][:-1]
357 output_lengths = input_lengths * multiples[axis + 1]
|