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 __anon130e45de0302()
|
| 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 __anon7c94107b0a02()
637 c->set_output(i, c->UnknownShapeOfRank(rank)); in __anon7c94107b0b02()
853 c->Concatenate(diag_prefix, c->UnknownShapeOfRank(2), &diag)); in __anon7c94107b1102()
1130 c->set_output(0, c->UnknownShapeOfRank(c->Rank(params_shape) + in __anon7c94107b1802()
1855 c->set_output(0, c->UnknownShapeOfRank(input_rank)); in __anon7c94107b2c02()
|
| D | lookup_ops.cc | 271 ShapeHandle keys = c->UnknownShapeOfRank(1); in __anon8a0811960802()
|
| D | math_ops.cc | 1510 c->set_output(0, c->UnknownShapeOfRank(1)); in __anonb22bfa860d02()
1529 c->set_output(0, c->UnknownShapeOfRank(1)); in __anonb22bfa860e02()
|
| D | nn_ops.cc | 394 padded = c->UnknownShapeOfRank(4); in CommonFusedConvCalculations()
1346 c->set_output(0, c->UnknownShapeOfRank(4)); in __anon3e672dd82802()
|
| /external/tensorflow/tensorflow/compiler/xrt/ops/ |
| D | xrt_compile_ops.cc | 29 c->set_output(1, c->UnknownShapeOfRank(1)); in __anonbe2fd7610102()
|
| /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 __anonc06300730302()
|
| /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 __anon7e18408e0102()
|
| /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
|