Searched refs:kFft (Results 1 – 20 of 20) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | plugin_registry.cc | 36 case PluginKind::kFft: in PluginKindString()
133 case PluginKind::kFft: in SetDefaultFactory()
156 case PluginKind::kFft: in HasFactory()
|
| D | plugin.h | 41 kFft, enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk.cc | 59 case Thunk::kFft: in ThunkKindToString()
|
| D | thunk.h | 56 kFft, enumerator
|
| D | fft_thunk.cc | 106 : Thunk(Kind::kFft, thunk_info), in FftThunk()
|
| D | gpu_layout_assignment.cc | 240 } else if (instruction->opcode() == HloOpcode::kFft) { in AddBackendConstraints()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_opcode.h | 88 V(kFft, "fft", 1) \
|
| D | instruction_fusion.cc | 161 case HloOpcode::kFft: in IsExpensive()
|
| D | hlo_instruction.cc | 166 case HloOpcode::kFft: { in CreateFromProto()
1663 case HloOpcode::kFft: in CloneWithNewOperands()
2202 case HloOpcode::kFft: in IdenticalSlowPath()
3128 case HloOpcode::kFft: in Visit()
|
| D | hlo_graph_dumper.cc | 1043 case HloOpcode::kFft: in GetInstructionColor()
|
| D | sharding_propagation.cc | 252 case HloOpcode::kFft: in PickRepresentativeOperand()
|
| D | hlo_parser.cc | 109 case HloOpcode::kFft: in CanInferShape()
1658 case HloOpcode::kFft: { in ParseInstructionRhs()
|
| D | layout_assignment.cc | 2230 case HloOpcode::kFft: in InstructionCanChangeLayout()
|
| D | hlo_instructions.cc | 171 : HloInstruction(HloOpcode::kFft, shape), fft_type_(fft_type) { in HloFftInstruction()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_fft.cc | 592 rocm::kROCmPlatformId, PluginKind::kFft, gpu::kRocFftPlugin); in initialize_rocfft()
616 rocm::kROCmPlatformId, PluginKind::kFft, gpu::kRocFftPlugin); in initialize_rocfft()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_fft.cc | 586 cuda::kCudaPlatformId, PluginKind::kFft, gpu::kCuFftPlugin); in initialize_cufft()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 447 return xla::HloOpcode::kFft; in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 725 case HloOpcode::kFft: { in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 298 case HloOpcode::kFft: in EmitOp()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1641 return AddInstruction(std::move(instr), HloOpcode::kFft, {operand}); in FftInternal()
|