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 __anon8bfbd4920102()
167 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon8bfbd4920102()
213 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon8bfbd4920202()
214 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon8bfbd4920202()
215 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anon8bfbd4920202()
216 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); in __anon8bfbd4920202()
259 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon8bfbd4920302()
260 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon8bfbd4920302()
261 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anon8bfbd4920302()
262 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); in __anon8bfbd4920302()
[all …]
Dboosted_trees_ops.cc37 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anon8a3fd5e50102()
64 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &node_id_range_shape)); in __anon8a3fd5e50202()
69 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &summary_shape_base)); in __anon8a3fd5e50202()
75 TF_RETURN_IF_ERROR(c->WithRank(c->input(1 + i), 3, &summary_shape)); in __anon8a3fd5e50202()
80 c->WithRank(c->input(num_features + 1), 0, &unused_shape)); in __anon8a3fd5e50202()
82 c->WithRank(c->input(num_features + 2), 0, &unused_shape)); in __anon8a3fd5e50202()
84 c->WithRank(c->input(num_features + 3), 0, &unused_shape)); in __anon8a3fd5e50202()
106 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anon8a3fd5e50302()
107 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anon8a3fd5e50302()
108 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_input)); in __anon8a3fd5e50302()
[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 __anone6e195410202()
56 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &b_indices)); in __anone6e195410202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &a_shape)); in __anone6e195410302()
100 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &unused)); // a_indices in __anone6e195410402()
[all …]
Ddataset_ops.cc110 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anonfa0360990102()
122 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anonfa0360990202()
134 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &count_shape)); in __anonfa0360990302()
173 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonfa0360990402()
242 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonfa0360990502()
243 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anonfa0360990502()
244 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anonfa0360990502()
245 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anonfa0360990502()
258 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonfa0360990602()
272 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonfa0360990702()
[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 __anonec23b74b0202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &s)); in __anonec23b74b0202()
95 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &shape0)); in __anonec23b74b0302()
96 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &shape1)); in __anonec23b74b0302()
97 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &shape2)); in __anonec23b74b0302()
139 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anonec23b74b0402()
140 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonec23b74b0402()
156 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in __anonec23b74b0502()
[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 __anon4650176f0202()
111 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in __anon4650176f0302()
116 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &vec)); in __anon4650176f0302()
143 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 4, &input)); in __anon4650176f0402()
150 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 1, &vec)); in __anon4650176f0402()
293 TF_RETURN_IF_ERROR(c->WithRank(s, 4, &s)); in __anon4650176f0502()
316 TF_RETURN_IF_ERROR(c->WithRank(s, 4, &s)); in __anon4650176f0602()
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 __anon6500dca60102()
76 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon6500dca60202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon6500dca60202()
78 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon6500dca60202()
79 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anon6500dca60202()
80 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused)); in __anon6500dca60202()
81 TF_RETURN_IF_ERROR(c->WithRank(c->input(6), 0, &unused)); in __anon6500dca60202()
83 TF_RETURN_IF_ERROR(c->WithRank(c->input(7), 1, &unused)); in __anon6500dca60202()
123 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon6500dca60302()
125 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &unused)); in __anon6500dca60302()
[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 __anona2f81a440802()
436 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anona2f81a440a02()
446 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anona2f81a440b02()
477 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anona2f81a440d02()
490 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anona2f81a440e02()
573 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused)); in __anona2f81a441102()
602 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anona2f81a441202()
624 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anona2f81a441302()
667 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anona2f81a441402()
[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 __anonc8e9eead0202()
167 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anonc8e9eead0302()
195 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anonc8e9eead0402()
211 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anonc8e9eead0502()
223 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anonc8e9eead0602()
250 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &handle)); in __anonc8e9eead0702()
270 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &handle)); in __anonc8e9eead0802()
[all …]
Dctc_ops.cc43 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anon7dc6794d0102()
44 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &labels_indices)); in __anon7dc6794d0102()
45 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &labels_values)); in __anon7dc6794d0102()
46 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &sequence_length)); in __anon7dc6794d0102()
76 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anon7dc6794d0202()
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &sequence_length)); in __anon7dc6794d0202()
106 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &inputs)); in __anon7dc6794d0302()
107 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &sequence_length)); in __anon7dc6794d0302()
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 __anon62f9110b0102()
67 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon62f9110b0102()
82 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &seed)); in __anon62f9110b0202()
89 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &logits_shape)); in __anon62f9110b0202()
90 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon62f9110b0202()
Dstring_ops.cc46 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonf38d6eaa0102()
47 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anonf38d6eaa0102()
66 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonf38d6eaa0202()
182 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anonf38d6eaa0502()
183 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonf38d6eaa0502()
200 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anonf38d6eaa0602()
201 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anonf38d6eaa0602()
243 TF_RETURN_IF_ERROR(c->WithRank(pos_shape, c->Rank(len_shape), &unused)); in __anonf38d6eaa0702()
275 TF_RETURN_IF_ERROR(c->WithRank(input_inner_values_shape, 1, &unused)); in __anonf38d6eaa0802()
279 TF_RETURN_IF_ERROR(c->WithRank(splits_shape, 1, &unused)); in __anonf38d6eaa0802()
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 __anondb9326b20602()
470 TF_RETURN_IF_ERROR(c->WithRank(shape_in, 1, &shape_in)); in __anondb9326b20702()
839 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), c->Rank(input) - 1, &diag)); in __anondb9326b21102()
906 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &dims)); in __anondb9326b21302()
909 TF_RETURN_IF_ERROR(c->WithRank(input, c->Value(dims_dim), &input)); in __anondb9326b21302()
[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 __anon72691d0a0102()
43 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anon72691d0a0102()
68 TF_RETURN_IF_ERROR(c->WithRank(c->input(i), 0, &unused_input)); in __anon72691d0a0202()
70 c->WithRank(c->input(num_resource_handles), 0, &unused_input)); in __anon72691d0a0202()
72 TF_RETURN_IF_ERROR(c->WithRank(c->input(num_resource_handles + i + 1), in __anon72691d0a0202()
75 TF_RETURN_IF_ERROR(c->WithRank( in __anon72691d0a0202()
78 TF_RETURN_IF_ERROR(c->WithRank( in __anon72691d0a0202()
83 TF_RETURN_IF_ERROR(c->WithRank( in __anon72691d0a0202()
115 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anon72691d0a0302()
116 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anon72691d0a0302()
[all …]
Dtraining_ops.cc32 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anon448b41f80102()
33 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anon448b41f80102()
34 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_input)); in __anon448b41f80102()
35 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &unused_input)); in __anon448b41f80102()
66 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anon448b41f80202()
67 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anon448b41f80202()
68 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_input)); in __anon448b41f80202()
69 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused_input)); in __anon448b41f80202()
71 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused_input)); in __anon448b41f80202()
73 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &unused_input)); in __anon448b41f80202()
[all …]
Dmodel_ops.cc31 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anonfb6f82cd0102()
45 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anonfb6f82cd0202()
46 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anonfb6f82cd0202()
47 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_input)); in __anonfb6f82cd0202()
63 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anonfb6f82cd0302()
80 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anonfb6f82cd0402()
99 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anonfb6f82cd0502()
100 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused_input)); in __anonfb6f82cd0502()
101 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused_input)); in __anonfb6f82cd0502()
121 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_input)); in __anonfb6f82cd0602()
Dsplit_handler_ops.cc46 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_shape)); in __anond11615e50102()
49 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &partition_ids_shape)); in __anond11615e50102()
51 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &bucket_ids_shape)); in __anond11615e50102()
61 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 1, &bucket_boundaries_shape)); in __anond11615e50102()
118 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_shape)); in __anond11615e50202()
121 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &partition_ids_shape)); in __anond11615e50202()
123 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &bucket_ids_shape)); in __anond11615e50202()
133 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 1, &bucket_boundaries_shape)); in __anond11615e50202()
189 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 0, &unused_shape)); in __anond11615e50302()
192 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &partition_ids_shape)); in __anond11615e50302()
[all …]
/external/tensorflow/tensorflow/contrib/rnn/ops/
Dlstm_ops.cc47 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anonad8b69780102()
48 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anonad8b69780102()
134 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anonad8b69780202()
135 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &cs_prev)); in __anonad8b69780202()
202 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &x)); in __anonad8b69780302()
203 TF_RETURN_IF_ERROR(c->WithRank(c->input(c->num_inputs() - 1), 1, &b)); in __anonad8b69780302()
294 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &x)); in __anonad8b69780402()
295 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &cs_prev)); in __anonad8b69780402()
296 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 2, &h_prev)); in __anonad8b69780402()
297 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 2, &w)); in __anonad8b69780402()
[all …]
Dgru_ops.cc37 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon6fce7f260102()
38 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon6fce7f260102()
115 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &x)); in __anon6fce7f260202()
116 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &h_prev)); in __anon6fce7f260202()
117 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &w_ru)); in __anon6fce7f260202()
/external/tensorflow/tensorflow/contrib/text/ops/
Dskip_gram_ops.cc37 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 1, &unused)); in __anon2a5bd20c0102()
39 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused)); in __anon2a5bd20c0102()
40 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &unused)); in __anon2a5bd20c0102()
41 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &unused)); in __anon2a5bd20c0102()
42 TF_RETURN_IF_ERROR(c->WithRank(c->input(4), 0, &unused)); in __anon2a5bd20c0102()
/external/tensorflow/tensorflow/contrib/seq2seq/ops/
Dbeam_search_ops.cc38 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &step_ids)); in __anon802694180102()
39 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &parent_ids)); in __anon802694180102()
40 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &max_sequence_lengths)); in __anon802694180102()
41 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &end_token)); in __anon802694180102()

123