Home
last modified time | relevance | path

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

12345678910

/external/tensorflow/tensorflow/core/ops/
Dcudnn_rnn_ops.cc61 c->set_output(0, c->Vector(1)); in __anon815c5c9f0202()
100 c->set_output(0, output_shape); in __anon815c5c9f0302()
101 c->set_output(1, output_h_shape); in __anon815c5c9f0302()
102 c->set_output(2, output_c_shape); in __anon815c5c9f0302()
103 c->set_output(3, c->UnknownShape()); in __anon815c5c9f0302()
143 c->set_output(0, output_shape); in __anon815c5c9f0402()
144 c->set_output(1, output_h_shape); in __anon815c5c9f0402()
145 c->set_output(2, output_c_shape); in __anon815c5c9f0402()
146 c->set_output(3, c->UnknownShape()); in __anon815c5c9f0402()
147 c->set_output(4, c->UnknownShape()); in __anon815c5c9f0402()
[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 __anon36a8fc880202()
58 c->set_output(1, c->Vector(c->Dim(b_indices, 0))); in __anon36a8fc880202()
78 c->set_output( in __anon36a8fc880302()
80 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anon36a8fc880302()
81 c->set_output(2, a_shape); in __anon36a8fc880302()
116 c->set_output(0, c->Matrix(output_left, output_right)); in __anon36a8fc880402()
132 c->set_output(0, c->Vector(3)); in __anon36a8fc880502()
148 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 3)); in __anon36a8fc880602()
[all …]
Drnn_ops.cc46 c->set_output(i, output); in __anon652ace880102()
78 c->set_output(0, x); in __anon652ace880202()
79 c->set_output(1, batch_cell_shape); in __anon652ace880202()
80 c->set_output(2, batch_cell_shape); in __anon652ace880202()
81 c->set_output(3, c->Matrix(batch_size, twice_cell_size)); in __anon652ace880202()
114 c->set_output(i, output); in __anon652ace880302()
154 c->set_output(0, c->Matrix(batch_size, cell_size)); in __anon652ace880402()
155 c->set_output(1, c->Matrix(batch_size, cell_size_times_4)); in __anon652ace880402()
156 c->set_output(2, cell_size_vec); in __anon652ace880402()
157 c->set_output(3, cell_size_vec); in __anon652ace880402()
[all …]
Dcount_ops.cc37 c->set_output(0, c->Matrix(nvals, rank)); // out.indices in DenseCountSparseOutputShapeFn()
38 c->set_output(1, c->Vector(nvals)); // out.values in DenseCountSparseOutputShapeFn()
39 c->set_output(2, c->Vector(rank)); // out.dense_shape in DenseCountSparseOutputShapeFn()
46 c->set_output(0, c->Matrix(nvals, rank)); // out.indices in SparseCountSparseOutputShapeFn()
47 c->set_output(1, c->Vector(nvals)); // out.values in SparseCountSparseOutputShapeFn()
48 c->set_output(2, c->Vector(rank)); // out.dense_shape in SparseCountSparseOutputShapeFn()
58 c->set_output(0, c->Matrix(nvals, rank)); // out.indices in RaggedCountSparseOutputShapeFn()
59 c->set_output(1, c->Vector(nvals)); // out.values in RaggedCountSparseOutputShapeFn()
60 c->set_output(2, c->Vector(rank)); // out.dense_shape in RaggedCountSparseOutputShapeFn()
Dboosted_trees_ops.cc39 c->set_output(0, c->Scalar()); in __anon770f674c0102()
89 TF_RETURN_IF_ERROR(c->set_output("node_ids_list", output_shapes_vec)); in __anon770f674c0202()
90 TF_RETURN_IF_ERROR(c->set_output("gains_list", output_shapes_vec)); in __anon770f674c0202()
91 TF_RETURN_IF_ERROR(c->set_output("thresholds_list", output_shapes_vec)); in __anon770f674c0202()
95 c->set_output("left_node_contribs_list", output_shapes_contribs)); in __anon770f674c0202()
97 c->set_output("right_node_contribs_list", output_shapes_contribs)); in __anon770f674c0202()
130 c->set_output(0, rank_1_output_shape); in __anon770f674c0302()
131 c->set_output(1, rank_1_output_shape); in __anon770f674c0302()
132 c->set_output(2, rank_1_output_shape); in __anon770f674c0302()
133 c->set_output(3, rank_1_output_shape); in __anon770f674c0302()
[all …]
Dstring_ops.cc48 c->set_output(0, c->input(0)); in __anon4354d5f10102()
67 c->set_output(0, c->input(0)); in __anon4354d5f10202()
149 c->set_output(0, c->Scalar()); in __anon4354d5f10302()
165 c->set_output(0, c->Scalar()); in __anon4354d5f10402()
179 c->set_output(0, out); in __anon4354d5f10402()
195 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anon4354d5f10502()
196 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anon4354d5f10502()
197 c->set_output(2, c->Vector(2)); in __anon4354d5f10502()
213 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anon4354d5f10602()
214 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anon4354d5f10602()
[all …]
Dctc_ops.cc60 c->set_output(0, c->Vector(batch_size)); in __anonbca2cdf40102()
61 c->set_output(1, inputs); in __anonbca2cdf40102()
97 c->set_output(0, c->Vector(batch_size)); in __anonbca2cdf40202()
98 c->set_output(1, inputs); in __anonbca2cdf40202()
124 c->set_output(0, c->Matrix(total_decoded_outputs, 2)); in __anonbca2cdf40302()
125 c->set_output(1, c->Vector(total_decoded_outputs)); in __anonbca2cdf40302()
126 c->set_output(2, c->Vector(2)); in __anonbca2cdf40302()
127 c->set_output(3, c->Matrix(batch_size, 1)); in __anonbca2cdf40302()
160 c->set_output(out_idx++, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anonbca2cdf40402()
163 c->set_output(out_idx++, c->Vector(InferenceContext::kUnknownDim)); in __anonbca2cdf40402()
[all …]
Dnn_ops.cc62 c->set_output(0, c->MakeShape(output_dims)); in FractionalPoolShapeFn()
63 c->set_output(1, c->Vector(output_dims[1])); in FractionalPoolShapeFn()
64 c->set_output(2, c->Vector(output_dims[2])); in FractionalPoolShapeFn()
119 c->set_output(0, out); in __anonad10f3360202()
153 c->set_output(0, dx); in __anonad10f3360302()
156 c->set_output(1, vector_shape); in __anonad10f3360302()
157 c->set_output(2, vector_shape); in __anonad10f3360302()
158 c->set_output(3, vector_shape); in __anonad10f3360302()
159 c->set_output(4, vector_shape); in __anonad10f3360302()
377 c->set_output(0, s); in __anonad10f3360402()
[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()
114 c->set_output(0, c->UnknownShape()); in DecodeImageV2ShapeFn()
117 c->set_output(0, in DecodeImageV2ShapeFn()
127 c->set_output(0, c->Scalar()); in EncodeImageShapeFn()
140 c->set_output(0, out); in ColorspaceShapeFn()
164 c->set_output(0, c->Vector(c->UnknownDim())); in NMSShapeFn()
189 c->set_output(0, c->Vector(c->UnknownDim())); in SoftNMSShapeFn()
190 c->set_output(1, c->Vector(c->UnknownDim())); in SoftNMSShapeFn()
251 c->set_output(0, c->MakeShape({batch_dim, output_size, 4})); in CombinedNMSShapeFn()
[all …]
Dcontrol_flow_ops.cc32 c->set_output(0, out); in SwitchShape()
33 c->set_output(1, out); in SwitchShape()
51 c->set_output(i, out); in SwitchNShape()
103 c->set_output(0, c->UnknownShape()); in __anon92ff84320202()
111 c->set_output(0, c->UnknownShape()); in __anon92ff84320202()
115 c->set_output(0, first_input); in __anon92ff84320202()
141 c->set_output(0, out); in MergeShape()
142 c->set_output(1, c->Scalar()); in MergeShape()
172 c->set_output(0, c->UnknownShape()); in __anon92ff84320402()
183 c->set_output(0, c->input(0)); in __anon92ff84320402()
Darray_ops.cc77 c->set_output(0, c->MakeShape(dims)); in PadKnown()
103 c->set_output(0, c->UnknownShapeOfRank(c->Value(n_dim))); in PadShapeFn()
105 c->set_output(0, c->UnknownShape()); in PadShapeFn()
130 c->set_output(0, c->UnknownShape()); in TransposeShapeFn()
147 c->set_output(0, input); in TransposeShapeFn()
183 c->set_output(0, c->MakeShape(dims)); in TransposeShapeFn()
194 c->set_output(0, out); in SetOutputShapeForReshape()
278 c->set_output(0, out); in SetOutputShapeForReshape()
318 c->set_output(0, passed_shape); in __anon42d741190202()
338 c->set_output(0, c->UnknownShape()); in __anon42d741190302()
[all …]
Dmath_ops.cc41 c->set_output(0, cur); in __anon725c8f640102()
681 c->set_output(0, output); in __anon725c8f640202()
724 c->set_output(0, output); \
816 c->set_output(0, data); in __anon725c8f640402()
828 c->set_output(0, data); in __anon725c8f640402()
836 c->set_output(0, data); in __anon725c8f640402()
843 c->set_output(0, data); in __anon725c8f640402()
851 c->set_output(0, data); in __anon725c8f640402()
855 c->set_output(0, data); in __anon725c8f640402()
907 c->set_output(0, output); in __anon725c8f640502()
[all …]
Dmkl_nn_ops.cc69 c->set_output(0, s); in __anondbc33bd90102()
95 c->set_output(0, s); in __anondbc33bd90202()
135 c->set_output(0, s); in __anondbc33bd90302()
154 c->set_output(0, s); in __anondbc33bd90402()
645 c->set_output(1, c->Scalar()); in __anondbc33bd90502()
646 c->set_output(2, c->Scalar()); in __anondbc33bd90502()
693 c->set_output(1, c->Scalar()); in __anondbc33bd90602()
694 c->set_output(2, c->Scalar()); in __anondbc33bd90602()
744 c->set_output(1, c->Scalar()); in __anondbc33bd90702()
745 c->set_output(2, c->Scalar()); in __anondbc33bd90702()
[all …]
Dlist_ops.cc64 c->set_output(0, c->Scalar()); in __anonabedecb60202()
82 c->set_output(0, c->Scalar()); in __anonabedecb60302()
128 c->set_output(0, input_handles); in __anonabedecb60402()
197 c->set_output(1, tensor_shape); in __anonabedecb60502()
198 c->set_output(0, c->Scalar()); in __anonabedecb60502()
247 c->set_output(0, result); in __anonabedecb60602()
279 c->set_output(0, result); in TensorListConcatShapeInference()
281 c->set_output(0, c->UnknownShape()); in TensorListConcatShapeInference()
283 c->set_output(1, c->MakeShape({c->UnknownDim()})); in TensorListConcatShapeInference()
325 c->set_output(0, c->Scalar()); in __anonabedecb60902()
[all …]
Dmap_ops.cc27 c->set_output(0, c->Scalar()); in __anon3f27e5980202()
43 c->set_output(0, c->UnknownShape()); in __anon3f27e5980302()
55 c->set_output(0, c->Scalar()); in __anon3f27e5980402()
66 c->set_output(0, c->Scalar()); // output map in __anon3f27e5980502()
82 c->set_output(0, c->UnknownShape()); // output keys in __anon3f27e5980602()
Dstateless_random_ops_v2.cc40 c->set_output(0, out); in StatelessShapeV2()
112 c->set_output(0, c->MakeShape({RNG_KEY_SIZE})); in __anonea62c6790202()
113 c->set_output(1, c->MakeShape({RNG_MAX_COUNTER_SIZE})); in __anonea62c6790202()
114 c->set_output(2, c->MakeShape({})); in __anonea62c6790202()
132 c->set_output(0, c->MakeShape({RNG_KEY_SIZE})); in __anonea62c6790302()
133 c->set_output(1, c->MakeShape({RNG_MAX_COUNTER_SIZE})); in __anonea62c6790302()
141 c->set_output(0, c->MakeShape({})); in __anonea62c6790402()
Dset_ops.cc85 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank)); in __anonda6063e60102()
86 c->set_output(1, c->Vector(c->UnknownDim())); in __anonda6063e60102()
87 c->set_output(2, c->Vector(output_rank)); in __anonda6063e60102()
129 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank_dim)); in __anonda6063e60202()
130 c->set_output(1, c->Vector(c->UnknownDim())); in __anonda6063e60202()
131 c->set_output(2, c->Vector(output_rank_dim)); in __anonda6063e60202()
185 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank_dim)); in __anonda6063e60302()
186 c->set_output(1, c->Vector(c->UnknownDim())); in __anonda6063e60302()
187 c->set_output(2, c->Vector(output_rank_dim)); in __anonda6063e60302()
Dlinalg_ops.cc46 c->set_output(0, out); in BatchUnchangedSquareShapeFn()
89 c->set_output(0, out); in BandedTriangularSolveShapeFn()
126 c->set_output(0, out); in MatrixSolveShapeFn()
154 c->set_output(0, out); in MatrixTriangularSolveShapeFn()
170 c->set_output(0, e_shape); in SelfAdjointEigV2ShapeFn()
176 c->set_output(1, v_shape); in SelfAdjointEigV2ShapeFn()
178 c->set_output(1, c->Vector(0ll)); in SelfAdjointEigV2ShapeFn()
202 c->set_output(0, lu_shape); in LuShapeFn()
203 c->set_output(1, p_shape); in LuShapeFn()
232 c->set_output(0, q_shape); in QrShapeFn()
[all …]
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 __anon52e493120302()
124 c->set_output(i, shape_handle); in __anon52e493120302()
203 c->set_output(0, c->UnknownShape()); in __anon52e493120702()
227 c->set_output(0, c->UnknownShape()); in __anon52e493120802()
238 c->set_output(0, shape_handle); in __anon52e493120802()
241 c->set_output(0, c->UnknownShape()); in __anon52e493120802()
383 c->set_output(0, out); in __anon52e493120902()
[all …]
Dfunctional_ops.cc44 c->set_output(i, handle_type->at(0).shape); in __anonb31eee6d0102()
46 c->set_output(i, c->UnknownShape()); in __anonb31eee6d0102()
49 c->set_output(i, c->input(i)); in __anonb31eee6d0102()
108 c->set_output(static_cast<int>(i), output_shape_handle); in IfShapeInferenceFn()
153 c->set_output(static_cast<int>(i), output_shape_handle); in CaseShapeInferenceFn()
189 c->set_output(i, c->input(i)); in __anonb31eee6d0202()
225 c->set_output(static_cast<int>(i), output_shape_handle); in WhileShapeInferenceFn()
229 c->set_output(i, c->input(i)); in WhileShapeInferenceFn()
310 c->set_output(0, out); in __anonb31eee6d0302()
Dcandidate_sampling_ops.cc38 c->set_output(0, num_sampled_v); in CandidateSamplerShapeFn()
39 c->set_output(1, c->Matrix(batch_size, num_true)); in CandidateSamplerShapeFn()
40 c->set_output(2, num_sampled_v); in CandidateSamplerShapeFn()
160 c->set_output(0, v); in __anonafdb00d10202()
161 c->set_output(1, v); in __anonafdb00d10202()
162 c->set_output(2, v); in __anonafdb00d10202()
Dragged_conversion_ops.cc181 c->set_output(0, c->Matrix(num_values, dense_dims)); // indices in RaggedTensorToSparseShapeFn()
182 c->set_output(1, c->Vector(num_values)); // values in RaggedTensorToSparseShapeFn()
183 c->set_output(2, c->Vector(dense_dims)); // dense_shape in RaggedTensorToSparseShapeFn()
203 c->set_output(0, c->Vector(num_rows)); in RaggedTensorToVariantShapeFn()
205 c->set_output(0, c->Scalar()); in RaggedTensorToVariantShapeFn()
218 c->set_output(0, shape); in RaggedTensorToVariantGradientShapeFn()
236 c->set_output(i, c->UnknownShapeOfRank(1)); in RaggedTensorFromVariantShapeFn()
238 c->set_output(output_ragged_rank, c->UnknownShape()); in RaggedTensorFromVariantShapeFn()
281 c->set_output(0, output_shape_handle); in RaggedTensorToTensorShapeFn()
Dbatch_ops.cc84 TF_RETURN_IF_ERROR(c->set_output("batched_tensors", out_shapes)); in __anond951767c0102()
85 TF_RETURN_IF_ERROR(c->set_output("id", {c->Scalar()})); in __anond951767c0102()
86 TF_RETURN_IF_ERROR(c->set_output( in __anond951767c0102()
106 c->set_output(0, out_shape); in __anond951767c0202()
120 c->set_output(0, c->UnknownShapeOfRank(c->Rank(c->input(2)))); in __anond951767c0302()
/external/tensorflow/tensorflow/core/tpu/ops/
Dtpu_compile_op.cc40 c->set_output(0, c->Scalar()); in __anondb58f4d00102()
43 c->set_output(i + 1, c->Vector(3)); in __anondb58f4d00102()
67 c->set_output(0, c->Vector(3)); in __anondb58f4d00202()
100 c->set_output(0, c->Scalar()); in __anondb58f4d00302()
103 c->set_output(i + 1, c->Vector(3)); in __anondb58f4d00302()
107 c->set_output(num_computations + i + 1, c->Scalar()); in __anondb58f4d00302()
Dtopk_ops.cc34 c->set_output(0, s); in __anonee73ab400102()
52 c->set_output(0, s); in __anonee73ab400202()
53 c->set_output(1, s); in __anonee73ab400202()
63 c->set_output(0, input); in __anonee73ab400302()
81 c->set_output(0, s); in __anonee73ab400402()
82 c->set_output(1, s); in __anonee73ab400402()

12345678910