Home
last modified time | relevance | path

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

123456789

/external/tensorflow/tensorflow/core/ops/
Dcudnn_rnn_ops.cc60 c->set_output(0, c->Vector(1)); in __anone1d60d010202()
99 c->set_output(0, output_shape); in __anone1d60d010302()
100 c->set_output(1, output_h_shape); in __anone1d60d010302()
101 c->set_output(2, output_c_shape); in __anone1d60d010302()
102 c->set_output(3, c->UnknownShape()); in __anone1d60d010302()
142 c->set_output(0, output_shape); in __anone1d60d010402()
143 c->set_output(1, output_h_shape); in __anone1d60d010402()
144 c->set_output(2, output_c_shape); in __anone1d60d010402()
145 c->set_output(3, c->UnknownShape()); in __anone1d60d010402()
146 c->set_output(4, c->UnknownShape()); in __anone1d60d010402()
[all …]
Dsparse_ops.cc36 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, in SparseSparseMinOrMaxShapeFn()
38 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in SparseSparseMinOrMaxShapeFn()
57 c->set_output(0, c->Vector(c->Dim(a_indices, 0))); in __anona7ffb82a0202()
58 c->set_output(1, c->Vector(c->Dim(b_indices, 0))); in __anona7ffb82a0202()
78 c->set_output( in __anona7ffb82a0302()
80 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anona7ffb82a0302()
81 c->set_output(2, a_shape); in __anona7ffb82a0302()
116 c->set_output(0, c->Matrix(output_left, output_right)); in __anona7ffb82a0402()
132 c->set_output(0, c->Vector(3)); in __anona7ffb82a0502()
148 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 3)); in __anona7ffb82a0602()
[all …]
Dparsing_ops.cc39 c->set_output(0, out); in __anon952516500102()
76 c->set_output(output_idx++, c->Matrix(c->UnknownDim(), 2)); in __anon952516500202()
79 c->set_output(output_idx++, c->Vector(c->UnknownDim())); in __anon952516500202()
82 c->set_output(output_idx++, c->Vector(2)); in __anon952516500202()
91 c->set_output(output_idx++, dense); in __anon952516500202()
119 c->set_output(output_idx++, c->Matrix(c->UnknownDim(), 1)); in __anon952516500302()
122 c->set_output(output_idx++, c->Vector(c->UnknownDim())); in __anon952516500302()
125 c->set_output(output_idx++, c->Vector(1)); in __anon952516500302()
133 c->set_output(output_idx++, dense); in __anon952516500302()
183 c->set_output(output_idx++, c->Matrix(c->UnknownDim(), 2)); in __anon952516500402()
[all …]
Dnn_ops.cc59 c->set_output(0, c->MakeShape(output_dims)); in FractionalPoolShapeFn()
60 c->set_output(1, c->Vector(output_dims[1])); in FractionalPoolShapeFn()
61 c->set_output(2, c->Vector(output_dims[2])); in FractionalPoolShapeFn()
92 c->set_output(0, s); in __anon3e672dd80202()
122 c->set_output(0, out); in __anon3e672dd80302()
156 c->set_output(0, dx); in __anon3e672dd80402()
159 c->set_output(1, vector_shape); in __anon3e672dd80402()
160 c->set_output(2, vector_shape); in __anon3e672dd80402()
161 c->set_output(3, vector_shape); in __anon3e672dd80402()
162 c->set_output(4, vector_shape); in __anon3e672dd80402()
[all …]
Dstring_ops.cc48 c->set_output(0, c->input(0)); in __anonb4ab91930102()
67 c->set_output(0, c->input(0)); in __anonb4ab91930202()
139 c->set_output(0, c->Scalar()); in __anonb4ab91930302()
155 c->set_output(0, c->Scalar()); in __anonb4ab91930402()
169 c->set_output(0, out); in __anonb4ab91930402()
185 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anonb4ab91930502()
186 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anonb4ab91930502()
187 c->set_output(2, c->Vector(2)); in __anonb4ab91930502()
203 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anonb4ab91930602()
204 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anonb4ab91930602()
[all …]
Dctc_ops.cc59 c->set_output(0, c->Vector(batch_size)); in __anon78c05cd60102()
60 c->set_output(1, inputs); in __anon78c05cd60102()
85 c->set_output(0, c->Matrix(total_decoded_outputs, 2)); in __anon78c05cd60202()
86 c->set_output(1, c->Vector(total_decoded_outputs)); in __anon78c05cd60202()
87 c->set_output(2, c->Vector(2)); in __anon78c05cd60202()
88 c->set_output(3, c->Matrix(batch_size, 1)); in __anon78c05cd60202()
120 c->set_output(out_idx++, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anon78c05cd60302()
123 c->set_output(out_idx++, c->Vector(InferenceContext::kUnknownDim)); in __anon78c05cd60302()
127 c->set_output(out_idx++, shape_v); in __anon78c05cd60302()
129 c->set_output(out_idx++, c->Matrix(batch_size, top_paths)); in __anon78c05cd60302()
Darray_ops.cc70 c->set_output(0, c->MakeShape(dims)); in PadKnown()
96 c->set_output(0, c->UnknownShapeOfRank(c->Value(n_dim))); in PadShapeFn()
98 c->set_output(0, c->UnknownShape()); in PadShapeFn()
123 c->set_output(0, c->UnknownShape()); in TransposeShapeFn()
140 c->set_output(0, input); in TransposeShapeFn()
176 c->set_output(0, c->MakeShape(dims)); in TransposeShapeFn()
187 c->set_output(0, out); in SetOutputShapeForReshape()
272 c->set_output(0, out); in SetOutputShapeForReshape()
312 c->set_output(0, passed_shape); in __anon7c94107b0202()
332 c->set_output(0, c->UnknownShape()); in __anon7c94107b0302()
[all …]
Dboosted_trees_ops.cc38 c->set_output(0, c->Scalar()); in __anone3ce78ae0102()
88 TF_RETURN_IF_ERROR(c->set_output("node_ids_list", output_shapes_vec)); in __anone3ce78ae0202()
89 TF_RETURN_IF_ERROR(c->set_output("gains_list", output_shapes_vec)); in __anone3ce78ae0202()
90 TF_RETURN_IF_ERROR(c->set_output("thresholds_list", output_shapes_vec)); in __anone3ce78ae0202()
94 c->set_output("left_node_contribs_list", output_shapes_contribs)); in __anone3ce78ae0202()
96 c->set_output("right_node_contribs_list", output_shapes_contribs)); in __anone3ce78ae0202()
134 c->set_output(0, c->Scalar()); in __anone3ce78ae0502()
135 c->set_output(1, c->Scalar()); in __anone3ce78ae0502()
136 c->set_output(2, c->Scalar()); in __anone3ce78ae0502()
137 c->set_output(3, c->Scalar()); in __anone3ce78ae0502()
[all …]
Dmkl_nn_ops.cc168 c->set_output(1, c->Scalar()); in __anon4d19f77b0102()
169 c->set_output(2, c->Scalar()); in __anon4d19f77b0102()
217 c->set_output(1, c->Scalar()); in __anon4d19f77b0202()
218 c->set_output(2, c->Scalar()); in __anon4d19f77b0202()
265 c->set_output(1, c->Scalar()); in __anon4d19f77b0302()
266 c->set_output(2, c->Scalar()); in __anon4d19f77b0302()
310 c->set_output(1, c->Scalar()); in __anon4d19f77b0402()
311 c->set_output(2, c->Scalar()); in __anon4d19f77b0402()
360 c->set_output(1, c->Scalar()); in __anon4d19f77b0502()
361 c->set_output(2, c->Scalar()); in __anon4d19f77b0502()
[all …]
Dimage_ops.cc58 c->set_output(0, c->MakeShape({batch_dim, height, width, channel_dim})); in SetOutputToSizedImage()
85 c->set_output(0, c->MakeShape({InferenceContext::kUnknownDim, in DecodeImageShapeFn()
93 c->set_output(0, c->Scalar()); in EncodeImageShapeFn()
106 c->set_output(0, out); in ColorspaceShapeFn()
130 c->set_output(0, c->Vector(c->UnknownDim())); in NMSShapeFn()
191 c->set_output(0, c->MakeShape({batch_dim, output_size, 4})); in CombinedNMSShapeFn()
192 c->set_output(1, c->MakeShape({batch_dim, output_size})); in CombinedNMSShapeFn()
193 c->set_output(2, c->MakeShape({batch_dim, output_size})); in CombinedNMSShapeFn()
194 c->set_output(3, c->Vector(batch_dim)); in CombinedNMSShapeFn()
228 c->set_output(0, c->input(1)); in __anon6a71d27f0202()
[all …]
Dcontrol_flow_ops.cc31 c->set_output(0, out); in SwitchShape()
32 c->set_output(1, out); in SwitchShape()
73 c->set_output(0, c->UnknownShape()); in __anonad90f1540202()
81 c->set_output(0, c->UnknownShape()); in __anonad90f1540202()
85 c->set_output(0, first_input); in __anonad90f1540202()
111 c->set_output(0, out); in MergeShape()
112 c->set_output(1, c->Scalar()); in MergeShape()
142 c->set_output(0, c->UnknownShape()); in __anonad90f1540402()
153 c->set_output(0, c->input(0)); in __anonad90f1540402()
Dlinalg_ops.cc46 c->set_output(0, out); in BatchUnchangedSquareShapeFn()
83 c->set_output(0, out); in MatrixSolveShapeFn()
99 c->set_output(0, e_shape); in SelfAdjointEigV2ShapeFn()
105 c->set_output(1, v_shape); in SelfAdjointEigV2ShapeFn()
107 c->set_output(1, c->Vector(0ll)); in SelfAdjointEigV2ShapeFn()
131 c->set_output(0, lu_shape); in LuShapeFn()
132 c->set_output(1, p_shape); in LuShapeFn()
161 c->set_output(0, q_shape); in QrShapeFn()
162 c->set_output(1, r_shape); in QrShapeFn()
183 c->set_output(0, e_shape); in SvdShapeFn()
[all …]
Dmath_ops.cc41 c->set_output(0, cur); in __anonb22bfa860102()
157 c->set_output(0, out); in __anonb22bfa860202()
653 c->set_output(0, output); in __anonb22bfa860302()
779 c->set_output(0, data); in __anonb22bfa860502()
791 c->set_output(0, data); in __anonb22bfa860502()
799 c->set_output(0, data); in __anonb22bfa860502()
806 c->set_output(0, data); in __anonb22bfa860502()
814 c->set_output(0, data); in __anonb22bfa860502()
818 c->set_output(0, data); in __anonb22bfa860502()
933 c->set_output(0, c->MakeShape(dims)); in ArgOpShape()
[all …]
Dfunctional_ops.cc44 c->set_output(i, handle_type->at(0).shape); in __anon22ceab0f0102()
46 c->set_output(i, c->UnknownShape()); in __anon22ceab0f0102()
49 c->set_output(i, c->input(i)); in __anon22ceab0f0102()
130 c->set_output(static_cast<int>(i), output_shape_handle); in __anon22ceab0f0202()
159 c->set_output(static_cast<int>(i), output_shape_handle); in __anon22ceab0f0302()
174 c->set_output(i, c->input(i)); in __anon22ceab0f0402()
219 c->set_output(static_cast<int>(i), output_shape_handle); in __anon22ceab0f0502()
223 c->set_output(i, c->input(i)); in __anon22ceab0f0502()
237 c->set_output(i, c->input(i)); in __anon22ceab0f0602()
288 c->set_output(0, out); in __anon22ceab0f0702()
Dlist_ops.cc58 c->set_output(0, c->Scalar()); in __anonebbd57d80202()
76 c->set_output(0, c->Scalar()); in __anonebbd57d80302()
122 c->set_output(0, input_handles); in __anonebbd57d80402()
191 c->set_output(1, tensor_shape); in __anonebbd57d80502()
192 c->set_output(0, c->Scalar()); in __anonebbd57d80502()
236 c->set_output(0, result); in __anonebbd57d80602()
268 c->set_output(0, result); in TensorListConcatShapeInference()
270 c->set_output(0, c->UnknownShape()); in TensorListConcatShapeInference()
272 c->set_output(1, c->MakeShape({c->UnknownDim()})); in TensorListConcatShapeInference()
314 c->set_output(0, c->Scalar()); in __anonebbd57d80902()
[all …]
Dset_ops.cc85 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank)); in __anon967df2c80102()
86 c->set_output(1, c->Vector(c->UnknownDim())); in __anon967df2c80102()
87 c->set_output(2, c->Vector(output_rank)); in __anon967df2c80102()
129 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank_dim)); in __anon967df2c80202()
130 c->set_output(1, c->Vector(c->UnknownDim())); in __anon967df2c80202()
131 c->set_output(2, c->Vector(output_rank_dim)); in __anon967df2c80202()
185 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank_dim)); in __anon967df2c80302()
186 c->set_output(1, c->Vector(c->UnknownDim())); in __anon967df2c80302()
187 c->set_output(2, c->Vector(output_rank_dim)); in __anon967df2c80302()
Dio_ops.cc35 c->set_output(i, c->Scalar()); in ScalarInputsAndOutputs()
48 c->set_output(i, c->Scalar()); in TwoElementVectorAndScalarOutputs()
54 c->set_output(0, c->Vector(2)); in TwoElementOutput()
112 c->set_output(i, c->UnknownShape()); in __anone43acdb40302()
124 c->set_output(i, shape_handle); in __anone43acdb40302()
203 c->set_output(0, c->UnknownShape()); in __anone43acdb40702()
227 c->set_output(0, c->UnknownShape()); in __anone43acdb40802()
238 c->set_output(0, shape_handle); in __anone43acdb40802()
241 c->set_output(0, c->UnknownShape()); in __anone43acdb40802()
383 c->set_output(0, out); in __anone43acdb40902()
[all …]
/external/tensorflow/tensorflow/contrib/rnn/ops/
Dlstm_ops.cc54 c->set_output(i, output); in __anon4050c6010102()
143 c->set_output(0, c->Matrix(batch_size, cell_size)); in __anon4050c6010202()
144 c->set_output(1, c->Matrix(batch_size, cell_size_times_4)); in __anon4050c6010202()
145 c->set_output(2, cell_size_vec); in __anon4050c6010202()
146 c->set_output(3, cell_size_vec); in __anon4050c6010202()
147 c->set_output(4, cell_size_vec); in __anon4050c6010202()
214 c->set_output(i, output); in __anon4050c6010302()
303 c->set_output(0, x); in __anon4050c6010402()
304 c->set_output(1, cs_prev); in __anon4050c6010402()
305 c->set_output(2, h_prev); in __anon4050c6010402()
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/ops/
Dstats_accumulator_ops.cc118 c->set_output(0, c->Scalar()); in __anone13797730302()
119 c->set_output(1, c->Vector(c->UnknownDim())); in __anone13797730302()
120 c->set_output(2, c->UnknownShape()); in __anone13797730302()
121 c->set_output(3, c->Vector(c->UnknownDim())); in __anone13797730302()
122 c->set_output(4, c->Vector(c->UnknownDim())); in __anone13797730302()
201 c->set_output(0, c->Scalar()); in __anone13797730502()
203 c->set_output(1, c->Scalar()); in __anone13797730502()
204 c->set_output(2, c->Vector(c->UnknownDim())); in __anone13797730502()
205 c->set_output(3, c->UnknownShape()); in __anone13797730502()
206 c->set_output(4, c->Vector(c->UnknownDim())); in __anone13797730502()
[all …]
Dtraining_ops.cc36 c->set_output(0, c->Scalar()); in __anond8bb5a410102()
105 c->set_output(0, c->Scalar()); in __anond8bb5a410302()
106 c->set_output(1, c->Scalar()); in __anond8bb5a410302()
107 c->set_output(2, c->Scalar()); in __anond8bb5a410302()
108 c->set_output(3, c->Scalar()); in __anond8bb5a410302()
109 c->set_output(4, c->Scalar()); in __anond8bb5a410302()
110 c->set_output(5, c->Scalar()); in __anond8bb5a410302()
Dsplit_handler_ops.cc62 c->set_output(0, c->Vector(c->UnknownDim())); in __anon295dc3ce0102()
63 c->set_output(1, c->Vector(c->UnknownDim())); in __anon295dc3ce0102()
64 c->set_output(2, c->Vector(c->UnknownDim())); in __anon295dc3ce0102()
134 c->set_output(0, c->Vector(c->UnknownDim())); in __anon295dc3ce0202()
135 c->set_output(1, c->Vector(c->UnknownDim())); in __anon295dc3ce0202()
136 c->set_output(2, c->Vector(c->UnknownDim())); in __anon295dc3ce0202()
203 c->set_output(0, c->Vector(c->UnknownDim())); in __anon295dc3ce0302()
204 c->set_output(1, c->Vector(c->UnknownDim())); in __anon295dc3ce0302()
205 c->set_output(2, c->Vector(c->UnknownDim())); in __anon295dc3ce0302()
Dprediction_ops.cc36 c->set_output(0, {c->Matrix(InferenceContext::kUnknownDim, in ApplyGradientTreesPredictionShapeFn()
39 c->set_output(1, {c->UnknownShape()}); in ApplyGradientTreesPredictionShapeFn()
52 c->set_output(0, {c->Matrix(InferenceContext::kUnknownDim, in ApplyGradientTreesPredictionVerboseShapeFn()
55 c->set_output(1, {c->UnknownShape()}); in ApplyGradientTreesPredictionVerboseShapeFn()
56 c->set_output(2, {c->Matrix(InferenceContext::kUnknownDim, in ApplyGradientTreesPredictionVerboseShapeFn()
178 return c->set_output("partition_ids", in __anonbe66d2360102()
Dmodel_ops.cc32 c->set_output(0, c->Scalar()); in __anond0e057b60102()
64 c->set_output(0, c->Scalar()); in __anond0e057b60302()
81 c->set_output(0, c->Scalar()); in __anond0e057b60402()
82 c->set_output(1, c->Scalar()); in __anond0e057b60402()
122 c->set_output(0, c->Scalar()); in __anond0e057b60602()
125 c->set_output(1, {c->Vector(num_all_handlers)}); in __anond0e057b60602()
/external/tensorflow/tensorflow/contrib/layers/ops/
Dsparse_feature_cross_op.cc36 c->set_output(0, c->Matrix(c->UnknownDim(), 2)); in __anon9c7e7a520102()
37 c->set_output(1, c->Vector(c->UnknownDim())); in __anon9c7e7a520102()
38 c->set_output(2, c->Vector(2)); in __anon9c7e7a520102()
108 c->set_output(0, c->Matrix(c->UnknownDim(), 2)); in __anon9c7e7a520202()
109 c->set_output(1, c->Vector(c->UnknownDim())); in __anon9c7e7a520202()
110 c->set_output(2, c->Vector(2)); in __anon9c7e7a520202()
/external/tensorflow/tensorflow/contrib/mpi_collectives/ops/
Dmpi_ops.cc36 c->set_output(0, c->Scalar()); in __anon107d6f1b0102()
51 c->set_output(0, c->Scalar()); in __anon107d6f1b0202()
66 c->set_output(0, c->Scalar()); in __anon107d6f1b0302()
83 c->set_output(0, c->input(0)); in __anon107d6f1b0402()
110 c->set_output(0, output); in __anon107d6f1b0502()

123456789