Searched refs:ComputeStrides (Results 1 – 3 of 3) sorted by relevance
962 std::vector<int64> ComputeStrides(const absl::Span<const int64> lengths, in ComputeStrides() function985 std::vector<int64> ComputeStrides(const absl::Span<const int64> lengths) { in ComputeStrides() function986 return ComputeStrides(lengths, in ComputeStrides()991 std::vector<int64> ComputeStrides(const absl::Span<const int64> lengths, in ComputeStrides() function994 ? ComputeStrides(lengths, literal.shape().layout()) in ComputeStrides()995 : ComputeStrides(lengths); in ComputeStrides()1389 const auto fft_strides = ComputeStrides(fft_lengths); in HandleFft()1412 const auto input_strides = ComputeStrides(input_lengths, input_literal); in HandleFft()1413 const auto output_strides = ComputeStrides(output_lengths, output_literal); in HandleFft()
747 void ComputeStrides(Dims<N>* dims) { in ComputeStrides() function
896 void ComputeStrides(Dims<N>* dims) {