OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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