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