Home
last modified time | relevance | path

Searched refs:UnknownShapeOfRank (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dragged_array_ops.cc84 ShapeHandle out_values = c->UnknownShapeOfRank(1); in __anon9deb1dc20102()
85 ShapeHandle out_splits = c->UnknownShapeOfRank(1); in __anon9deb1dc20102()
146 c->set_output(i, c->UnknownShapeOfRank(1)); in RaggedGatherShapeFn()
153 TF_RETURN_IF_ERROR(c->Concatenate(c->UnknownShapeOfRank(1), value, &values)); in RaggedGatherShapeFn()
Dragged_math_ops.cc77 c->set_output(1, c->UnknownShapeOfRank(1)); in RaggedRangeShapeFn()
Dbatch_ops.cc120 c->set_output(0, c->UnknownShapeOfRank(c->Rank(c->input(2)))); in __anond951767c0302()
Dparsing_ops.cc372 TF_RETURN_IF_ERROR(c->Concatenate(input, c->UnknownShapeOfRank(1), in __anonf8f6e66e0802()
448 c->UnknownShapeOfRank(1), c, in __anonf8f6e66e0902()
Dragged_conversion_ops.cc236 c->set_output(i, c->UnknownShapeOfRank(1)); in RaggedTensorFromVariantShapeFn()
Darray_ops.cc103 c->set_output(0, c->UnknownShapeOfRank(c->Value(n_dim))); in PadShapeFn()
582 out = c->UnknownShapeOfRank(c->Rank(input)); in __anon42d741190a02()
643 c->set_output(i, c->UnknownShapeOfRank(rank)); in __anon42d741190b02()
882 c->Concatenate(diag_batch_shape, c->UnknownShapeOfRank(2), &diag)); in __anon42d741191102()
1198 c->set_output(0, c->UnknownShapeOfRank(c->Rank(params_shape) + in __anon42d741191802()
1511 c->set_output(0, c->UnknownShapeOfRank(c->Rank(c->input(0)))); in __anon42d741191f02()
1542 c->set_output(0, c->UnknownShapeOfRank(c->Rank(c->input(0)))); in __anon42d741192102()
2010 c->set_output(0, c->UnknownShapeOfRank(input_rank)); in __anon42d741192c02()
Dstring_ops.cc383 c->set_output(0, c->UnknownShapeOfRank(1)); in __anon4354d5f10b02()
Dmath_ops.cc1636 c->set_output(0, c->UnknownShapeOfRank(1)); in __anon725c8f640d02()
1655 c->set_output(0, c->UnknownShapeOfRank(1)); in __anon725c8f640e02()
Dnn_ops.cc474 padded = c->UnknownShapeOfRank(4); in CommonFusedConvCalculations()
1440 c->set_output(0, c->UnknownShapeOfRank(4)); in __anonad10f3362202()
/external/tensorflow/tensorflow/compiler/xrt/ops/
Dxrt_compile_ops.cc29 c->set_output(1, c->UnknownShapeOfRank(1)); in __anone85cc47f0102()
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc35 c->set_output(0, c->UnknownShapeOfRank(c->Rank(c->input(0)))); in UnchangedRank()
553 0, c->UnknownShapeOfRank(rank - dimensions_to_reduce.size())); in __anonbbc1c2d10502()
604 i, c->UnknownShapeOfRank(rank - dimensions_to_reduce.size())); in __anonbbc1c2d10702()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc641 auto s_u1 = c.UnknownShapeOfRank(1); in TEST_F()
642 auto s_u2 = c.UnknownShapeOfRank(1); in TEST_F()
733 auto s_u1 = c.UnknownShapeOfRank(1); in TEST_F()
734 auto s_u2 = c.UnknownShapeOfRank(1); in TEST_F()
1229 TEST_F(ShapeInferenceTest, UnknownShapeOfRank) { in TEST_F() argument
1234 auto unknown_shape_of_rank_3 = c.UnknownShapeOfRank(3); in TEST_F()
1237 auto unknown_shape_of_rank_0 = c.UnknownShapeOfRank(0); in TEST_F()
Dshape_inference.h451 ShapeHandle UnknownShapeOfRank(int64 rank);
Dcommon_shape_fns.cc1487 c->Concatenate(diag_prefix, c->UnknownShapeOfRank(2), &diag_shape)); in MatrixSetDiagV2Shape()
1850 c->set_output(0, c->UnknownShapeOfRank(c->Rank(input))); in ReductionShape()
2217 c->set_output(0, c->UnknownShapeOfRank(c->Rank(input))); in SliceShape()
Dshape_inference.cc693 ShapeHandle InferenceContext::UnknownShapeOfRank(int64 rank) { in UnknownShapeOfRank() function in tensorflow::shape_inference::InferenceContext