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.cc379 if (d_old.SameHandle(d_new)) { in Relax()
405 if (d0.SameHandle(d1)) { in Merge()
454 if (s_old.SameHandle(s_new)) { in Relax()
474 if (d0.SameHandle(d1)) continue; in Relax()
499 if (s0.SameHandle(s1)) { in Merge()
524 if (d0.SameHandle(d1)) continue; in Merge()
1160 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
249 if (inputs_[idx].SameHandle(new_shape)) { in RelaxInput()
Dcommon_shape_fns.cc1940 } else if (dim_y.SameHandle(dim_x)) { in BroadcastBinaryOpOutputShapeFnHelper()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc58 return h1.SameHandle(h2); in operator ()()
854 !ic->input(dst_input).SameHandle(src_ic->output(src_output))) { in UpdateNode()
981 if (shape1.SameHandle(shape2)) { in OutputAsUnion()
991 if (!ctx->Dim(shape1, d).SameHandle(ctx->Dim(shape2, d))) { in OutputAsUnion()
1005 if (s1.SameHandle(s2)) { in EquivalentShapes()
1016 if (!InferenceContext::DimKnownRank(s1, i).SameHandle( in EquivalentShapes()
1039 if (inferred_shape.SameHandle(annotated_shape)) { in CompatibleShapes()
1052 .SameHandle( in CompatibleShapes()
1068 if (inferred_shape.SameHandle(annotated_shape)) { in SameShapes()
1967 if (!ic->output(0).SameHandle(input)) { in UpdateEnter()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc87 if (outer_context->input(index).SameHandle(ShapeHandle())) { in InferShapesForFunctionSubNode()
760 if (s0.SameHandle(s1)) { in SameDefinedShape()
770 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/compiler/mlir/tensorflow/translate/
Dimport_model.cc800 if (s0.SameHandle(s1) || (!c->RankKnown(s0) && !c->RankKnown(s1))) { in AddNodesToShapeRefiner()
807 if (!c->Dim(s0, i).SameHandle(c->Dim(s1, i))) { in AddNodesToShapeRefiner()