Home
last modified time | relevance | path

Searched refs:MergePrefix (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc769 TEST_F(ShapeInferenceTest, MergePrefix) { in TEST_F() argument
789 EXPECT_TRUE(c.MergePrefix(s_unknown, s_u_2, &s_out, &s_prefix_out).ok()); in TEST_F()
792 EXPECT_TRUE(c.MergePrefix(s_1_u_3, s_unknown, &s_out, &s_prefix_out).ok()); in TEST_F()
796 EXPECT_TRUE(c.MergePrefix(s_1_u_3, s_u_2, &s_out, &s_prefix_out).ok()); in TEST_F()
809 c.MergePrefix(s_1_u_3, s_2_4, &s_out, &s_prefix_out).ToString(), in TEST_F()
818 c.MergePrefix(s_2_4, s_1_u_3, &s_out, &s_prefix_out).ToString(), in TEST_F()
Dshape_inference.h403 Status MergePrefix(ShapeHandle s, ShapeHandle prefix, ShapeHandle* s_out,
Dshape_inference.cc428 Status InferenceContext::MergePrefix(ShapeHandle s, ShapeHandle prefix, in MergePrefix() function in tensorflow::shape_inference::InferenceContext
Dcommon_shape_fns.cc2101 c->MergePrefix(s_data, s_segment_ids, &s_data, &s_segment_ids)); in UnsortedSegmentReductionShapeFn()
/external/tensorflow/tensorflow/core/ops/
Ddata_flow_ops.cc70 c->MergePrefix(data_shape, partitions_shape, &unused, &unused)); in __anonea01102b0202()
114 c->MergePrefix(data_shape, indices_shape, &unused, &unused)); in DynamicStitchShapeFunction()
818 c->MergePrefix(c->input(2), indices, &value_shape, &indices)); in __anonea01102b1b02()
Dlist_ops.cc126 c->MergePrefix(tensor, input_handles, &tensor, &input_handles)); in __anonabedecb60402()