Searched refs:UnknownShapeOfRank (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_array_ops.cc | 72 c->set_output(i, c->UnknownShapeOfRank(1)); in RaggedGatherShapeFn()
79 TF_RETURN_IF_ERROR(c->Concatenate(c->UnknownShapeOfRank(1), value, &values)); in RaggedGatherShapeFn()
|
| D | ragged_math_ops.cc | 76 c->set_output(1, c->UnknownShapeOfRank(1)); in RaggedRangeShapeFn()
|
| D | batch_ops.cc | 101 c->set_output(0, c->UnknownShapeOfRank(c->Rank(c->input(2)))); in __anon720d5c150302()
|
| D | array_ops.cc | 96 c->set_output(0, c->UnknownShapeOfRank(c->Value(n_dim))); in PadShapeFn()
576 out = c->UnknownShapeOfRank(c->Rank(input)); in __anondb9326b20a02()
637 c->set_output(i, c->UnknownShapeOfRank(rank)); in __anondb9326b20b02()
853 c->Concatenate(diag_prefix, c->UnknownShapeOfRank(2), &diag)); in __anondb9326b21102()
1130 c->set_output(0, c->UnknownShapeOfRank(c->Rank(params_shape) + in __anondb9326b21802()
1855 c->set_output(0, c->UnknownShapeOfRank(input_rank)); in __anondb9326b22c02()
|
| D | lookup_ops.cc | 271 ShapeHandle keys = c->UnknownShapeOfRank(1); in __anonc8e9eead0802()
|
| D | math_ops.cc | 1510 c->set_output(0, c->UnknownShapeOfRank(1)); in __anon57f5a5dd0d02()
1529 c->set_output(0, c->UnknownShapeOfRank(1)); in __anon57f5a5dd0e02()
|
| D | nn_ops.cc | 394 padded = c->UnknownShapeOfRank(4); in CommonFusedConvCalculations()
1346 c->set_output(0, c->UnknownShapeOfRank(4)); in __anon4650176f2802()
|
| /external/tensorflow/tensorflow/compiler/xrt/ops/ |
| D | xrt_compile_ops.cc | 29 c->set_output(1, c->UnknownShapeOfRank(1)); in __anon273601d80102()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 31 c->set_output(0, c->UnknownShapeOfRank(c->Rank(c->input(0)))); in UnchangedRank()
336 0, c->UnknownShapeOfRank(rank - dimensions_to_reduce.size())); in __anon3500048a0302()
|
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/ |
| D | reduce_slice_ops.cc | 55 c->set_output(0, c->UnknownShapeOfRank(c->Rank(c->input(0)))); in ReduceSliceShapeFn()
|
| /external/tensorflow/tensorflow/contrib/ffmpeg/ |
| D | decode_video_op.cc | 101 c->set_output(0, c->UnknownShapeOfRank(4)); in __anon8bd361450102()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 626 auto s_u1 = c.UnknownShapeOfRank(1); in TEST_F()
627 auto s_u2 = c.UnknownShapeOfRank(1); in TEST_F()
718 auto s_u1 = c.UnknownShapeOfRank(1); in TEST_F()
719 auto s_u2 = c.UnknownShapeOfRank(1); in TEST_F()
1221 TEST_F(ShapeInferenceTest, UnknownShapeOfRank) { in TEST_F() argument
1226 auto unknown_shape_of_rank_3 = c.UnknownShapeOfRank(3); in TEST_F()
1229 auto unknown_shape_of_rank_0 = c.UnknownShapeOfRank(0); in TEST_F()
|
| D | shape_inference.h | 465 ShapeHandle UnknownShapeOfRank(int64 rank);
|
| D | common_shape_fns.cc | 1176 c->set_output(0, c->UnknownShapeOfRank(c->Rank(input))); in ReductionShape()
1484 c->set_output(0, c->UnknownShapeOfRank(c->Rank(input))); in SliceShape()
|
| D | shape_inference.cc | 751 ShapeHandle InferenceContext::UnknownShapeOfRank(int64 rank) { in UnknownShapeOfRank() function in tensorflow::shape_inference::InferenceContext
|