Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/ops/
Dboosted_trees_ops.cc38 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anon770f674c0102()
65 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &node_id_range_shape)); in __anon770f674c0202()
70 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &summary_shape_base)); in __anon770f674c0202()
76 TF_RETURN_IF_ERROR(c->WithRank(c->input(1 + i), 3, &summary_shape)); in __anon770f674c0202()
81 c->WithRank(c->input(num_features + 1), 0, &unused_shape)); in __anon770f674c0202()
83 c->WithRank(c->input(num_features + 2), 0, &unused_shape)); in __anon770f674c0202()
85 c->WithRank(c->input(num_features + 3), 0, &unused_shape)); in __anon770f674c0202()
121 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &node_id_range_shape)); in __anon770f674c0302()
124 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 4, &unused_shape)); in __anon770f674c0302()
125 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_shape)); in __anon770f674c0302()
[all …]
Dmkl_nn_ops.cc68 TF_RETURN_IF_ERROR(c->WithRank(s, 5, &s)); in __anondbc33bd90102()
94 TF_RETURN_IF_ERROR(c->WithRank(s, 5, &s)); in __anondbc33bd90202()
134 TF_RETURN_IF_ERROR(c->WithRank(s, 4, &s)); in __anondbc33bd90302()
153 TF_RETURN_IF_ERROR(c->WithRank(s, 4, &s)); in __anondbc33bd90402()
643 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anondbc33bd90502()
644 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anondbc33bd90502()
689 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anondbc33bd90602()
690 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anondbc33bd90602()
691 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anondbc33bd90602()
692 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); in __anondbc33bd90602()
[all …]
Ddataset_ops.cc110 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anon42b7b0c00102()
122 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anon42b7b0c00202()
134 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anon42b7b0c00302()
191 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon42b7b0c00402()
294 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon42b7b0c00502()
295 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon42b7b0c00502()
296 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon42b7b0c00502()
297 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anon42b7b0c00502()
310 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon42b7b0c00602()
325 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon42b7b0c00702()
[all …]
Dimage_ops.cc34 TF_RETURN_IF_ERROR(c->WithRank(c->input(size_input_idx), 1, &size)); in SetOutputToSizedImage()
64 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in ResizeShapeFn()
71 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in DecodeImageShapeFn()
96 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in DecodeImageV2ShapeFn()
126 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &unused)); in EncodeImageShapeFn()
148 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &boxes)); in NMSShapeFn()
150 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &scores)); in NMSShapeFn()
152 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &max_output_size)); in NMSShapeFn()
154 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &iou_threshold)); in NMSShapeFn()
156 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &score_threshold)); in NMSShapeFn()
[all …]
Drnn_ops.cc39 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon652ace880102()
40 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon652ace880102()
69 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon652ace880202()
70 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon652ace880202()
71 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &w_ru)); in __anon652ace880202()
107 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon652ace880302()
108 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anon652ace880302()
145 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon652ace880402()
146 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anon652ace880402()
185 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &x)); in __anon652ace880502()
[all …]
Dexperimental_dataset_ops.cc29 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon4afff42d0102()
42 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &unused)); in __anon4afff42d0202()
55 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &unused)); in __anon4afff42d0302()
88 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &tag_shape)); in __anon4afff42d0402()
100 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &tag_shape)); in __anon4afff42d0502()
175 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon4afff42d0602()
176 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon4afff42d0602()
177 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon4afff42d0602()
178 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anon4afff42d0602()
179 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); in __anon4afff42d0602()
[all …]
Dio_ops.cc32 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 0, &unused)); in ScalarInputsAndOutputs()
44 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &handle)); in TwoElementVectorAndScalarOutputs()
73 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in __anon52e493120202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &s)); in __anon52e493120202()
95 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &shape0)); in __anon52e493120302()
96 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &shape1)); in __anon52e493120302()
97 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &shape2)); in __anon52e493120302()
139 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anon52e493120402()
140 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon52e493120402()
156 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in __anon52e493120502()
[all …]
Dsparse_ops.cc30 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &unused)); // a_indices in SparseSparseMinOrMaxShapeFn()
31 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &unused)); // a_values in SparseSparseMinOrMaxShapeFn()
32 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &unused)); // a_shape in SparseSparseMinOrMaxShapeFn()
33 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 2, &unused)); // b_indices in SparseSparseMinOrMaxShapeFn()
34 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 1, &unused)); // b_values in SparseSparseMinOrMaxShapeFn()
35 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 1, &unused)); // b_shape in SparseSparseMinOrMaxShapeFn()
55 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &a_indices)); in __anon36a8fc880202()
56 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &b_indices)); in __anon36a8fc880202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &a_shape)); in __anon36a8fc880302()
100 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &unused)); // a_indices in __anon36a8fc880402()
[all …]
Dtraining_ops.cc61 TF_RETURN_IF_ERROR(c->WithRank(c->input(grad_idx + 1), 1, &indices)); in HandleGradAndIndicesInputs()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); // alpha in ApplyGradientDescentShapeFn()
106 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); // alpha in ApplyProximalGradientDescentShapeFn()
107 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); // l1 in ApplyProximalGradientDescentShapeFn()
108 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); // l2 in ApplyProximalGradientDescentShapeFn()
175 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); // lr in ApplyAdadeltaShapeFn()
176 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); // rho in ApplyAdadeltaShapeFn()
177 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); // epsilon in ApplyAdadeltaShapeFn()
249 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); // lr in ApplyAdagradShapeFn()
311 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); // lr in ApplyAdagradV2ShapeFn()
[all …]
Dnn_ops.cc37 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in FractionalPoolShapeFn()
108 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in __anonad10f3360202()
113 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &vec)); in __anonad10f3360202()
140 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in __anonad10f3360302()
147 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &vec)); in __anonad10f3360302()
376 TF_RETURN_IF_ERROR(c->WithRank(s, 4, &s)); in __anonad10f3360402()
427 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in CommonFusedConvCalculations()
451 TF_RETURN_IF_ERROR(c->WithRank(c->input(paddings_index), 2, &paddings)); in CommonFusedConvCalculations()
453 c->WithRank(resized, c->Value(c->Dim(paddings, 0)), &resized)); in CommonFusedConvCalculations()
479 TF_RETURN_IF_ERROR(c->WithRank(c->input(filter_index), 4, &filter)); in CommonFusedConvCalculations()
[all …]
Dctc_ops.cc44 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anonbca2cdf40102()
45 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &labels_indices)); in __anonbca2cdf40102()
46 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &labels_values)); in __anonbca2cdf40102()
47 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &sequence_length)); in __anonbca2cdf40102()
81 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anonbca2cdf40202()
82 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &labels_indices)); in __anonbca2cdf40202()
83 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &labels_values)); in __anonbca2cdf40202()
84 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &sequence_length)); in __anonbca2cdf40202()
115 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anonbca2cdf40302()
116 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &sequence_length)); in __anonbca2cdf40302()
[all …]
Ddata_flow_ops.cc165 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &handle)); in TwoElementVectorInputsAndScalarOutputs()
412 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonea01102b0802()
436 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonea01102b0a02()
446 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonea01102b0b02()
466 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonea01102b0c02()
490 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonea01102b0e02()
500 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonea01102b0f02()
532 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonea01102b1102()
545 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonea01102b1202()
628 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in __anonea01102b1502()
[all …]
Dlookup_ops.cc36 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &handle)); in TwoElementVectorInputsAndScalarOutputs()
48 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in ScalarAndTwoElementVectorInputsAndScalarOutputs()
50 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &handle)); in ScalarAndTwoElementVectorInputsAndScalarOutputs()
79 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anon18b155f40202()
170 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon18b155f40302()
193 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anon18b155f40402()
209 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon18b155f40502()
221 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon18b155f40602()
248 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anon18b155f40702()
268 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon18b155f40802()
[all …]
Dstateful_random_ops.cc27 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in StatefulRandomShape()
63 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon0d38c5e20102()
64 Status s = c->WithRank(c->input(3), 0, &unused); in __anon0d38c5e20102()
70 s = c->WithRank(c->input(4), 0, &unused); in __anon0d38c5e20102()
89 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon0d38c5e20202()
90 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon0d38c5e20202()
101 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon0d38c5e20302()
102 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon0d38c5e20302()
Dstateless_random_ops_v2.cc30 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &key)); in StatelessShapeV2()
31 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &counter)); in StatelessShapeV2()
33 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused_shape)); in StatelessShapeV2()
73 Status s = c->WithRank(c->input(4), 0, &unused); in __anonea62c6790102()
79 s = c->WithRank(c->input(5), 0, &unused); in __anonea62c6790102()
107 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &seed)); in __anonea62c6790202()
127 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &seed)); in __anonea62c6790302()
Dmath_ops.cc1047 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &dimension_shape)); in ArgOpShape()
1126 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &segment_ids_shape)); in SegmentReductionShapeFn()
1143 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &indices_shape)); in SparseSegmentReductionShapeFn()
1146 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &segment_ids_shape)); in SparseSegmentReductionShapeFn()
1167 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &indices_shape)); in SparseSegmentReductionGradShapeFn()
1174 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in SparseSegmentReductionGradShapeFn()
1205 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &indices_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1208 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &segment_ids_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1211 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &num_segments_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1468 TF_RETURN_WITH_CONTEXT_IF_ERROR(c->WithRank(c->input(0), 0, &unused), in __anon725c8f640902()
[all …]
Dstring_ops.cc46 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon4354d5f10102()
47 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon4354d5f10102()
66 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon4354d5f10202()
192 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anon4354d5f10502()
193 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon4354d5f10502()
210 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anon4354d5f10602()
211 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon4354d5f10602()
266 TF_RETURN_IF_ERROR(c->WithRank(pos_shape, c->Rank(len_shape), &unused)); in __anon4354d5f10702()
300 TF_RETURN_IF_ERROR(c->WithRank(input_inner_values_shape, 1, &unused)); in __anon4354d5f10802()
304 TF_RETURN_IF_ERROR(c->WithRank(splits_shape, 1, &unused)); in __anon4354d5f10802()
[all …]
Dstateless_random_ops.cc28 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &seed)); in StatelessShape()
66 Status s = c->WithRank(c->input(2), 0, &unused); in __anona1237d120102()
72 s = c->WithRank(c->input(3), 0, &unused); in __anona1237d120102()
101 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &seed)); in __anona1237d120202()
108 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &logits_shape)); in __anona1237d120202()
109 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anona1237d120202()
141 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &seed)); in __anona1237d120302()
Dmkl_array_ops.cc54 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 0, &unused)); in __anon741deadc0102()
84 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 0, &unused)); in __anon741deadc0202()
121 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), minmax_rank, &minmax)); in __anon741deadc0302()
122 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), minmax_rank, &minmax)); in __anon741deadc0302()
152 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon741deadc0402()
153 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon741deadc0402()
Daudio_ops.cc31 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in DecodeWavShapeFn()
64 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &unused)); in EncodeWavShapeFn()
65 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in EncodeWavShapeFn()
72 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &input)); in SpectrogramShapeFn()
105 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &spectrogram)); in MfccShapeFn()
107 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in MfccShapeFn()
Dtpu_configuration_ops.cc88 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 0, &input)); in __anon6e35756a0102()
119 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &input)); in __anon6e35756a0202()
143 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &input)); in __anon6e35756a0302()
173 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &input)); in __anon6e35756a0402()
Darray_ops.cc84 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &paddings)); in PadShapeFn()
92 TF_RETURN_IF_ERROR(c->WithRank(input, c->Value(n_dim), &input)); in PadShapeFn()
111 TF_RETURN_IF_ERROR(c->WithRank(input, num_dims, &input)); in PadShapeFn()
153 TF_RETURN_IF_ERROR(c->WithRank(input, rank, &input)); in TransposeShapeFn()
155 TF_RETURN_IF_ERROR(c->WithRank(perm_shape, 1, &perm_shape)); in TransposeShapeFn()
457 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &dims)); in __anon42d741190602()
476 TF_RETURN_IF_ERROR(c->WithRank(shape_in, 1, &shape_in)); in __anon42d741190702()
868 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), c->Rank(input) - 1, &diag)); in __anon42d741191102()
973 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &dims)); in __anon42d741191302()
976 TF_RETURN_IF_ERROR(c->WithRank(input, c->Value(dims_dim), &input)); in __anon42d741191302()
[all …]
Dcheckpoint_ops.cc30 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in __anon62d746c20102()
31 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon62d746c20102()
61 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon62d746c20202()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
Dsentencepiece_tokenizer_op.cc45 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &unused)); in __anon18bebb200102()
46 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon18bebb200102()
47 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon18bebb200102()
48 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anon18bebb200102()
49 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); in __anon18bebb200102()
50 TF_RETURN_IF_ERROR(c->WithRank(c->input(6), 0, &unused)); in __anon18bebb200102()
/external/tensorflow/tensorflow/core/tpu/ops/
Dtopk_ops.cc30 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &input)); in __anonee73ab400102()
45 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &input)); in __anonee73ab400202()
62 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &input)); in __anonee73ab400302()
74 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &input)); in __anonee73ab400402()

123