diff options
Diffstat (limited to 'tensorflow/core/kernels/spectrogram_op_test.cc')
-rw-r--r-- | tensorflow/core/kernels/spectrogram_op_test.cc | 104 |
1 files changed, 0 insertions, 104 deletions
diff --git a/tensorflow/core/kernels/spectrogram_op_test.cc b/tensorflow/core/kernels/spectrogram_op_test.cc deleted file mode 100644 index 5c3cbeeeb9..0000000000 --- a/tensorflow/core/kernels/spectrogram_op_test.cc +++ /dev/null @@ -1,104 +0,0 @@ -/* 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. -==============================================================================*/ - -#define EIGEN_USE_THREADS - -#include <functional> -#include <memory> -#include <vector> - -#include "tensorflow/cc/client/client_session.h" -#include "tensorflow/cc/ops/audio_ops.h" -#include "tensorflow/cc/ops/const_op.h" -#include "tensorflow/cc/ops/math_ops.h" -#include "tensorflow/core/framework/tensor_testutil.h" -#include "tensorflow/core/framework/types.h" -#include "tensorflow/core/framework/types.pb.h" -#include "tensorflow/core/kernels/ops_util.h" -#include "tensorflow/core/lib/core/status_test_util.h" -#include "tensorflow/core/platform/test.h" - -namespace tensorflow { - -using namespace ops; // NOLINT(build/namespaces) - -TEST(SpectrogramOpTest, SimpleTest) { - Scope root = Scope::NewRootScope(); - - Tensor audio_tensor(DT_FLOAT, TensorShape({8, 1})); - test::FillValues<float>(&audio_tensor, - {-1.0f, 0.0f, 1.0f, 0.0f, -1.0f, 0.0f, 1.0f, 0.0f}); - - Output audio_const_op = Const(root.WithOpName("audio_const_op"), - Input::Initializer(audio_tensor)); - - AudioSpectrogram spectrogram_op = - AudioSpectrogram(root.WithOpName("spectrogram_op"), audio_const_op, 8, 1); - - TF_ASSERT_OK(root.status()); - - ClientSession session(root); - std::vector<Tensor> outputs; - - TF_EXPECT_OK(session.Run(ClientSession::FeedType(), - {spectrogram_op.spectrogram}, &outputs)); - - const Tensor& spectrogram_tensor = outputs[0]; - - EXPECT_EQ(3, spectrogram_tensor.dims()); - EXPECT_EQ(5, spectrogram_tensor.dim_size(2)); - EXPECT_EQ(1, spectrogram_tensor.dim_size(1)); - EXPECT_EQ(1, spectrogram_tensor.dim_size(0)); - - test::ExpectTensorNear<float>( - spectrogram_tensor, - test::AsTensor<float>({0, 1, 2, 1, 0}, TensorShape({1, 1, 5})), 1e-3); -} - -TEST(SpectrogramOpTest, SquaredTest) { - Scope root = Scope::NewRootScope(); - - Tensor audio_tensor(DT_FLOAT, TensorShape({8, 1})); - test::FillValues<float>(&audio_tensor, - {-1.0f, 0.0f, 1.0f, 0.0f, -1.0f, 0.0f, 1.0f, 0.0f}); - - Output audio_const_op = Const(root.WithOpName("audio_const_op"), - Input::Initializer(audio_tensor)); - - AudioSpectrogram spectrogram_op = - AudioSpectrogram(root.WithOpName("spectrogram_op"), audio_const_op, 8, 1, - AudioSpectrogram::Attrs().MagnitudeSquared(true)); - - TF_ASSERT_OK(root.status()); - - ClientSession session(root); - std::vector<Tensor> outputs; - - TF_EXPECT_OK(session.Run(ClientSession::FeedType(), - {spectrogram_op.spectrogram}, &outputs)); - - const Tensor& spectrogram_tensor = outputs[0]; - - EXPECT_EQ(3, spectrogram_tensor.dims()); - EXPECT_EQ(5, spectrogram_tensor.dim_size(2)); - EXPECT_EQ(1, spectrogram_tensor.dim_size(1)); - EXPECT_EQ(1, spectrogram_tensor.dim_size(0)); - - test::ExpectTensorNear<float>( - spectrogram_tensor, - test::AsTensor<float>({0, 1, 4, 1, 0}, TensorShape({1, 1, 5})), 1e-3); -} - -} // namespace tensorflow |