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

  /external/tensorflow/tensorflow/stream_executor/
fft.h 128 // input_embed, output_embed:
142 uint64 input_stride, uint64 input_distance, uint64 *output_embed,
151 // input_embed, output_embed:
165 uint64 input_stride, uint64 input_distance, uint64 *output_embed,
236 uint64 input_stride, uint64 input_distance, uint64 *output_embed, \
241 uint64 input_stride, uint64 input_distance, uint64 *output_embed, \
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_fft.cc 114 uint64 *output_embed, uint64 output_stride, uint64 output_distance,
126 if (output_embed) {
127 output_embed_[i] = output_embed[i];
132 if (batch_count == 1 && input_embed == nullptr && output_embed == nullptr) {
233 output_embed ? output_embed_ : nullptr, output_stride,
258 output_embed ? output_embed_ : nullptr, output_stride,
279 /*output_embed=*/nullptr, /*output_stride=*/0,
418 uint64 input_stride, uint64 input_distance, uint64 *output_embed,
424 input_distance, output_embed, output_stride, output_distance, type,
436 uint64 input_stride, uint64 input_distance, uint64 *output_embed,
    [all...]
cuda_fft.h 71 uint64 *output_embed, uint64 output_stride,
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
fft_thunk.cc 147 uint64 output_embed[3]; local
156 output_embed[i] = output_shape_.dimensions(dim_offset);
164 input_distance, output_embed, output_stride, output_distance,
  /external/tensorflow/tensorflow/core/kernels/
fft_ops.cc 383 uint64 output_embed[3]; variable
391 output_embed[i] = output_shape.dim_size(dim_offset);
405 input_distance, output_embed, output_stride, output_distance,

Completed in 243 milliseconds