Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/ops/
Dmkl_nn_ops.cc166 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon4d19f77b0102()
167 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon4d19f77b0102()
213 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon4d19f77b0202()
214 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon4d19f77b0202()
215 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anon4d19f77b0202()
216 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); in __anon4d19f77b0202()
259 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon4d19f77b0302()
260 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon4d19f77b0302()
261 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anon4d19f77b0302()
262 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); in __anon4d19f77b0302()
[all …]
Dboosted_trees_ops.cc37 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anone3ce78ae0102()
64 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &node_id_range_shape)); in __anone3ce78ae0202()
69 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &summary_shape_base)); in __anone3ce78ae0202()
75 TF_RETURN_IF_ERROR(c->WithRank(c->input(1 + i), 3, &summary_shape)); in __anone3ce78ae0202()
80 c->WithRank(c->input(num_features + 1), 0, &unused_shape)); in __anone3ce78ae0202()
82 c->WithRank(c->input(num_features + 2), 0, &unused_shape)); in __anone3ce78ae0202()
84 c->WithRank(c->input(num_features + 3), 0, &unused_shape)); in __anone3ce78ae0202()
106 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anone3ce78ae0302()
107 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anone3ce78ae0302()
108 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_input)); in __anone3ce78ae0302()
[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 __anona7ffb82a0202()
56 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &b_indices)); in __anona7ffb82a0202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &a_shape)); in __anona7ffb82a0302()
100 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &unused)); // a_indices in __anona7ffb82a0402()
[all …]
Ddataset_ops.cc110 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anondee5e0a20102()
122 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anondee5e0a20202()
134 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anondee5e0a20302()
173 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anondee5e0a20402()
242 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anondee5e0a20502()
243 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anondee5e0a20502()
244 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anondee5e0a20502()
245 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anondee5e0a20502()
258 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anondee5e0a20602()
272 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anondee5e0a20702()
[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 __anone43acdb40202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &s)); in __anone43acdb40202()
95 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &shape0)); in __anone43acdb40302()
96 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &shape1)); in __anone43acdb40302()
97 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &shape2)); in __anone43acdb40302()
139 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anone43acdb40402()
140 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anone43acdb40402()
156 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in __anone43acdb40502()
[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()
138 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &boxes)); in CombinedNMSShapeFn()
[all …]
Dnn_ops.cc34 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in FractionalPoolShapeFn()
91 TF_RETURN_IF_ERROR(c->WithRank(s, 4, &s)); in __anon3e672dd80202()
111 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in __anon3e672dd80302()
116 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &vec)); in __anon3e672dd80302()
143 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in __anon3e672dd80402()
150 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &vec)); in __anon3e672dd80402()
293 TF_RETURN_IF_ERROR(c->WithRank(s, 4, &s)); in __anon3e672dd80502()
316 TF_RETURN_IF_ERROR(c->WithRank(s, 4, &s)); in __anon3e672dd80602()
347 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in CommonFusedConvCalculations()
371 TF_RETURN_IF_ERROR(c->WithRank(c->input(paddings_index), 2, &paddings)); in CommonFusedConvCalculations()
[all …]
Dtraining_ops.cc46 TF_RETURN_IF_ERROR(c->WithRank(c->input(grad_idx + 1), 1, &indices)); in HandleGradAndIndicesInputs()
62 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); // alpha in ApplyGradientDescentShapeFn()
91 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); // alpha in ApplyProximalGradientDescentShapeFn()
92 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); // l1 in ApplyProximalGradientDescentShapeFn()
93 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); // l2 in ApplyProximalGradientDescentShapeFn()
162 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); // lr in ApplyAdadeltaShapeFn()
163 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); // rho in ApplyAdadeltaShapeFn()
164 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); // epsilon in ApplyAdadeltaShapeFn()
239 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); // lr in ApplyAdagradShapeFn()
277 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); // lr in ApplyProximalAdagradShapeFn()
[all …]
Dexperimental_dataset_ops.cc37 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &tag_shape)); in __anondb35634f0102()
76 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anondb35634f0202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anondb35634f0202()
78 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anondb35634f0202()
79 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anondb35634f0202()
80 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); in __anondb35634f0202()
81 TF_RETURN_IF_ERROR(c->WithRank(c->input(6), 0, &unused)); in __anondb35634f0202()
83 TF_RETURN_IF_ERROR(c->WithRank(c->input(7), 1, &unused)); in __anondb35634f0202()
123 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anondb35634f0302()
125 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &unused)); in __anondb35634f0302()
[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 __anon4a7ac08d0802()
436 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon4a7ac08d0a02()
446 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon4a7ac08d0b02()
477 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon4a7ac08d0d02()
490 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon4a7ac08d0e02()
573 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in __anon4a7ac08d1102()
602 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anon4a7ac08d1202()
624 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anon4a7ac08d1302()
667 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anon4a7ac08d1402()
[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 __anon8a0811960202()
167 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon8a0811960302()
195 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anon8a0811960402()
211 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon8a0811960502()
223 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon8a0811960602()
250 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anon8a0811960702()
270 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anon8a0811960802()
[all …]
Dctc_ops.cc43 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anon78c05cd60102()
44 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &labels_indices)); in __anon78c05cd60102()
45 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &labels_values)); in __anon78c05cd60102()
46 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &sequence_length)); in __anon78c05cd60102()
76 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anon78c05cd60202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &sequence_length)); in __anon78c05cd60202()
106 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anon78c05cd60302()
107 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &sequence_length)); in __anon78c05cd60302()
Dmath_ops.cc910 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &dimension_shape)); in ArgOpShape()
989 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &segment_ids_shape)); in SegmentReductionShapeFn()
1006 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &indices_shape)); in SparseSegmentReductionShapeFn()
1009 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &segment_ids_shape)); in SparseSegmentReductionShapeFn()
1030 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &indices_shape)); in SparseSegmentReductionGradShapeFn()
1037 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in SparseSegmentReductionGradShapeFn()
1068 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &indices_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1071 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &segment_ids_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1074 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &num_segments_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1106 TF_RETURN_IF_ERROR(c->WithRank(s_num_segments, 0, &s_num_segments)); in UnsortedSegmentReductionShapeFn()
[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 __anon530aab340102()
67 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon530aab340102()
82 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &seed)); in __anon530aab340202()
89 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &logits_shape)); in __anon530aab340202()
90 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon530aab340202()
Dstring_ops.cc46 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonb4ab91930102()
47 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anonb4ab91930102()
66 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonb4ab91930202()
182 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anonb4ab91930502()
183 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonb4ab91930502()
200 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anonb4ab91930602()
201 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonb4ab91930602()
243 TF_RETURN_IF_ERROR(c->WithRank(pos_shape, c->Rank(len_shape), &unused)); in __anonb4ab91930702()
275 TF_RETURN_IF_ERROR(c->WithRank(input_inner_values_shape, 1, &unused)); in __anonb4ab91930802()
279 TF_RETURN_IF_ERROR(c->WithRank(splits_shape, 1, &unused)); in __anonb4ab91930802()
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()
Darray_ops.cc77 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &paddings)); in PadShapeFn()
85 TF_RETURN_IF_ERROR(c->WithRank(input, c->Value(n_dim), &input)); in PadShapeFn()
104 TF_RETURN_IF_ERROR(c->WithRank(input, num_dims, &input)); in PadShapeFn()
146 TF_RETURN_IF_ERROR(c->WithRank(input, rank, &input)); in TransposeShapeFn()
148 TF_RETURN_IF_ERROR(c->WithRank(perm_shape, 1, &perm_shape)); in TransposeShapeFn()
451 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &dims)); in __anon7c94107b0602()
470 TF_RETURN_IF_ERROR(c->WithRank(shape_in, 1, &shape_in)); in __anon7c94107b0702()
839 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), c->Rank(input) - 1, &diag)); in __anon7c94107b1102()
906 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &dims)); in __anon7c94107b1302()
909 TF_RETURN_IF_ERROR(c->WithRank(input, c->Value(dims_dim), &input)); in __anon7c94107b1302()
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/ops/
Dstats_accumulator_ops.cc41 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anone13797730102()
43 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anone13797730102()
68 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 0, &unused_input)); in __anone13797730202()
70 c->WithRank(c->input(num_resource_handles), 0, &unused_input)); in __anone13797730202()
72 TF_RETURN_IF_ERROR(c->WithRank(c->input(num_resource_handles + i + 1), in __anone13797730202()
75 TF_RETURN_IF_ERROR(c->WithRank( in __anone13797730202()
78 TF_RETURN_IF_ERROR(c->WithRank( in __anone13797730202()
83 TF_RETURN_IF_ERROR(c->WithRank( in __anone13797730202()
115 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anone13797730302()
116 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anone13797730302()
[all …]
Dtraining_ops.cc32 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anond8bb5a410102()
33 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anond8bb5a410102()
34 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_input)); in __anond8bb5a410102()
35 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &unused_input)); in __anond8bb5a410102()
66 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anond8bb5a410202()
67 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anond8bb5a410202()
68 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_input)); in __anond8bb5a410202()
69 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused_input)); in __anond8bb5a410202()
71 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused_input)); in __anond8bb5a410202()
73 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused_input)); in __anond8bb5a410202()
[all …]
Dmodel_ops.cc31 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anond0e057b60102()
45 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anond0e057b60202()
46 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anond0e057b60202()
47 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_input)); in __anond0e057b60202()
63 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anond0e057b60302()
80 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anond0e057b60402()
99 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anond0e057b60502()
100 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anond0e057b60502()
101 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_input)); in __anond0e057b60502()
121 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anond0e057b60602()
Dsplit_handler_ops.cc46 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_shape)); in __anon295dc3ce0102()
49 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &partition_ids_shape)); in __anon295dc3ce0102()
51 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &bucket_ids_shape)); in __anon295dc3ce0102()
61 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 1, &bucket_boundaries_shape)); in __anon295dc3ce0102()
118 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_shape)); in __anon295dc3ce0202()
121 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &partition_ids_shape)); in __anon295dc3ce0202()
123 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &bucket_ids_shape)); in __anon295dc3ce0202()
133 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 1, &bucket_boundaries_shape)); in __anon295dc3ce0202()
189 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_shape)); in __anon295dc3ce0302()
192 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &partition_ids_shape)); in __anon295dc3ce0302()
[all …]
/external/tensorflow/tensorflow/contrib/rnn/ops/
Dlstm_ops.cc47 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon4050c6010102()
48 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anon4050c6010102()
134 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon4050c6010202()
135 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anon4050c6010202()
202 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &x)); in __anon4050c6010302()
203 TF_RETURN_IF_ERROR(c->WithRank(c->input(c->num_inputs() - 1), 1, &b)); in __anon4050c6010302()
294 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &x)); in __anon4050c6010402()
295 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &cs_prev)); in __anon4050c6010402()
296 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 2, &h_prev)); in __anon4050c6010402()
297 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 2, &w)); in __anon4050c6010402()
[all …]
Dgru_ops.cc37 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon07a5ee8f0102()
38 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon07a5ee8f0102()
115 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon07a5ee8f0202()
116 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon07a5ee8f0202()
117 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &w_ru)); in __anon07a5ee8f0202()
/external/tensorflow/tensorflow/contrib/text/ops/
Dskip_gram_ops.cc37 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anon1ca0b1550102()
39 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon1ca0b1550102()
40 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon1ca0b1550102()
41 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon1ca0b1550102()
42 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anon1ca0b1550102()
/external/tensorflow/tensorflow/contrib/seq2seq/ops/
Dbeam_search_ops.cc38 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &step_ids)); in __anon559769010102()
39 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &parent_ids)); in __anon559769010102()
40 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &max_sequence_lengths)); in __anon559769010102()
41 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &end_token)); in __anon559769010102()

123