MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetPlan

Method GetPlan

tensorflow/stream_executor/cuda/cuda_fft.h:58–64  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

56 // Get FFT direction in cuFFT based on FFT type.
57 int GetFftDirection() const;
58 cufftHandle GetPlan() const {
59 if (IsInitialized()) {
60 return plan_;
61 } else {
62 LOG(FATAL) << "Try to get cufftHandle value before initialization.";
63 }
64 }
65
66 // Initialize function for batched plan
67 port::Status Initialize(GpuExecutor* parent, Stream* stream, int rank,

Callers 2

DoFftInternalMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected