HomeSort by relevance Sort by last modified time
    Searched refs:neg_sizes (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
runtime_fft_impl.h 137 auto neg_sizes = in_dims; local
138 neg_sizes[FFTRank] = fft_shape[FFTRank - 1] - in_dims[FFTRank];
166 if (neg_sizes[FFTRank] != 0) {
167 full_fft.slice(neg_target_indices, neg_sizes).device(device) =
168 full_fft.slice(neg_start_indices, neg_sizes)
  /external/tensorflow/tensorflow/core/kernels/
fft_ops.cc 195 auto neg_sizes = input_slice_sizes; variable
196 neg_sizes[FFTRank] =
227 if (neg_sizes[FFTRank] != 0) {
228 full_fft.slice(neg_target_indices, neg_sizes).device(device) =
229 full_fft.slice(neg_start_indices, neg_sizes)

Completed in 227 milliseconds