D | shape_inference_test.cc | 58 void Relax(InferenceContext* c, DimensionHandle d0, DimensionHandle d1, in Relax() function in tensorflow::shape_inference::ShapeInferenceTest 60 c->Relax(d0, d1, out); in Relax() 62 void Relax(InferenceContext* c, ShapeHandle s0, ShapeHandle s1, in Relax() function in tensorflow::shape_inference::ShapeInferenceTest 64 c->Relax(s0, s1, out); in Relax() 547 Relax(&c, d2, d_unknown, &out); in TEST_F() 551 Relax(&c, d_unknown, d2, &out); in TEST_F() 554 Relax(&c, d_unknown, d_unknown_b, &out); in TEST_F() 560 Relax(&c, d2, d2, &out); in TEST_F() 562 Relax(&c, d_unknown, d_unknown, &out); in TEST_F() 566 Relax(&c, d2, d2_b, &out); in TEST_F() [all …]
|