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.cc60 bool SameHandle(DimensionHandle a, DimensionHandle b) { in SameHandle() function in tensorflow::shape_inference::ShapeInferenceTest
61 return a.SameHandle(b); in SameHandle()
63 bool SameHandle(ShapeHandle a, ShapeHandle b) { return a.SameHandle(b); } in SameHandle() function in tensorflow::shape_inference::ShapeInferenceTest
293 EXPECT_TRUE(SameHandle(d, c.Dim(in1, -3))); in TEST_F()
299 EXPECT_TRUE(SameHandle(d, c.Dim(in1, -2))); in TEST_F()
303 EXPECT_TRUE(SameHandle(d, c.Dim(in1, -1))); in TEST_F()
322 EXPECT_FALSE(SameHandle(c.Dim(c.input(1), 1), c.NumElements(c.input(1)))); in TEST_F()
343 EXPECT_FALSE(SameHandle(s1, s2)); in TEST_F()
344 EXPECT_FALSE(SameHandle(c.Dim(s2, 0), c.Dim(s2, 1))); in TEST_F()
348 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.cc381 if (d_old.SameHandle(d_new)) { in Relax()
407 if (d0.SameHandle(d1)) { in Merge()
456 if (s_old.SameHandle(s_new)) { in Relax()
476 if (d0.SameHandle(d1)) continue; in Relax()
501 if (s0.SameHandle(s1)) { in Merge()
526 if (d0.SameHandle(d1)) continue; in Merge()
1195 if (!existing.shape.SameHandle(new_values[i].shape)) { in MergeHandleShapesAndTypes()
Dshape_inference.h57 bool SameHandle(DimensionHandle d) const { return ptr_ == d.ptr_; } in SameHandle() function
97 bool SameHandle(ShapeHandle s) const { return ptr_ == s.ptr_; } in SameHandle() function
254 if (inputs_[idx].SameHandle(new_shape)) { in RelaxInput()
Dcommon_shape_fns.cc2044 } 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()
769 if (s0.SameHandle(s1)) { in SameDefinedShape()
779 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()
1370 EXPECT_TRUE(SameHandle(ctx->Dim(ctx->output(0), 0), ctx->Dim(shp, 0))); in TEST_F()
1380 EXPECT_TRUE(SameHandle(ctx->Dim(shp, 0), ctx->Dim(shp2, 0))); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc64 return h1.SameHandle(h2); in operator ()()
1013 !ic->input(dst_input).SameHandle(src_ic->output(src_output))) { in UpdateNode()
1144 if (shape1.SameHandle(shape2)) { in OutputAsUnion()
1154 if (!ctx->Dim(shape1, d).SameHandle(ctx->Dim(shape2, d))) { in OutputAsUnion()
1168 if (s1.SameHandle(s2)) { in EquivalentShapes()
1179 if (!InferenceContext::DimKnownRank(s1, i).SameHandle( in EquivalentShapes()
1202 if (inferred_shape.SameHandle(annotated_shape)) { in CompatibleShapes()
1215 .SameHandle( in CompatibleShapes()
1231 if (inferred_shape.SameHandle(annotated_shape)) { in SameShapes()
2282 if (!ic->output(0).SameHandle(input)) { in UpdateEnter()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc879 if (s0.SameHandle(s1) || (!c->RankKnown(s0) && !c->RankKnown(s1))) { in AddNodesToShapeRefiner()
886 if (!c->Dim(s0, i).SameHandle(c->Dim(s1, i))) { in AddNodesToShapeRefiner()