Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc59 bool SameHandle(DimensionHandle a, DimensionHandle b) { in SameHandle() function in tensorflow::shape_inference::ShapeInferenceTest
60 return a.SameHandle(b); in SameHandle()
62 bool SameHandle(ShapeHandle a, ShapeHandle b) { return a.SameHandle(b); } in SameHandle() function in tensorflow::shape_inference::ShapeInferenceTest
301 EXPECT_TRUE(SameHandle(d, c.Dim(in1, -3))); in TEST_F()
307 EXPECT_TRUE(SameHandle(d, c.Dim(in1, -2))); in TEST_F()
311 EXPECT_TRUE(SameHandle(d, c.Dim(in1, -1))); in TEST_F()
330 EXPECT_FALSE(SameHandle(c.Dim(c.input(1), 1), c.NumElements(c.input(1)))); in TEST_F()
351 EXPECT_FALSE(SameHandle(s1, s2)); in TEST_F()
352 EXPECT_FALSE(SameHandle(c.Dim(s2, 0), c.Dim(s2, 1))); in TEST_F()
356 EXPECT_FALSE(SameHandle(s1, s2)); in TEST_F()
[all …]
Dshape_inference_testutil.cc97 if (c.input(i).SameHandle(out)) { in InferShapes()
161 if (c.Dim(in, j).SameHandle(out_dim)) { in InferShapes()
Dshape_inference.cc388 if (d_old.SameHandle(d_new)) { in Relax()
414 if (d0.SameHandle(d1)) { in Merge()
463 if (s_old.SameHandle(s_new)) { in Relax()
483 if (d0.SameHandle(d1)) continue; in Relax()
508 if (s0.SameHandle(s1)) { in Merge()
533 if (d0.SameHandle(d1)) continue; in Merge()
1228 if (!existing.shape.SameHandle(new_values[i].shape)) { in MergeHandleShapesAndTypes()
Dshape_inference.h132 bool SameHandle(DimensionHandle d) const { return ptr_ == d.ptr_; } in SameHandle() function
172 bool SameHandle(ShapeHandle s) const { return ptr_ == s.ptr_; } in SameHandle() function
330 if (inputs_[idx].SameHandle(new_shape)) { in RelaxInput()
Dcommon_shape_fns.cc2135 } else if (dim_y.SameHandle(dim_x)) { in BroadcastBinaryOpOutputShapeFnHelper()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc89 if (outer_context->input(index).SameHandle(ShapeHandle())) { in InferShapesForFunctionSubNode()
790 if (s0.SameHandle(s1)) { in SameDefinedShape()
800 if (!c->Dim(s0, i).SameHandle(c->Dim(s1, i))) { in SameDefinedShape()
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()
1418 EXPECT_TRUE(SameHandle(ctx->Dim(ctx->output(0), 0), ctx->Dim(shp, 0))); in TEST_F()
1428 EXPECT_TRUE(SameHandle(ctx->Dim(shp, 0), ctx->Dim(shp2, 0))); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc68 return h1.SameHandle(h2); in operator ()()
1042 !ic->input(dst_input).SameHandle(src_ic->output(src_output))) { in UpdateNode()
1181 if (shape1.SameHandle(shape2)) { in OutputAsUnion()
1191 if (!ctx->Dim(shape1, d).SameHandle(ctx->Dim(shape2, d))) { in OutputAsUnion()
1205 if (s1.SameHandle(s2)) { in EquivalentShapes()
1216 if (!InferenceContext::DimKnownRank(s1, i).SameHandle( in EquivalentShapes()
1239 if (inferred_shape.SameHandle(annotated_shape)) { in CompatibleShapes()
1252 .SameHandle( in CompatibleShapes()
1268 if (inferred_shape.SameHandle(annotated_shape)) { in SameShapes()
2326 if (!ic->output(0).SameHandle(input)) { in UpdateEnter()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc989 if (s0.SameHandle(s1) || (!c->RankKnown(s0) && !c->RankKnown(s1))) { in AddNodesToShapeRefiner()
996 if (!c->Dim(s0, i).SameHandle(c->Dim(s1, i))) { in AddNodesToShapeRefiner()