/external/tensorflow/tensorflow/contrib/seq2seq/ops/ |
beam_search_ops.cc | 47 TF_RETURN_IF_ERROR(c->MergePrefix(step_ids, step_ids_prefix, &step_ids,
|
/external/tensorflow/tensorflow/core/framework/ |
shape_inference_test.cc | 759 TEST_F(ShapeInferenceTest, MergePrefix) { 779 EXPECT_TRUE(c.MergePrefix(s_unknown, s_u_2, &s_out, &s_prefix_out).ok()); 782 EXPECT_TRUE(c.MergePrefix(s_1_u_3, s_unknown, &s_out, &s_prefix_out).ok()); 786 EXPECT_TRUE(c.MergePrefix(s_1_u_3, s_u_2, &s_out, &s_prefix_out).ok()); 800 c.MergePrefix(s_1_u_3, s_2_4, &s_out, &s_prefix_out).ToString()) 811 c.MergePrefix(s_2_4, s_1_u_3, &s_out, &s_prefix_out).ToString()) [all...] |
shape_inference.h | 408 Status MergePrefix(ShapeHandle s, ShapeHandle prefix, ShapeHandle* s_out,
|
shape_inference.cc | 449 Status InferenceContext::MergePrefix(ShapeHandle s, ShapeHandle prefix, [all...] |
/external/tensorflow/tensorflow/core/ops/ |
data_flow_ops.cc | 70 c->MergePrefix(data_shape, partitions_shape, &unused, &unused)); 114 c->MergePrefix(data_shape, indices_shape, &unused, &unused)); [all...] |
math_ops.cc | 988 c->MergePrefix(s_data, s_segment_ids, &s_data, &s_segment_ids)); [all...] |