Searched refs:kFft (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | plugin_registry.cc | 33 case PluginKind::kFft: in PluginKindString()
130 case PluginKind::kFft: in SetDefaultFactory()
153 case PluginKind::kFft: in HasFactory()
|
| D | plugin.h | 41 kFft, enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk.cc | 39 case Thunk::kFft: in ThunkKindToString()
|
| D | thunk.h | 53 kFft, enumerator
|
| D | fft_thunk.cc | 100 : Thunk(Kind::kFft, hlo), in FftThunk()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_opcode.h | 87 V(kFft, "fft", 1) \
|
| D | instruction_fusion.cc | 136 case HloOpcode::kFft: in IsExpensive()
|
| D | hlo_instruction.cc | 160 case HloOpcode::kFft: { in CreateFromProto()
1352 case HloOpcode::kFft: in CloneWithNewOperands()
1802 case HloOpcode::kFft: in IdenticalSlowPath()
2547 case HloOpcode::kFft: in Visit()
|
| D | hlo_graph_dumper.cc | 1020 case HloOpcode::kFft: in GetInstructionColor()
|
| D | layout_assignment.cc | 2004 case HloOpcode::kFft: in InstructionCanChangeLayout()
|
| D | hlo_instructions.cc | 168 : HloInstruction(HloOpcode::kFft, shape), fft_type_(fft_type) { in HloFftInstruction()
|
| D | hlo_parser.cc | 1106 case HloOpcode::kFft: { in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.cc | 143 opcode == HloOpcode::kFft || opcode == HloOpcode::kInfeed || in GetTargetParallelTaskCount()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_fft.cc | 587 rocm::kROCmPlatformId, PluginKind::kFft, gpu::kRocFftPlugin); in initialize_rocfft()
611 rocm::kROCmPlatformId, PluginKind::kFft, gpu::kRocFftPlugin); in initialize_rocfft()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_fft.cc | 557 cuda::kCudaPlatformId, PluginKind::kFft, gpu::kCuFftPlugin); in initialize_cufft()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1277 return AddInstruction(std::move(instr), HloOpcode::kFft, {operand}); in Fft()
|