Searched refs:MergePrefix (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/seq2seq/ops/ |
| D | beam_search_ops.cc | 47 TF_RETURN_IF_ERROR(c->MergePrefix(step_ids, step_ids_prefix, &step_ids, in __anon559769010102()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 754 TEST_F(ShapeInferenceTest, MergePrefix) { in TEST_F() argument
774 EXPECT_TRUE(c.MergePrefix(s_unknown, s_u_2, &s_out, &s_prefix_out).ok()); in TEST_F()
777 EXPECT_TRUE(c.MergePrefix(s_1_u_3, s_unknown, &s_out, &s_prefix_out).ok()); in TEST_F()
781 EXPECT_TRUE(c.MergePrefix(s_1_u_3, s_u_2, &s_out, &s_prefix_out).ok()); in TEST_F()
794 c.MergePrefix(s_1_u_3, s_2_4, &s_out, &s_prefix_out).ToString(), in TEST_F()
803 c.MergePrefix(s_2_4, s_1_u_3, &s_out, &s_prefix_out).ToString(), in TEST_F()
|
| D | shape_inference.h | 417 Status MergePrefix(ShapeHandle s, ShapeHandle prefix, ShapeHandle* s_out,
|
| D | shape_inference.cc | 486 Status InferenceContext::MergePrefix(ShapeHandle s, ShapeHandle prefix, in MergePrefix() function in tensorflow::shape_inference::InferenceContext
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | data_flow_ops.cc | 70 c->MergePrefix(data_shape, partitions_shape, &unused, &unused)); in __anon4a7ac08d0202()
114 c->MergePrefix(data_shape, indices_shape, &unused, &unused)); in DynamicStitchShapeFunction()
763 c->MergePrefix(c->input(2), indices, &value_shape, &indices)); in __anon4a7ac08d1702()
|
| D | list_ops.cc | 120 c->MergePrefix(tensor, input_handles, &tensor, &input_handles)); in __anonebbd57d80402()
|
| D | math_ops.cc | 1114 c->MergePrefix(s_data, s_segment_ids, &s_data, &s_segment_ids)); in UnsortedSegmentReductionShapeFn()
|