Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/ops/
Dmkl_nn_ops.cc191 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonfe9dedfd0102()
192 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anonfe9dedfd0102()
237 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anonfe9dedfd0202()
238 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anonfe9dedfd0202()
239 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anonfe9dedfd0202()
240 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); in __anonfe9dedfd0202()
286 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anonfe9dedfd0302()
287 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anonfe9dedfd0302()
290 TF_RETURN_IF_ERROR(c->WithRank(c->input(6), 0, &unused)); in __anonfe9dedfd0302()
291 TF_RETURN_IF_ERROR(c->WithRank(c->input(7), 0, &unused)); in __anonfe9dedfd0302()
[all …]
Dboosted_trees_ops.cc38 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anon9fefc8f00102()
65 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &node_id_range_shape)); in __anon9fefc8f00202()
70 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &summary_shape_base)); in __anon9fefc8f00202()
76 TF_RETURN_IF_ERROR(c->WithRank(c->input(1 + i), 3, &summary_shape)); in __anon9fefc8f00202()
81 c->WithRank(c->input(num_features + 1), 0, &unused_shape)); in __anon9fefc8f00202()
83 c->WithRank(c->input(num_features + 2), 0, &unused_shape)); in __anon9fefc8f00202()
85 c->WithRank(c->input(num_features + 3), 0, &unused_shape)); in __anon9fefc8f00202()
121 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &node_id_range_shape)); in __anon9fefc8f00302()
124 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 4, &unused_shape)); in __anon9fefc8f00302()
125 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_shape)); in __anon9fefc8f00302()
[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 __anon5983aeac0202()
56 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &b_indices)); in __anon5983aeac0202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &a_shape)); in __anon5983aeac0302()
100 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &unused)); // a_indices in __anon5983aeac0402()
[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()
92 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &unused)); in EncodeImageShapeFn()
114 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &boxes)); in NMSShapeFn()
116 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &scores)); in NMSShapeFn()
118 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &max_output_size)); in NMSShapeFn()
120 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &iou_threshold)); in NMSShapeFn()
122 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &score_threshold)); in NMSShapeFn()
137 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &boxes)); in SoftNMSShapeFn()
[all …]
Dexperimental_dataset_ops.cc29 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &unused)); in __anon03b595510102()
42 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &unused)); in __anon03b595510202()
74 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &tag_shape)); in __anon03b595510302()
86 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &tag_shape)); in __anon03b595510402()
143 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon03b595510502()
144 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon03b595510502()
145 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon03b595510502()
146 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anon03b595510502()
147 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); in __anon03b595510502()
148 TF_RETURN_IF_ERROR(c->WithRank(c->input(6), 0, &unused)); in __anon03b595510502()
[all …]
Drnn_ops.cc39 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon95c2132c0102()
40 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon95c2132c0102()
69 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon95c2132c0202()
70 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon95c2132c0202()
71 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &w_ru)); in __anon95c2132c0202()
107 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon95c2132c0302()
108 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anon95c2132c0302()
145 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon95c2132c0402()
146 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anon95c2132c0402()
185 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &x)); in __anon95c2132c0502()
[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 __anon2d93d3360202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &s)); in __anon2d93d3360202()
95 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &shape0)); in __anon2d93d3360302()
96 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &shape1)); in __anon2d93d3360302()
97 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &shape2)); in __anon2d93d3360302()
139 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anon2d93d3360402()
140 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon2d93d3360402()
156 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in __anon2d93d3360502()
[all …]
Ddataset_ops.cc110 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anonc0e8a7640102()
122 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anonc0e8a7640202()
134 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anonc0e8a7640302()
175 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonc0e8a7640402()
260 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonc0e8a7640502()
261 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anonc0e8a7640502()
262 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anonc0e8a7640502()
263 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anonc0e8a7640502()
276 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonc0e8a7640602()
291 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonc0e8a7640702()
[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.cc35 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in FractionalPoolShapeFn()
92 TF_RETURN_IF_ERROR(c->WithRank(s, 4, &s)); in __anon87c0335a0202()
112 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in __anon87c0335a0302()
117 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &vec)); in __anon87c0335a0302()
144 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in __anon87c0335a0402()
151 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &vec)); in __anon87c0335a0402()
357 TF_RETURN_IF_ERROR(c->WithRank(s, 4, &s)); in __anon87c0335a0502()
380 TF_RETURN_IF_ERROR(c->WithRank(s, 4, &s)); in __anon87c0335a0602()
412 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in CommonFusedConvCalculations()
436 TF_RETURN_IF_ERROR(c->WithRank(c->input(paddings_index), 2, &paddings)); in CommonFusedConvCalculations()
[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 __anonb84a3fcf0802()
436 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonb84a3fcf0a02()
446 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonb84a3fcf0b02()
466 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonb84a3fcf0c02()
490 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonb84a3fcf0e02()
500 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonb84a3fcf0f02()
532 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonb84a3fcf1102()
545 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonb84a3fcf1202()
628 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in __anonb84a3fcf1502()
[all …]
Dctc_ops.cc44 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anoned3a12980102()
45 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &labels_indices)); in __anoned3a12980102()
46 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &labels_values)); in __anoned3a12980102()
47 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &sequence_length)); in __anoned3a12980102()
81 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anoned3a12980202()
82 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &labels_indices)); in __anoned3a12980202()
83 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &labels_values)); in __anoned3a12980202()
84 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &sequence_length)); in __anoned3a12980202()
115 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anoned3a12980302()
116 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &sequence_length)); in __anoned3a12980302()
[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 __anon3b8c08180202()
167 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon3b8c08180302()
195 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anon3b8c08180402()
211 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon3b8c08180502()
223 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon3b8c08180602()
250 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anon3b8c08180702()
270 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon3b8c08180802()
[all …]
Dstateless_random_ops.cc28 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &seed)); in StatelessShape()
66 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anond38e2c360102()
67 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anond38e2c360102()
82 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &seed)); in __anond38e2c360202()
89 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &logits_shape)); in __anond38e2c360202()
90 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anond38e2c360202()
Dmath_ops.cc1030 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &dimension_shape)); in ArgOpShape()
1109 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &segment_ids_shape)); in SegmentReductionShapeFn()
1126 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &indices_shape)); in SparseSegmentReductionShapeFn()
1129 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &segment_ids_shape)); in SparseSegmentReductionShapeFn()
1150 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &indices_shape)); in SparseSegmentReductionGradShapeFn()
1157 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in SparseSegmentReductionGradShapeFn()
1188 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &indices_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1191 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &segment_ids_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1194 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &num_segments_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1439 TF_RETURN_WITH_CONTEXT_IF_ERROR(c->WithRank(c->input(0), 0, &unused), in __anonb5dc68880902()
[all …]
Dstring_ops.cc46 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon662f88150102()
47 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon662f88150102()
66 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon662f88150202()
192 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anon662f88150502()
193 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon662f88150502()
210 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anon662f88150602()
211 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon662f88150602()
266 TF_RETURN_IF_ERROR(c->WithRank(pos_shape, c->Rank(len_shape), &unused)); in __anon662f88150702()
300 TF_RETURN_IF_ERROR(c->WithRank(input_inner_values_shape, 1, &unused)); in __anon662f88150802()
304 TF_RETURN_IF_ERROR(c->WithRank(splits_shape, 1, &unused)); in __anon662f88150802()
[all …]
Dstateful_random_ops.cc26 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in StatefulRandomShape()
62 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonefb820860102()
63 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anonefb820860102()
64 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anonefb820860102()
78 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonefb820860202()
79 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anonefb820860202()
Dmkl_array_ops.cc54 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 0, &unused)); in __anon42671a800102()
84 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 0, &unused)); in __anon42671a800202()
121 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), minmax_rank, &minmax)); in __anon42671a800302()
122 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), minmax_rank, &minmax)); in __anon42671a800302()
149 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon42671a800402()
150 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon42671a800402()
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 __anonedf6090e0102()
119 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &input)); in __anonedf6090e0202()
143 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &input)); in __anonedf6090e0302()
173 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &input)); in __anonedf6090e0402()
Darray_ops.cc83 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &paddings)); in PadShapeFn()
91 TF_RETURN_IF_ERROR(c->WithRank(input, c->Value(n_dim), &input)); in PadShapeFn()
110 TF_RETURN_IF_ERROR(c->WithRank(input, num_dims, &input)); in PadShapeFn()
152 TF_RETURN_IF_ERROR(c->WithRank(input, rank, &input)); in TransposeShapeFn()
154 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 __anonf6523ebd0602()
476 TF_RETURN_IF_ERROR(c->WithRank(shape_in, 1, &shape_in)); in __anonf6523ebd0702()
868 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), c->Rank(input) - 1, &diag)); in __anonf6523ebd1102()
973 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &dims)); in __anonf6523ebd1302()
976 TF_RETURN_IF_ERROR(c->WithRank(input, c->Value(dims_dim), &input)); in __anonf6523ebd1302()
[all …]
Dcheckpoint_ops.cc30 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in __anonfa466ae60102()
31 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonfa466ae60102()
61 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonfa466ae60202()
Dtensor_forest_ops.cc32 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anon8b5112eb0102()
52 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anon8b5112eb0202()
70 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &shape_handle)); in __anon8b5112eb0302()
Drandom_ops.cc48 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon492053ff0102()
49 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon492053ff0102()
117 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &logits_shape)); in __anon492053ff0302()
118 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon492053ff0302()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc224 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &a)); in MatMulShape()
227 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &b)); in MatMulShape()
308 c->WithRank(input_shape, num_named_labels, &input_shape), " for ", in EinsumShape()
505 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &bias_shape)); in BiasAddShape()
689 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), rank, &conv_input_shape)); in Conv2DShapeImpl()
695 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), rank, &filter_shape)); in Conv2DShapeImpl()
829 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 5, &input_shape)); in Conv3DShape()
831 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 5, &filter_shape)); in Conv3DShape()
939 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input_shape)); in DepthwiseConv2DNativeShape()
941 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 4, &filter_shape)); in DepthwiseConv2DNativeShape()
[all …]

12