D | shape_inference_test.cc | 66 void Relax(InferenceContext* c, DimensionHandle d0, DimensionHandle d1, in Relax() function in tensorflow::shape_inference::ShapeInferenceTest 68 c->Relax(d0, d1, out); in Relax() 70 void Relax(InferenceContext* c, ShapeHandle s0, ShapeHandle s1, in Relax() function in tensorflow::shape_inference::ShapeInferenceTest 72 c->Relax(s0, s1, out); in Relax() 562 Relax(&c, d2, d_unknown, &out); in TEST_F() 566 Relax(&c, d_unknown, d2, &out); in TEST_F() 569 Relax(&c, d_unknown, d_unknown_b, &out); in TEST_F() 575 Relax(&c, d2, d2, &out); in TEST_F() 577 Relax(&c, d_unknown, d_unknown, &out); in TEST_F() 581 Relax(&c, d2, d2_b, &out); in TEST_F() [all …]
|