| /external/tensorflow/tensorflow/core/ops/ |
| D | rpc_ops.cc | 29 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &address)); in RpcShapeOp()
33 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &method)); in RpcShapeOp()
37 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &request)); in RpcShapeOp()
|
| D | ragged_math_ops.cc | 49 TF_RETURN_IF_ERROR(c->WithRankAtMost(starts, 1, &starts)); in RaggedRangeShapeFn()
50 TF_RETURN_IF_ERROR(c->WithRankAtMost(limits, 1, &limits)); in RaggedRangeShapeFn()
51 TF_RETURN_IF_ERROR(c->WithRankAtMost(deltas, 1, &deltas)); in RaggedRangeShapeFn()
|
| D | random_ops.cc | 78 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &unused)); in __anon979c5d7d0202()
79 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &unused)); in __anon979c5d7d0202()
80 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(3), 1, &unused)); in __anon979c5d7d0202()
81 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(4), 1, &unused)); in __anon979c5d7d0202()
|
| D | manip_ops.cc | 36 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &unused)); in __anon323d2f510102()
38 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &unused)); in __anon323d2f510102()
|
| D | experimental_dataset_ops.cc | 73 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anondb35634f0202()
87 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(i), 1, &v)); in __anondb35634f0202()
249 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anondb35634f0602()
340 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &unused)); in __anondb35634f0802()
341 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 0, &unused)); in __anondb35634f0802()
|
| D | dataset_ops.cc | 412 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anondee5e0a20f02()
432 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anondee5e0a21002()
455 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anondee5e0a21102()
475 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anondee5e0a21202()
|
| D | lookup_ops.cc | 85 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &unused)); in __anon8a0811960202()
171 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &keys)); in __anon8a0811960302()
321 TF_RETURN_IF_ERROR(c->WithRankAtMost(key, 1, &key_s)); in MutableHashTableShape()
|
| D | parsing_ops.cc | 349 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(i), 1, &v)); in __anon952516500602()
|
| D | io_ops.cc | 489 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anone43acdb40e02()
|
| D | array_ops.cc | 487 TF_RETURN_IF_ERROR(c->WithRankAtMost(in, out_rank, &in)); in __anon7c94107b0702()
3121 TF_RETURN_IF_ERROR(c->WithRankAtMost(inputs, 4, &inputs)); in __anon7c94107b4a02()
|
| /external/tensorflow/tensorflow/compiler/xrt/ops/ |
| D | xrt_execute_op.cc | 35 c->WithRankAtMost(input_handle_shapes[i], 1, &unused)); in __anon8fb216580102()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 138 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 6, &h)); in TEST_F()
149 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &h)); in TEST_F()
175 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &h)); in TEST_F()
196 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &h)); in TEST_F()
219 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &h)); in TEST_F()
242 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &h)); in TEST_F()
349 TEST_F(ShapeInferenceTest, WithRankAtMost) { in TEST_F() argument
360 EXPECT_TRUE(c.WithRankAtMost(in0, 1, &s1).ok()); in TEST_F()
364 EXPECT_TRUE(c.WithRankAtMost(in0, 2, &s2).ok()); in TEST_F()
371 c.WithRankAtMost(in1, 2, &s1).ToString(), in TEST_F()
[all …]
|
| D | common_shape_fns.h | 194 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), rank, &out)); in UnchangedShapeWithRankAtMost()
|
| D | shape_inference.cc | 406 Status InferenceContext::WithRankAtMost(ShapeHandle shape, int64 rank, in WithRankAtMost() function in tensorflow::shape_inference::InferenceContext
778 TF_RETURN_IF_ERROR(WithRankAtMost(input(input_idx), 1, &input_shape)); in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
|
| D | shape_inference.h | 398 Status WithRankAtMost(ShapeHandle shape, int64 rank,
|
| D | common_shape_fns.cc | 1163 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &indices)); in ReductionShape()
|
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/ |
| D | reduce_slice_ops.cc | 39 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 2, &handle)); in ReduceSliceShapeFn()
|