Searched defs:fft_type (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_fft.cc | 26 const void* run_options_ptr, void* out, void* operand, int32 fft_type, in __xla_cpu_runtime_EigenSingleThreadedFft()
|
| D | runtime_fft.cc | 29 const void* run_options_ptr, void* out, void* operand, int32 fft_type, in __xla_cpu_runtime_EigenFft()
|
| D | runtime_fft_impl.h | 173 int32 fft_type, int64 input_batch, int64 fft_length0, in EigenFftWithRank()
208 int32 fft_type, int32 fft_rank, int64 input_batch, in EigenFftImpl()
|
| D | ir_emitter.cc | 1279 llvm::FunctionType* fft_type = llvm::FunctionType::get( in HandleFft() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fft_ops.cc | 42 explicit GenericFftOp(OpKernelConstruction* ctx, FftType fft_type, in GenericFftOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 95 FftThunk::FftThunk(FftType fft_type, absl::Span<const int64> fft_length, in FftThunk()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 144 xla.FftType fft_type = 31; field
|
| D | hlo_instructions.h | 107 FftType fft_type() const { return fft_type_; } in fft_type() function
|
| D | hlo_instruction.cc | 791 const Shape& shape, HloInstruction* operand, FftType fft_type, in CreateFft()
3222 FftType HloInstruction::fft_type() const { in fft_type() function in xla::HloInstruction
|
| D | shape_inference.cc | 1819 const Shape& in, const FftType fft_type, in InferFftShape()
|
| D | hlo_parser.cc | 1107 optional<FftType> fft_type; in ParseInstructionRhs() local
|
| D | hlo_instructions.cc | 166 HloInstruction* operand, FftType fft_type, in HloFftInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1264 XlaOp XlaBuilder::Fft(const XlaOp& operand, const FftType fft_type, in Fft()
3024 XlaOp Fft(const XlaOp& operand, FftType fft_type, in Fft()
|