diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-02 13:01:59 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-02 13:05:54 -0800 |
commit | 5bf26acd87d3d44183fc28cb9576cda10c0255ca (patch) | |
tree | 5486c0ab9077496e30b594d35c089a45f7b7e18c /tensorflow/compiler/xla/service/cpu/runtime_fft.cc | |
parent | 97a843db78745fe3a8e418b3b1e93ef79fbfff12 (diff) |
Automated g4 rollback of changelist 180000981
PiperOrigin-RevId: 180581912
Diffstat (limited to 'tensorflow/compiler/xla/service/cpu/runtime_fft.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/cpu/runtime_fft.cc | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/cpu/runtime_fft.cc b/tensorflow/compiler/xla/service/cpu/runtime_fft.cc new file mode 100644 index 0000000000..848d2d2241 --- /dev/null +++ b/tensorflow/compiler/xla/service/cpu/runtime_fft.cc @@ -0,0 +1,37 @@ +/* Copyright 2017 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +==============================================================================*/ + +#include "tensorflow/compiler/xla/service/cpu/runtime_fft.h" + +#define EIGEN_USE_THREADS + +#include "tensorflow/compiler/xla/executable_run_options.h" +#include "tensorflow/compiler/xla/service/cpu/runtime_fft_impl.h" +#include "tensorflow/core/platform/dynamic_annotations.h" +#include "tensorflow/core/platform/types.h" + +using tensorflow::int32; +using tensorflow::int64; + +TF_ATTRIBUTE_NO_SANITIZE_MEMORY void __xla_cpu_runtime_EigenFft( + const void* run_options_ptr, void* out, void* operand, int32 fft_type, + int32 fft_rank, int64 input_batch, int64 fft_length0, int64 fft_length1, + int64 fft_length2) { + const xla::ExecutableRunOptions* run_options = + static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); + tensorflow::xla::EigenFftImpl(*run_options->intra_op_thread_pool(), out, + operand, fft_type, fft_rank, input_batch, + fft_length0, fft_length1, fft_length2); +} |