Home
last modified time | relevance | path

Searched defs:fft_type (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_fft.cc26 const void* run_options_ptr, void* out, void* operand, int32 fft_type, in __xla_cpu_runtime_EigenSingleThreadedFft()
Druntime_fft.cc30 const void* run_options_ptr, void* out, void* operand, int32 fft_type, in __xla_cpu_runtime_EigenFft()
Druntime_fft_impl.h173 int32 fft_type, int64 input_batch, int64 fft_length0, in EigenFftWithRank()
208 int32 fft_type, int32 fft_rank, int64 input_batch, in EigenFftImpl()
Dir_emitter.cc1342 llvm::FunctionType* fft_type = llvm::FunctionType::get( in HandleFft() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfft_ops.cc42 explicit GenericFftOp(OpKernelConstruction* ctx, FftType fft_type, in GenericFftOp()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.cc95 FftThunk::FftThunk(FftType fft_type, absl::Span<const int64> fft_length, in FftThunk()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc987 void Sweep(int64 fft_rank, FftType fft_type, in Sweep()
1119 int64 fft_rank, FftType fft_type, int64 fft_size, in CopyDataFromInput()
1172 int64 fft_rank, FftType fft_type, in CopyDataToOutput()
1209 int64 fft_rank, FftType fft_type, int64 fft_size, in CopyDataFromInput()
1229 int64 fft_rank, FftType fft_type, in CopyDataToOutput()
1248 int64 fft_rank, FftType fft_type, in CheckParameters()
1363 const FftType fft_type = fft->fft_type(); in HandleFft() local
Dhlo.proto148 xla.FftType fft_type = 31; field
Dhlo_instructions.h107 FftType fft_type() const { return fft_type_; } in fft_type() function
Dhlo_instruction.cc869 const Shape& shape, HloInstruction* operand, FftType fft_type, in CreateFft()
3552 FftType HloInstruction::fft_type() const { in fft_type() function in xla::HloInstruction
Dshape_inference.cc1842 const Shape& in, const FftType fft_type, in InferFftShape()
Dhlo_parser.cc1169 optional<FftType> fft_type; in ParseInstructionRhs() local
Dhlo_instructions.cc167 HloInstruction* operand, FftType fft_type, in HloFftInstruction()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc509 auto fft_type = in ImportInstruction() local
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc529 se::fft::Plan* plan, const se::fft::Type fft_type, in DoFFTInternal()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1632 def Fft(self, operand, fft_type, fft_lengths): argument
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1323 XlaOp XlaBuilder::Fft(XlaOp operand, const FftType fft_type, in Fft()
3089 XlaOp Fft(const XlaOp operand, FftType fft_type, in Fft()