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