Home
last modified time | relevance | path

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

/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 __anon59a9d6c30502()
206 TF_RETURN_IF_ERROR(c->WithRankAtLeast(sparse_matrix, 2, &sparse_matrix)); in __anon59a9d6c30602()
236 TF_RETURN_IF_ERROR(c->WithRankAtLeast(a_shape, 2, &a_shape)); in __anon59a9d6c30702()
242 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &b_shape)); in __anon59a9d6c30702()
360 TF_RETURN_IF_ERROR(c->WithRankAtLeast(a_shape, 2, &a_shape)); in __anon59a9d6c30902()
368 TF_RETURN_IF_ERROR(c->WithRankAtLeast(b_shape, 2, &b_shape)); in __anon59a9d6c30902()
394 TF_RETURN_IF_ERROR(c->WithRankAtLeast(a_shape, 2, &a_shape)); in __anon59a9d6c30a02()
402 TF_RETURN_IF_ERROR(c->WithRankAtLeast(b_shape, 2, &b_shape)); in __anon59a9d6c30a02()
486 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, 2, &input)); in __anon59a9d6c30c02()
[all …]
Dset_ops.cc55 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anonda6063e60102()
77 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input1_shape, 2, &input1_shape)); in __anonda6063e60102()
117 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anonda6063e60202()
Dcollective_ops.cc52 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anon52c2f9040102()
138 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anon52c2f9040202()
Darray_ops.cc588 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); in __anon42d741190a02()
649 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); in __anon42d741190b02()
774 TF_RETURN_IF_ERROR(c->WithRankAtLeast(in, 1, &in)); in __anon42d741190e02()
820 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &in)); in __anon42d741191002()
865 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in __anon42d741191102()
866 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &diag)); in __anon42d741191102()
915 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &in)); in __anon42d741191202()
1165 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anon42d741191702()
1187 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &params_shape)); in __anon42d741191802()
1216 TF_RETURN_IF_ERROR(c->WithRankAtLeast( in __anon42d741191802()
[all …]
Dmanip_ops.cc34 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anonf8805fef0102()
Dencode_proto_ops.cc39 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, 2, &input)); in __anona4faf29b0102()
Dragged_conversion_ops.cc167 TF_RETURN_IF_ERROR(c->WithRankAtLeast(rt_dense_values, 1, &rt_dense_values)); in RaggedTensorToSparseShapeFn()
194 TF_RETURN_IF_ERROR(c->WithRankAtLeast(rt_dense_values, 1, &rt_dense_values)); in RaggedTensorToVariantShapeFn()
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()
Dmath_ops.cc1125 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SegmentReductionShapeFn()
1140 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SparseSegmentReductionShapeFn()
1164 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SparseSegmentReductionGradShapeFn()
1202 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1588 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &a_shape)); in __anon725c8f640c02()
1589 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &b_shape)); in __anon725c8f640c02()
1936 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon725c8f641702()
Dresource_variable_ops.cc259 TF_RETURN_IF_ERROR(c->WithRankAtLeast(handle_shape_and_type[0].shape, in __anon50d007a70402()
263 c->WithRankAtLeast(indices_shape, batch_dims, &unused)); in __anon50d007a70402()
Dmkl_array_ops.cc125 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), axis + 1, &input)); in __anon741deadc0302()
Dragged_array_ops.cc142 c->WithRankAtLeast(params_dense_values, 1, &params_dense_values)); in RaggedGatherShapeFn()
Dlist_ops.cc123 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &tensor)); in __anonabedecb60402()
331 TF_RETURN_IF_ERROR(c->WithRankAtLeast(tensor_shape, 1, &ignored)); in __anonabedecb60902()
571 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &ignored)); in __anonabedecb61302()
Dspectral_ops.cc77 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), rank, &out)); in RFFTShape()
Dlookup_ops.cc222 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &handle)); in __anon18b155f40602()
253 TF_RETURN_IF_ERROR(c->WithRankAtLeast(values, 1, &values)); in __anon18b155f40702()
Dimage_ops.cc133 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in ColorspaceShapeFn()
Dnn_ops.cc1295 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in TopKShapeFn()
1359 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anonad10f3362002()
Ddata_flow_ops.cc1176 TF_RETURN_IF_ERROR(c->WithRankAtLeast(values, 1, &values)); in __anonea01102b2b02()
/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()
1270 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input_shape)); in MatrixDiagPartV2Shape()
[all …]
Dcommon_shape_fns.h62 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), rank, &out)); in UnchangedShapeWithRankAtLeast()
Dshape_inference_test.cc404 TEST_F(ShapeInferenceTest, WithRankAtLeast) { in TEST_F() argument
415 EXPECT_TRUE(c.WithRankAtLeast(in0, 1, &s1).ok()); in TEST_F()
419 EXPECT_TRUE(c.WithRankAtLeast(in0, 2, &s2).ok()); in TEST_F()
426 c.WithRankAtLeast(in1, 4, &s1).ToString(), in TEST_F()
430 EXPECT_TRUE(c.WithRankAtLeast(in1, 3, &s1).ok()); in TEST_F()
432 EXPECT_TRUE(c.WithRankAtLeast(in1, 2, &s1).ok()); in TEST_F()
434 EXPECT_TRUE(c.WithRankAtLeast(in1, 0, &s1).ok()); in TEST_F()
Dshape_inference.h382 Status WithRankAtLeast(ShapeHandle shape, int64 rank,
Dshape_inference.cc333 Status InferenceContext::WithRankAtLeast(ShapeHandle shape, int64 rank, in WithRankAtLeast() function in tensorflow::shape_inference::InferenceContext
438 TF_RETURN_IF_ERROR(WithRankAtLeast(s, rank, &s)); in MergePrefix()
/external/tensorflow/tensorflow/c/
Dops.cc218 DEFINE_RANK_FUNC(WithRankAtLeast)