1// Copyright 2017 The TensorFlow Authors. All Rights Reserved.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14// =============================================================================
15#include "tensorflow/core/framework/node_def_builder.h"
16#include "tensorflow/core/framework/op.h"
17#include "tensorflow/core/framework/shape_inference_testutil.h"
18#include "tensorflow/core/framework/tensor.h"
19#include "tensorflow/core/framework/tensor_testutil.h"
20#include "tensorflow/core/lib/core/status.h"
21#include "tensorflow/core/lib/core/status_test_util.h"
22#include "tensorflow/core/platform/test.h"
23
24namespace tensorflow {
25
26TEST(StatsOpsTest, CreateFertileStatsVariable_ShapeFn) {
27  ShapeInferenceTestOp op("CreateFertileStatsVariable");
28  INFER_OK(op, "[1];[1]", "");
29}
30
31TEST(StatsOpsTest, FertileStatsSerialize_ShapeFn) {
32  ShapeInferenceTestOp op("FertileStatsSerialize");
33  INFER_OK(op, "[1]", "[]");
34}
35
36TEST(StatsOpsTest, FertileStatsDeserialize_ShapeFn) {
37  ShapeInferenceTestOp op("FertileStatsDeserialize");
38  INFER_OK(op, "[1];[1]", "");
39}
40
41TEST(StatsOpsTest, GrowTreeV4_ShapeFn) {
42  ShapeInferenceTestOp op("GrowTreeV4");
43  INFER_OK(op, "[1];[1];?", "");
44}
45
46TEST(StatsOpsTest, ProcessInputV4_ShapeFn) {
47  ShapeInferenceTestOp op("ProcessInputV4");
48  INFER_OK(op, "[1];[1];?;?;?;?;?;?;?", "[?]");
49}
50
51TEST(StatsOpsTest, FinalizeTree_ShapeFn) {
52  ShapeInferenceTestOp op("FinalizeTree");
53  INFER_OK(op, "[1];[1]", "");
54}
55
56}  // namespace tensorflow
57