Searched refs:ShapeInferenceTestutil (Results 1 – 5 of 5) sorted by relevance
47 class ShapeInferenceTestutil {74 ShapeInferenceTestutil() {} in ShapeInferenceTestutil() function85 "", ::tensorflow::shape_inference::ShapeInferenceTestutil::InferShapes( \91 ::tensorflow::shape_inference::ShapeInferenceTestutil::InferShapes( \
63 return ShapeInferenceTestutil::InferShapes(op, ins, expected_outs) in RunInferShapes()98 auto error_message = ShapeInferenceTestutil::InferShapes( in TEST()
29 Status ShapeInferenceTestutil::InferShapes(ShapeInferenceTestOp op, in InferShapes()229 Status ShapeInferenceTestutil::MakeShapeFromString( in MakeShapeFromString()
71 friend class ShapeInferenceTestutil; variable109 friend class ShapeInferenceTestutil; variable674 friend class ShapeInferenceTestutil; // For testing shapes. variable
81 shape_inference::ShapeInferenceTestutil::InferShapes(op, "[1,2]", "in0")); in TEST()99 TF_ASSERT_OK(shape_inference::ShapeInferenceTestutil::InferShapes( in TEST()129 TF_ASSERT_OK(shape_inference::ShapeInferenceTestutil::InferShapes( in TEST()