Home
last modified time | relevance | path

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

123456789

/external/tensorflow/tensorflow/core/ops/
Dcudnn_rnn_ops.cc61 c->set_output(0, c->Vector(1)); in __anon4fa58c430202()
100 c->set_output(0, output_shape); in __anon4fa58c430302()
101 c->set_output(1, output_h_shape); in __anon4fa58c430302()
102 c->set_output(2, output_c_shape); in __anon4fa58c430302()
103 c->set_output(3, c->UnknownShape()); in __anon4fa58c430302()
143 c->set_output(0, output_shape); in __anon4fa58c430402()
144 c->set_output(1, output_h_shape); in __anon4fa58c430402()
145 c->set_output(2, output_c_shape); in __anon4fa58c430402()
146 c->set_output(3, c->UnknownShape()); in __anon4fa58c430402()
147 c->set_output(4, c->UnknownShape()); in __anon4fa58c430402()
[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 __anon5983aeac0202()
58 c->set_output(1, c->Vector(c->Dim(b_indices, 0))); in __anon5983aeac0202()
78 c->set_output( in __anon5983aeac0302()
80 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anon5983aeac0302()
81 c->set_output(2, a_shape); in __anon5983aeac0302()
116 c->set_output(0, c->Matrix(output_left, output_right)); in __anon5983aeac0402()
132 c->set_output(0, c->Vector(3)); in __anon5983aeac0502()
148 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 3)); in __anon5983aeac0602()
[all …]
Drnn_ops.cc46 c->set_output(i, output); in __anon95c2132c0102()
78 c->set_output(0, x); in __anon95c2132c0202()
79 c->set_output(1, batch_cell_shape); in __anon95c2132c0202()
80 c->set_output(2, batch_cell_shape); in __anon95c2132c0202()
81 c->set_output(3, c->Matrix(batch_size, twice_cell_size)); in __anon95c2132c0202()
114 c->set_output(i, output); in __anon95c2132c0302()
154 c->set_output(0, c->Matrix(batch_size, cell_size)); in __anon95c2132c0402()
155 c->set_output(1, c->Matrix(batch_size, cell_size_times_4)); in __anon95c2132c0402()
156 c->set_output(2, cell_size_vec); in __anon95c2132c0402()
157 c->set_output(3, cell_size_vec); in __anon95c2132c0402()
[all …]
Dboosted_trees_ops.cc39 c->set_output(0, c->Scalar()); in __anon9fefc8f00102()
89 TF_RETURN_IF_ERROR(c->set_output("node_ids_list", output_shapes_vec)); in __anon9fefc8f00202()
90 TF_RETURN_IF_ERROR(c->set_output("gains_list", output_shapes_vec)); in __anon9fefc8f00202()
91 TF_RETURN_IF_ERROR(c->set_output("thresholds_list", output_shapes_vec)); in __anon9fefc8f00202()
95 c->set_output("left_node_contribs_list", output_shapes_contribs)); in __anon9fefc8f00202()
97 c->set_output("right_node_contribs_list", output_shapes_contribs)); in __anon9fefc8f00202()
130 c->set_output(0, rank_1_output_shape); in __anon9fefc8f00302()
131 c->set_output(1, rank_1_output_shape); in __anon9fefc8f00302()
132 c->set_output(2, rank_1_output_shape); in __anon9fefc8f00302()
133 c->set_output(3, rank_1_output_shape); in __anon9fefc8f00302()
[all …]
Dstring_ops.cc48 c->set_output(0, c->input(0)); in __anon662f88150102()
67 c->set_output(0, c->input(0)); in __anon662f88150202()
149 c->set_output(0, c->Scalar()); in __anon662f88150302()
165 c->set_output(0, c->Scalar()); in __anon662f88150402()
179 c->set_output(0, out); in __anon662f88150402()
195 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anon662f88150502()
196 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anon662f88150502()
197 c->set_output(2, c->Vector(2)); in __anon662f88150502()
213 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anon662f88150602()
214 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anon662f88150602()
[all …]
Dctc_ops.cc60 c->set_output(0, c->Vector(batch_size)); in __anoned3a12980102()
61 c->set_output(1, inputs); in __anoned3a12980102()
97 c->set_output(0, c->Vector(batch_size)); in __anoned3a12980202()
98 c->set_output(1, inputs); in __anoned3a12980202()
124 c->set_output(0, c->Matrix(total_decoded_outputs, 2)); in __anoned3a12980302()
125 c->set_output(1, c->Vector(total_decoded_outputs)); in __anoned3a12980302()
126 c->set_output(2, c->Vector(2)); in __anoned3a12980302()
127 c->set_output(3, c->Matrix(batch_size, 1)); in __anoned3a12980302()
160 c->set_output(out_idx++, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anoned3a12980402()
163 c->set_output(out_idx++, c->Vector(InferenceContext::kUnknownDim)); in __anoned3a12980402()
[all …]
Dnn_ops.cc60 c->set_output(0, c->MakeShape(output_dims)); in FractionalPoolShapeFn()
61 c->set_output(1, c->Vector(output_dims[1])); in FractionalPoolShapeFn()
62 c->set_output(2, c->Vector(output_dims[2])); in FractionalPoolShapeFn()
93 c->set_output(0, s); in __anon87c0335a0202()
123 c->set_output(0, out); in __anon87c0335a0302()
157 c->set_output(0, dx); in __anon87c0335a0402()
160 c->set_output(1, vector_shape); in __anon87c0335a0402()
161 c->set_output(2, vector_shape); in __anon87c0335a0402()
162 c->set_output(3, vector_shape); in __anon87c0335a0402()
163 c->set_output(4, vector_shape); in __anon87c0335a0402()
[all …]
Dmkl_nn_ops.cc193 c->set_output(1, c->Scalar()); in __anonfe9dedfd0102()
194 c->set_output(2, c->Scalar()); in __anonfe9dedfd0102()
241 c->set_output(1, c->Scalar()); in __anonfe9dedfd0202()
242 c->set_output(2, c->Scalar()); in __anonfe9dedfd0202()
292 c->set_output(1, c->Scalar()); in __anonfe9dedfd0302()
293 c->set_output(2, c->Scalar()); in __anonfe9dedfd0302()
338 c->set_output(1, channel); in __anonfe9dedfd0402()
339 c->set_output(2, channel); in __anonfe9dedfd0402()
389 c->set_output(1, c->Scalar()); in __anonfe9dedfd0502()
390 c->set_output(2, c->Scalar()); in __anonfe9dedfd0502()
[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()
155 c->set_output(0, c->Vector(c->UnknownDim())); in SoftNMSShapeFn()
156 c->set_output(1, c->Vector(c->UnknownDim())); in SoftNMSShapeFn()
217 c->set_output(0, c->MakeShape({batch_dim, output_size, 4})); in CombinedNMSShapeFn()
218 c->set_output(1, c->MakeShape({batch_dim, output_size})); in CombinedNMSShapeFn()
219 c->set_output(2, c->MakeShape({batch_dim, output_size})); 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 __anonc2c841560202()
111 c->set_output(0, c->UnknownShape()); in __anonc2c841560202()
115 c->set_output(0, first_input); in __anonc2c841560202()
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 __anonc2c841560402()
183 c->set_output(0, c->input(0)); in __anonc2c841560402()
Darray_ops.cc76 c->set_output(0, c->MakeShape(dims)); in PadKnown()
102 c->set_output(0, c->UnknownShapeOfRank(c->Value(n_dim))); in PadShapeFn()
104 c->set_output(0, c->UnknownShape()); in PadShapeFn()
129 c->set_output(0, c->UnknownShape()); in TransposeShapeFn()
146 c->set_output(0, input); in TransposeShapeFn()
182 c->set_output(0, c->MakeShape(dims)); in TransposeShapeFn()
193 c->set_output(0, out); in SetOutputShapeForReshape()
278 c->set_output(0, out); in SetOutputShapeForReshape()
318 c->set_output(0, passed_shape); in __anonf6523ebd0202()
338 c->set_output(0, c->UnknownShape()); in __anonf6523ebd0302()
[all …]
Dlinalg_ops.cc46 c->set_output(0, out); in BatchUnchangedSquareShapeFn()
83 c->set_output(0, out); in MatrixSolveShapeFn()
111 c->set_output(0, out); in MatrixTriangularSolveShapeFn()
127 c->set_output(0, e_shape); in SelfAdjointEigV2ShapeFn()
133 c->set_output(1, v_shape); in SelfAdjointEigV2ShapeFn()
135 c->set_output(1, c->Vector(0ll)); in SelfAdjointEigV2ShapeFn()
159 c->set_output(0, lu_shape); in LuShapeFn()
160 c->set_output(1, p_shape); in LuShapeFn()
189 c->set_output(0, q_shape); in QrShapeFn()
190 c->set_output(1, r_shape); in QrShapeFn()
[all …]
Dmath_ops.cc41 c->set_output(0, cur); in __anonb5dc68880102()
680 c->set_output(0, output); in __anonb5dc68880202()
726 c->set_output(0, output); \
818 c->set_output(0, data); in __anonb5dc68880402()
830 c->set_output(0, data); in __anonb5dc68880402()
838 c->set_output(0, data); in __anonb5dc68880402()
845 c->set_output(0, data); in __anonb5dc68880402()
853 c->set_output(0, data); in __anonb5dc68880402()
857 c->set_output(0, data); in __anonb5dc68880402()
909 c->set_output(0, output); in __anonb5dc68880502()
[all …]
Dlist_ops.cc58 c->set_output(0, c->Scalar()); in __anonef6dc5da0202()
76 c->set_output(0, c->Scalar()); in __anonef6dc5da0302()
122 c->set_output(0, input_handles); in __anonef6dc5da0402()
191 c->set_output(1, tensor_shape); in __anonef6dc5da0502()
192 c->set_output(0, c->Scalar()); in __anonef6dc5da0502()
241 c->set_output(0, result); in __anonef6dc5da0602()
273 c->set_output(0, result); in TensorListConcatShapeInference()
275 c->set_output(0, c->UnknownShape()); in TensorListConcatShapeInference()
277 c->set_output(1, c->MakeShape({c->UnknownDim()})); in TensorListConcatShapeInference()
319 c->set_output(0, c->Scalar()); in __anonef6dc5da0902()
[all …]
Dset_ops.cc85 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank)); in __anon0af7a88a0102()
86 c->set_output(1, c->Vector(c->UnknownDim())); in __anon0af7a88a0102()
87 c->set_output(2, c->Vector(output_rank)); in __anon0af7a88a0102()
129 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank_dim)); in __anon0af7a88a0202()
130 c->set_output(1, c->Vector(c->UnknownDim())); in __anon0af7a88a0202()
131 c->set_output(2, c->Vector(output_rank_dim)); in __anon0af7a88a0202()
185 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank_dim)); in __anon0af7a88a0302()
186 c->set_output(1, c->Vector(c->UnknownDim())); in __anon0af7a88a0302()
187 c->set_output(2, c->Vector(output_rank_dim)); in __anon0af7a88a0302()
Dfunctional_ops.cc44 c->set_output(i, handle_type->at(0).shape); in __anon4a8e12910102()
46 c->set_output(i, c->UnknownShape()); in __anon4a8e12910102()
49 c->set_output(i, c->input(i)); in __anon4a8e12910102()
108 c->set_output(static_cast<int>(i), output_shape_handle); in IfShapeInferenceFn()
162 c->set_output(static_cast<int>(i), output_shape_handle); in __anon4a8e12910202()
177 c->set_output(i, c->input(i)); in __anon4a8e12910302()
213 c->set_output(static_cast<int>(i), output_shape_handle); in WhileShapeInferenceFn()
217 c->set_output(i, c->input(i)); in WhileShapeInferenceFn()
292 c->set_output(0, out); in __anon4a8e12910402()
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 __anon2d93d3360302()
124 c->set_output(i, shape_handle); in __anon2d93d3360302()
203 c->set_output(0, c->UnknownShape()); in __anon2d93d3360702()
227 c->set_output(0, c->UnknownShape()); in __anon2d93d3360802()
238 c->set_output(0, shape_handle); in __anon2d93d3360802()
241 c->set_output(0, c->UnknownShape()); in __anon2d93d3360802()
383 c->set_output(0, out); in __anon2d93d3360902()
[all …]
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 __anon27ae08f50202()
161 c->set_output(1, v); in __anon27ae08f50202()
162 c->set_output(2, v); in __anon27ae08f50202()
Dbatch_ops.cc65 TF_RETURN_IF_ERROR(c->set_output("batched_tensors", out_shapes)); in __anon8ccc74200102()
66 TF_RETURN_IF_ERROR(c->set_output("id", {c->Scalar()})); in __anon8ccc74200102()
67 TF_RETURN_IF_ERROR(c->set_output( in __anon8ccc74200102()
87 c->set_output(0, out_shape); in __anon8ccc74200202()
101 c->set_output(0, c->UnknownShapeOfRank(c->Rank(c->input(2)))); in __anon8ccc74200302()
Dsparse_csr_matrix_ops.cc81 c->set_output(0, c->Scalar()); in __anond96a6a670102()
105 c->set_output(0, indices); in __anond96a6a670202()
106 c->set_output(1, values); in __anond96a6a670202()
107 c->set_output(2, dense_shape); in __anond96a6a670202()
147 c->set_output(0, c->Scalar()); in __anond96a6a670302()
163 c->set_output(0, sparse_matrix); in __anond96a6a670402()
193 c->set_output(0, row_ptrs); in __anond96a6a670502()
194 c->set_output(1, c->Vector(c->UnknownDim())); in __anond96a6a670502()
195 c->set_output(2, c->Vector(c->UnknownDim())); in __anond96a6a670502()
217 c->set_output(0, out); in __anond96a6a670602()
[all …]
Dmkl_array_ops.cc56 c->set_output(1, c->Scalar()); in __anon42671a800102()
57 c->set_output(2, c->Scalar()); in __anon42671a800102()
86 c->set_output(1, c->Scalar()); in __anon42671a800202()
87 c->set_output(2, c->Scalar()); in __anon42671a800202()
130 c->set_output(1, minmax); in __anon42671a800302()
131 c->set_output(2, minmax); in __anon42671a800302()
Dragged_conversion_ops.cc171 c->set_output(0, c->Matrix(num_values, dense_dims)); // indices in RaggedTensorToSparseShapeFn()
172 c->set_output(1, c->Vector(num_values)); // values in RaggedTensorToSparseShapeFn()
173 c->set_output(2, c->Vector(dense_dims)); // dense_shape in RaggedTensorToSparseShapeFn()
193 c->set_output(0, c->Vector(num_rows)); in RaggedTensorToVariantShapeFn()
195 c->set_output(0, c->Scalar()); in RaggedTensorToVariantShapeFn()
218 c->set_output(i, c->UnknownShapeOfRank(1)); in RaggedTensorFromVariantShapeFn()
220 c->set_output(output_ragged_rank, c->UnknownShape()); in RaggedTensorFromVariantShapeFn()
263 c->set_output(0, output_shape_handle); in RaggedTensorToTensorShapeFn()
Dstateful_random_ops.cc30 c->set_output(0, out); in StatefulRandomShape()
68 c->set_output(0, out); in __anonefb820860102()
93 c->set_output(0, out); in __anonefb820860302()
112 c->set_output(0, out); in __anonefb820860402()
131 c->set_output(0, out); in __anonefb820860502()
Dfunction_ops.cc56 context->set_output(0, shape_handle); in __anona02766840102()
61 context->set_output(0, context->UnknownShape()); in __anona02766840102()
75 context->set_output(0, shape_handle); in __anona02766840102()
77 context->set_output(0, context->UnknownShape()); in __anona02766840102()
108 context->set_output(0, context->UnknownShape()); in __anona02766840202()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_testutil_test.cc71 c->set_output(0, c->input(0)); in TEST()
75 c->set_output(0, c->input(2)); in TEST()
80 c->set_output(i, c->UnknownShape()); in TEST()
85 c->set_output(0, c->Matrix(1, 2)); in TEST()
89 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in TEST()
141 c->set_output(0, c->MakeShape({c->Dim(c->input(0), 1), c->MakeDim(2), in TEST()

123456789