Home
last modified time | relevance | path

Searched refs:WithRankAtMost (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dsparse_csr_matrix_ops.cc49 TF_RETURN_IF_ERROR(c->WithRankAtMost(matrix_shape, 3, &out)); in ValidateSquareMatrixShape()
97 TF_RETURN_IF_ERROR(c->WithRankAtMost(sparse_matrix, 3, &sparse_matrix)); in __anon59a9d6c30202()
159 TF_RETURN_IF_ERROR(c->WithRankAtMost(sparse_matrix, 3, &sparse_matrix)); in __anon59a9d6c30402()
181 c->WithRankAtMost(csr_sparse_matrix, 3, &csr_sparse_matrix)); in __anon59a9d6c30502()
207 TF_RETURN_IF_ERROR(c->WithRankAtMost(sparse_matrix, 3, &sparse_matrix)); in __anon59a9d6c30602()
237 TF_RETURN_IF_ERROR(c->WithRankAtMost(a_shape, 3, &a_shape)); in __anon59a9d6c30702()
243 TF_RETURN_IF_ERROR(c->WithRankAtMost(b_shape, 3, &b_shape)); in __anon59a9d6c30702()
307 TF_RETURN_IF_ERROR(c->WithRankAtMost(a_shape, 3, &a_shape)); in __anon59a9d6c30802()
312 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 3, &b_shape)); in __anon59a9d6c30802()
361 TF_RETURN_IF_ERROR(c->WithRankAtMost(a_shape, 3, &a_shape)); in __anon59a9d6c30902()
[all …]
Dmkl_nn_ops.cc740 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(4), 1, &unused)); in __anondbc33bd90702()
741 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(5), 1, &unused)); in __anondbc33bd90702()
788 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(5), 1, &channel)); in __anondbc33bd90802()
789 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(6), 1, &channel)); in __anondbc33bd90802()
839 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(5), 1, &unused)); in __anondbc33bd90902()
840 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(6), 1, &unused)); in __anondbc33bd90902()
881 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(4), 1, &channel)); in __anondbc33bd90a02()
882 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(5), 1, &channel)); in __anondbc33bd90a02()
927 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(4), 1, &unused)); in __anondbc33bd90b02()
928 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(5), 1, &unused)); in __anondbc33bd90b02()
[all …]
Dragged_math_ops.cc50 TF_RETURN_IF_ERROR(c->WithRankAtMost(starts, 1, &starts)); in RaggedRangeShapeFn()
51 TF_RETURN_IF_ERROR(c->WithRankAtMost(limits, 1, &limits)); in RaggedRangeShapeFn()
52 TF_RETURN_IF_ERROR(c->WithRankAtMost(deltas, 1, &deltas)); in RaggedRangeShapeFn()
Drandom_ops.cc88 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &unused)); in __anon2645a1db0202()
89 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &unused)); in __anon2645a1db0202()
90 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(3), 1, &unused)); in __anon2645a1db0202()
91 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(4), 1, &unused)); in __anon2645a1db0202()
Dmanip_ops.cc36 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &unused)); in __anonf8805fef0102()
38 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &unused)); in __anonf8805fef0102()
Dexperimental_dataset_ops.cc172 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anon4afff42d0602()
186 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(i), 1, &v)); in __anon4afff42d0602()
215 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anon4afff42d0702()
232 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(i), 1, &v)); in __anon4afff42d0702()
260 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anon4afff42d0802()
274 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(i), 1, &v)); in __anon4afff42d0802()
578 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anon4afff42d0f02()
590 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anon4afff42d1002()
880 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &unused)); in __anon4afff42d1402()
881 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 0, &unused)); in __anon4afff42d1402()
[all …]
Dparsing_ops.cc209 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &input)); in __anonf8f6e66e0502()
211 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &names)); in __anonf8f6e66e0502()
368 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &input)); in __anonf8f6e66e0802()
370 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &names)); in __anonf8f6e66e0802()
483 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(i), 1, &v)); in __anonf8f6e66e0a02()
Dnn_ops.cc2743 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(5), 1, &channel)); in __anonad10f3363702()
2744 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(6), 1, &channel)); in __anonad10f3363702()
2777 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(5), 1, &channel)); in __anonad10f3363802()
2778 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(6), 1, &channel)); in __anonad10f3363802()
2809 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(4), 1, &channel)); in __anonad10f3363902()
2810 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(5), 1, &channel)); in __anonad10f3363902()
2840 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(4), 1, &channel)); in __anonad10f3363a02()
2841 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(5), 1, &channel)); in __anonad10f3363a02()
2874 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(5), 1, &channel)); in __anonad10f3363b02()
2875 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(6), 1, &channel)); in __anonad10f3363b02()
[all …]
Ddataset_ops.cc628 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anon42b7b0c01902()
648 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anon42b7b0c01a02()
671 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anon42b7b0c01b02()
691 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anon42b7b0c01c02()
Dlookup_ops.cc85 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &unused)); in __anon18b155f40202()
321 TF_RETURN_IF_ERROR(c->WithRankAtMost(key, 1, &key_s)); in MutableHashTableShape()
Dboosted_trees_ops.cc469 c->WithRankAtMost(c->input(i + 1), 2, &feature_shape)); in __anon770f674c0c02()
498 c->WithRankAtMost(c->input(i + 1), 2, &feature_shape)); in __anon770f674c0d02()
542 c->WithRankAtMost(c->input(i + 3), 2, &feature_shape)); in __anon770f674c0f02()
Dio_ops.cc490 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 1, &unused)); in __anon52e493120e02()
Dmath_ops.cc1680 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 2, &unused)); in __anon725c8f640f02()
Darray_ops.cc493 TF_RETURN_IF_ERROR(c->WithRankAtMost(in, out_rank, &in)); in __anon42d741190702()
3299 TF_RETURN_IF_ERROR(c->WithRankAtMost(inputs, 4, &inputs)); in __anon42d741194c02()
/external/tensorflow/tensorflow/compiler/xrt/ops/
Dxrt_execute_op.cc35 c->WithRankAtMost(input_handle_shapes[i], 1, &unused)); in __anonc746d7b60102()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc146 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 6, &h)); in TEST_F()
157 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &h)); in TEST_F()
182 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &h)); in TEST_F()
204 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &h)); in TEST_F()
230 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &h)); in TEST_F()
256 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), 0, &h)); in TEST_F()
366 TEST_F(ShapeInferenceTest, WithRankAtMost) { in TEST_F() argument
377 EXPECT_TRUE(c.WithRankAtMost(in0, 1, &s1).ok()); in TEST_F()
381 EXPECT_TRUE(c.WithRankAtMost(in0, 2, &s2).ok()); in TEST_F()
388 c.WithRankAtMost(in1, 2, &s1).ToString(), in TEST_F()
[all …]
Dcommon_shape_fns.h71 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(0), rank, &out)); in UnchangedShapeWithRankAtMost()
Dcommon_shape_fns.cc277 c->WithRankAtMost(output_bcast_shape, 0, &output_bcast_shape), in EinsumShape()
1271 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &diag_index_shape)); in MatrixDiagPartV2Shape()
1325 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &diag_index_shape)); in MatrixDiagV2Shape()
1426 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(2), 1, &diag_index_shape)); in MatrixSetDiagV2Shape()
1456 c->WithRankAtMost(c->input(1), input_rank, &diag_shape)); in MatrixSetDiagV2Shape()
1837 TF_RETURN_IF_ERROR(c->WithRankAtMost(c->input(1), 1, &indices)); in ReductionShape()
Dshape_inference.h384 Status WithRankAtMost(ShapeHandle shape, int64 rank,
Dshape_inference.cc348 Status InferenceContext::WithRankAtMost(ShapeHandle shape, int64 rank, in WithRankAtMost() function in tensorflow::shape_inference::InferenceContext
720 TF_RETURN_IF_ERROR(WithRankAtMost(input(input_idx), 1, &input_shape)); in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
/external/tensorflow/tensorflow/c/
Dops.cc219 DEFINE_RANK_FUNC(WithRankAtMost)