Home
last modified time | relevance | path

Searched refs:SameHandle (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc52 bool SameHandle(DimensionHandle a, DimensionHandle b) { in SameHandle() function in tensorflow::shape_inference::ShapeInferenceTest
53 return a.SameHandle(b); in SameHandle()
55 bool SameHandle(ShapeHandle a, ShapeHandle b) { return a.SameHandle(b); } in SameHandle() function in tensorflow::shape_inference::ShapeInferenceTest
276 EXPECT_TRUE(SameHandle(d, c.Dim(in1, -3))); in TEST_F()
282 EXPECT_TRUE(SameHandle(d, c.Dim(in1, -2))); in TEST_F()
286 EXPECT_TRUE(SameHandle(d, c.Dim(in1, -1))); in TEST_F()
305 EXPECT_FALSE(SameHandle(c.Dim(c.input(1), 1), c.NumElements(c.input(1)))); in TEST_F()
326 EXPECT_FALSE(SameHandle(s1, s2)); in TEST_F()
327 EXPECT_FALSE(SameHandle(c.Dim(s2, 0), c.Dim(s2, 1))); in TEST_F()
331 EXPECT_FALSE(SameHandle(s1, s2)); in TEST_F()
[all …]
Dshape_inference_testutil.cc98 if (c.input(i).SameHandle(out)) { in InferShapes()
163 if (c.Dim(in, j).SameHandle(out_dim)) { in InferShapes()
Dshape_inference.cc439 if (d_old.SameHandle(d_new)) { in Relax()
465 if (d0.SameHandle(d1)) { in Merge()
514 if (s_old.SameHandle(s_new)) { in Relax()
534 if (d0.SameHandle(d1)) continue; in Relax()
559 if (s0.SameHandle(s1)) { in Merge()
584 if (d0.SameHandle(d1)) continue; in Merge()
1220 if (!existing.shape.SameHandle(new_values[i].shape)) { in MergeHandleShapesAndTypes()
Dshape_inference.h59 bool SameHandle(DimensionHandle d) const { return ptr_ == d.ptr_; } in SameHandle() function
102 bool SameHandle(ShapeHandle s) const { return ptr_ == s.ptr_; } in SameHandle() function
281 if (inputs_[idx].SameHandle(new_shape)) { in RelaxInput()
Dcommon_shape_fns.cc1355 } else if (dim_y.SameHandle(dim_x)) { in BroadcastBinaryOpOutputShapeFnHelper()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc57 return h1.SameHandle(h2); in operator ()()
831 !ic->input(dst_input).SameHandle(src_ic->output(src_output))) { in UpdateNode()
932 if (shape1.SameHandle(shape2)) { in OutputAsUnion()
942 if (!ctx->Dim(shape1, d).SameHandle(ctx->Dim(shape2, d))) { in OutputAsUnion()
956 if (s1.SameHandle(s2)) { in EquivalentShapes()
967 if (!InferenceContext::DimKnownRank(s1, i).SameHandle( in EquivalentShapes()
990 if (inferred_shape.SameHandle(annotated_shape)) { in CompatibleShapes()
1003 .SameHandle( in CompatibleShapes()
1849 if (!ic->output(0).SameHandle(input)) { in UpdateEnter()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner_test.cc38 bool SameHandle(shape_inference::DimensionHandle a, in SameHandle() function in tensorflow::ShapeRefinerTest
40 return a.SameHandle(b); in SameHandle()
43 bool SameHandle(shape_inference::ShapeHandle a, in SameHandle() function in tensorflow::ShapeRefinerTest
45 return a.SameHandle(b); in SameHandle()
1385 EXPECT_TRUE(SameHandle(ctx->Dim(ctx->output(0), 0), ctx->Dim(shp, 0))); in TEST_F()
1395 EXPECT_TRUE(SameHandle(ctx->Dim(shp, 0), ctx->Dim(shp2, 0))); in TEST_F()
Dshape_refiner.cc726 if (s0.SameHandle(s1)) { in SameDefinedShape()
736 if (!c->Dim(s0, i).SameHandle(c->Dim(s1, i))) { in SameDefinedShape()