Searched refs:SameHandle (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | shape_inference_test.cc | 52 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 …]
|
D | shape_inference_testutil.cc | 98 if (c.input(i).SameHandle(out)) { in InferShapes() 163 if (c.Dim(in, j).SameHandle(out_dim)) { in InferShapes()
|
D | shape_inference.cc | 439 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()
|
D | shape_inference.h | 59 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()
|
D | common_shape_fns.cc | 1355 } else if (dim_y.SameHandle(dim_x)) { in BroadcastBinaryOpOutputShapeFnHelper()
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | graph_properties.cc | 57 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/ |
D | shape_refiner_test.cc | 38 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()
|
D | shape_refiner.cc | 726 if (s0.SameHandle(s1)) { in SameDefinedShape() 736 if (!c->Dim(s0, i).SameHandle(c->Dim(s1, i))) { in SameDefinedShape()
|