Home
last modified time | relevance | path

Searched refs:WithRankAtLeast (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/
Dlinalg_ops.cc32 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, 2, &s)); in MakeBatchSquareMatrix()
55 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &lhs)); in BandedTriangularSolveShapeFn()
56 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &rhs)); in BandedTriangularSolveShapeFn()
101 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &lhs)); in MatrixSolveShapeFn()
103 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &rhs)); in MatrixSolveShapeFn()
136 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &rhs)); in MatrixTriangularSolveShapeFn()
188 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in LuShapeFn()
214 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in QrShapeFn()
245 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in SvdShapeFn()
291 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &superdiag)); in TridiagonalMatMulShapeFn()
[all …]
Dsparse_csr_matrix_ops.cc48 TF_RETURN_IF_ERROR(c->WithRankAtLeast(matrix_shape, 2, &out)); in ValidateSquareMatrixShape()
179 c->WithRankAtLeast(csr_sparse_matrix, 2, &csr_sparse_matrix)); in __anon6adce4610502()
206 TF_RETURN_IF_ERROR(c->WithRankAtLeast(sparse_matrix, 2, &sparse_matrix)); in __anon6adce4610602()
236 TF_RETURN_IF_ERROR(c->WithRankAtLeast(a_shape, 2, &a_shape)); in __anon6adce4610702()
242 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &b_shape)); in __anon6adce4610702()
360 TF_RETURN_IF_ERROR(c->WithRankAtLeast(a_shape, 2, &a_shape)); in __anon6adce4610902()
368 TF_RETURN_IF_ERROR(c->WithRankAtLeast(b_shape, 2, &b_shape)); in __anon6adce4610902()
394 TF_RETURN_IF_ERROR(c->WithRankAtLeast(a_shape, 2, &a_shape)); in __anon6adce4610a02()
402 TF_RETURN_IF_ERROR(c->WithRankAtLeast(b_shape, 2, &b_shape)); in __anon6adce4610a02()
486 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, 2, &input)); in __anon6adce4610c02()
[all …]
Dset_ops.cc55 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anon1e42d5040102()
77 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input1_shape, 2, &input1_shape)); in __anon1e42d5040102()
117 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anon1e42d5040202()
Darray_ops.cc589 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); in __anon091a71b70a02()
650 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); in __anon091a71b70b02()
792 TF_RETURN_IF_ERROR(c->WithRankAtLeast(in, 1, &in)); in __anon091a71b70e02()
838 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &in)); in __anon091a71b71002()
883 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in __anon091a71b71102()
884 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &diag)); in __anon091a71b71102()
933 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &in)); in __anon091a71b71202()
1184 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anon091a71b71702()
1206 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &params_shape)); in __anon091a71b71802()
1238 TF_RETURN_IF_ERROR(c->WithRankAtLeast( in __anon091a71b71802()
[all …]
Dcollective_ops.cc54 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anone3133c620102()
161 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anone3133c620302()
Dmanip_ops.cc34 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anonbec3908d0102()
Dencode_proto_ops.cc39 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, 2, &input)); in __anon8a6985790102()
Dstate_ops.cc143 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &indices_shape)); in ScatterNdUpdateShape()
145 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(2), 1, &updates_shape)); in ScatterNdUpdateShape()
Dragged_conversion_ops.cc168 TF_RETURN_IF_ERROR(c->WithRankAtLeast(rt_dense_values, 1, &rt_dense_values)); in RaggedTensorToSparseShapeFn()
195 TF_RETURN_IF_ERROR(c->WithRankAtLeast(rt_dense_values, 1, &rt_dense_values)); in RaggedTensorToVariantShapeFn()
Drandom_index_shuffle_ops.cc35 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &seed_shape)); in StatelessRandomPermuteShape()
Dresource_variable_ops.cc275 TF_RETURN_IF_ERROR(c->WithRankAtLeast(handle_shape_and_type[0].shape, in __anon620315450402()
279 c->WithRankAtLeast(indices_shape, batch_dims, &unused)); in __anon620315450402()
Dspectral_ops.cc77 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), rank, &out)); in RFFTShape()
Dmath_ops.cc1158 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SegmentReductionShapeFn()
1173 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SparseSegmentReductionShapeFn()
1197 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SparseSegmentReductionGradShapeFn()
1235 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1646 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &a_shape)); in __anon328d24420c02()
1647 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &b_shape)); in __anon328d24420c02()
Dlist_ops.cc133 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &tensor)); in __anon6c1e81940402()
347 TF_RETURN_IF_ERROR(c->WithRankAtLeast(tensor_shape, 1, &ignored)); in __anon6c1e81940902()
613 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &ignored)); in __anon6c1e81941302()
Dragged_array_ops.cc156 c->WithRankAtLeast(params_dense_values, 1, &params_dense_values)); in RaggedGatherShapeFn()
Dlookup_ops.cc222 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &handle)); in __anona75a9a520602()
253 TF_RETURN_IF_ERROR(c->WithRankAtLeast(values, 1, &values)); in __anona75a9a520702()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2022-007.md37 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, min_rank, &input));
40 However, [`WithRankAtLeast`](https://github.com/tensorflow/tensorflow/blob/5100e359aef5c8021f2e71c7…
43 Status InferenceContext::WithRankAtLeast(ShapeHandle shape, int64_t rank,
Dtfsa-2022-004.md42 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), axis + 1, &input));
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc192 c->WithRankAtLeast(input_shape, num_named_labels, &input_shape), in EinsumShape()
312 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &a_shape)); in BatchMatMulV2Shape()
313 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &b_shape)); in BatchMatMulV2Shape()
349 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &a_shape)); in BatchMatMulShape()
350 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &b_shape)); in BatchMatMulShape()
390 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 3, &input_shape)); in BiasAddShape()
392 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input_shape)); in BiasAddShape()
448 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 3, &input_shape)); in BiasAddGradShape()
451 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input_shape)); in BiasAddGradShape()
1361 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input_shape)); in MatrixDiagPartV2Shape()
[all …]
Dshape_inference_test.cc415 TEST_F(ShapeInferenceTest, WithRankAtLeast) { in TEST_F() argument
426 TF_EXPECT_OK(c.WithRankAtLeast(in0, 1, &s1)); in TEST_F()
430 TF_EXPECT_OK(c.WithRankAtLeast(in0, 2, &s2)); in TEST_F()
437 c.WithRankAtLeast(in1, 4, &s1), in TEST_F()
442 TF_EXPECT_OK(c.WithRankAtLeast(in1, 3, &s1)); in TEST_F()
444 TF_EXPECT_OK(c.WithRankAtLeast(in1, 2, &s1)); in TEST_F()
446 TF_EXPECT_OK(c.WithRankAtLeast(in1, 0, &s1)); in TEST_F()
Dcommon_shape_fns.h62 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), rank, &out)); in UnchangedShapeWithRankAtLeast()
Dshape_inference.cc340 Status InferenceContext::WithRankAtLeast(ShapeHandle shape, int64_t rank, in WithRankAtLeast() function in tensorflow::shape_inference::InferenceContext
445 TF_RETURN_IF_ERROR(WithRankAtLeast(s, rank, &s)); in MergePrefix()
Dshape_inference.h467 Status WithRankAtLeast(ShapeHandle shape, int64_t rank,
/external/tensorflow/tensorflow/c/
Dops.cc218 DEFINE_RANK_FUNC(WithRankAtLeast)
/external/tensorflow/tensorflow/core/tpu/ops/
Dtpu_embedding_ops.cc292 TF_RETURN_IF_ERROR(c->WithRankAtLeast(accumulators[i][table_id], 1, in __anon8e82fb570702()

12