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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
fft_thunk.cc 149 uint64 output_distance = 1; local
157 output_distance *= output_shape_.dimensions(dim_offset);
164 input_distance, output_embed, output_stride, output_distance,
166 scale_factor_ = 1.0f / output_distance;
  /external/tensorflow/tensorflow/stream_executor/
fft.h 138 // output_distance: Indicates the distance between the first element of two
143 uint64 output_stride, uint64 output_distance, Type type,
161 // output_distance: Indicates the distance between the first element of two
166 uint64 output_stride, uint64 output_distance, Type type,
237 uint64 output_stride, uint64 output_distance, fft::Type type, \
242 uint64 output_stride, uint64 output_distance, fft::Type type, \
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_fft.h 72 uint64 output_distance, fft::Type type,
cuda_fft.cc 114 uint64 *output_embed, uint64 output_stride, uint64 output_distance,
234 output_distance, CUDAFftType(type), batch_count);
259 output_distance, CUDAFftType(type), batch_count,
280 /*output_distance=*/0, type, 1, scratch_allocator);
419 uint64 output_stride, uint64 output_distance, fft::Type type,
424 input_distance, output_embed, output_stride, output_distance, type,
437 uint64 output_stride, uint64 output_distance, fft::Type type,
442 input_distance, output_embed, output_stride, output_distance, type,
  /external/tensorflow/tensorflow/core/kernels/
fft_ops.cc 385 uint64 output_distance = 1; variable
392 output_distance *= output_shape.dim_size(dim_offset);
405 input_distance, output_embed, output_stride, output_distance,
423 auto alpha = 1.f / output_distance;
439 auto alpha = complex64(1.f / output_distance);

Completed in 233 milliseconds