Home
last modified time | relevance | path

Searched refs:SetOutput (Results 1 – 25 of 113) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc67 ctx->SetOutput(0, xla::ConvertElementType(xla::GetTupleElement(output, 0), in Compile()
69 ctx->SetOutput(1, xla::GetTupleElement(output, 1)); in Compile()
70 ctx->SetOutput(2, xla::GetTupleElement(output, 2)); in Compile()
76 ctx->SetOutput(3, xla::GetTupleElement(output, 1)); in Compile()
84 ctx->SetOutput(4, xla::Rsqrt(xla::Add( in Compile()
87 ctx->SetOutput(4, xla::GetTupleElement(output, 2)); in Compile()
93 ctx->SetOutput(0, xla::ConvertElementType(output, input_type)); in Compile()
95 ctx->SetOutput(1, ctx->Input(3)); in Compile()
96 ctx->SetOutput(2, ctx->Input(4)); in Compile()
97 ctx->SetOutput(3, ctx->Input(3)); in Compile()
[all …]
Dconst_op.cc53 ctx->SetOutput( in Compile()
61 ctx->SetOutput(0, xla::Broadcast(xla::ConstantR0<float>( in Compile()
69 ctx->SetOutput(0, xla::Broadcast(xla::ConstantR0<double>( in Compile()
77 ctx->SetOutput( in Compile()
88 ctx->SetOutput( in Compile()
99 ctx->SetOutput( in Compile()
107 ctx->SetOutput(0, xla::Broadcast(xla::ConstantR0<int64>( in Compile()
Dxla_svd_op.cc44 ctx->SetOutput(0, result.d); in Compile()
45 ctx->SetOutput(1, result.u); in Compile()
46 ctx->SetOutput(2, result.v); in Compile()
70 ctx->SetOutput(0, result.d); in Compile()
81 ctx->SetOutput(1, result.u); in Compile()
82 ctx->SetOutput(2, result.v); in Compile()
Dtile_ops.cc61 ctx->SetOutput(0, input); in Compile()
78 ctx->SetOutput(0, input); in Compile()
98 ctx->SetOutput(0, xla::Or(broadcasted_zero, input)); in Compile()
100 ctx->SetOutput(0, xla::Add(broadcasted_zero, input)); in Compile()
107 ctx->SetOutput(0, result.ValueOrDie()); in Compile()
Drelu_op.cc35 ctx->SetOutput(0, xla::Max(zero, ctx->Input(0))); in Compile()
48 ctx->SetOutput(0, xla::Clamp(zero, ctx->Input(0), six)); in Compile()
62 ctx->SetOutput(0, output); in Compile()
79 ctx->SetOutput(0, xla::Select(pred, ctx->Input(0), zero)); in Compile()
99 ctx->SetOutput(0, out); in Compile()
115 ctx->SetOutput(0, output); in Compile()
Dxla_self_adjoint_eig_op.cc34 ctx->SetOutput(0, result.w); in Compile()
35 ctx->SetOutput(1, result.v); in Compile()
55 ctx->SetOutput(0, result.w); in Compile()
56 ctx->SetOutput(1, result.v); in Compile()
Dsort_ops.cc28 context->SetOutput(0, xla::Sort(context->Input("input"))); in Compile()
42 context->SetOutput(0, xla::GetTupleElement(result, 0)); in Compile()
43 context->SetOutput(1, xla::GetTupleElement(result, 1)); in Compile()
Drandom_ops.cc54 ctx->SetOutput(0, result); in Compile()
80 ctx->SetOutput(0, input); in Compile()
142 ctx->SetOutput(0, curr); in Compile()
199 ctx->SetOutput(0, gather); in Compile()
230 ctx->SetOutput(0, xla::RngUniform(minval, maxval, xla_shape)); in Compile()
259 ctx->SetOutput(0, result); in Compile()
287 ctx->SetOutput(0, TruncatedNormal(uniform)); in Compile()
Dreverse_op.cc49 ctx->SetOutput(0, ctx->Input(0)); in Compile()
63 ctx->SetOutput(0, xla::Rev(ctx->Input(0), dimensions)); in Compile()
88 ctx->SetOutput(0, ctx->Input(0)); in Compile()
115 ctx->SetOutput(0, xla::Rev(ctx->Input(0), axes)); in Compile()
Dxla_broadcast_helper_op.cc58 context->SetOutput(0, lhs); in Compile()
59 context->SetOutput(1, rhs); in Compile()
97 context->SetOutput(0, lhs); in Compile()
98 context->SetOutput(1, rhs); in Compile()
Delu_op.cc38 ctx->SetOutput(0, xla::Select(pred, ctx->Input(0), expm1)); in Compile()
55 ctx->SetOutput(0, xla::Select(pred, grad, exp_grad)); in Compile()
75 ctx->SetOutput(0, xla::Select(pred, xla::Mul(scale, ctx->Input(0)), in Compile()
97 ctx->SetOutput(0, xla::Select(pred, lin_grad, exp_grad)); in Compile()
Dtranspose_op.cc89 ctx->SetOutput(0, xla::Conj(transposed)); in Compile()
91 ctx->SetOutput(0, transposed); in Compile()
156 ctx->SetOutput(0, xla::ConstantR1<int32>(ctx->builder(), output)); in Compile()
165 ctx->SetOutput(0, result.ValueOrDie()); in Compile()
Dpad_op.cc50 ctx->SetOutput(0, input); in Compile()
76 ctx->SetOutput(0, xla::Pad(input, ctx->Input("constant_values"), config)); in Compile()
79 ctx->SetOutput(0, xla::Pad(input, zero, config)); in Compile()
Dshape_op.cc108 ctx->SetOutput(0, size); in Compile()
154 ctx->SetOutput(0, xla::Reshape(ctx->Input("input"), new_shape)); in Compile()
212 ctx->SetOutput(0, xla::Reshape(ctx->Input(0), new_shape)); in Compile()
229 ctx->SetOutput(0, xla::Broadcast(zero, input_shape.dim_sizes())); in Compile()
243 ctx->SetOutput(0, xla::Broadcast(one, input_shape.dim_sizes())); in Compile()
Dsoftmax_op.cc81 ctx->SetOutput(0, softmax); in Compile()
168 ctx->SetOutput(0, loss); in Compile()
169 ctx->SetOutput(1, backprop); in Compile()
235 ctx->SetOutput(0, loss); in Compile()
236 ctx->SetOutput(1, backprop); in Compile()
Dqr_op.cc34 ctx->SetOutput(0, result.ValueOrDie().q); in Compile()
35 ctx->SetOutput(1, result.ValueOrDie().r); in Compile()
Ddiag_op.cc115 ctx->SetOutput(0, diag); in Compile()
157 ctx->SetOutput(0, output); in Compile()
186 ctx->SetOutput(0, diag); in Compile()
205 ctx->SetOutput(0, xla::GetMatrixDiagonal(input)); in Compile()
Dfake_quantize_ops.cc120 ctx->SetOutput(0, output); in Compile()
172 ctx->SetOutput(0, output); in Compile()
212 ctx->SetOutput(0, output); in Compile()
258 ctx->SetOutput(0, output0); in Compile()
267 ctx->SetOutput(1, output1); in Compile()
276 ctx->SetOutput(2, output2); in Compile()
Dtopk_op.cc52 context->SetOutput(0, xla::GetTupleElement(output_tuple, 0)); in Compile()
53 context->SetOutput(1, xla::GetTupleElement(output_tuple, 1)); in Compile()
Dconv_ops.cc57 ctx->SetOutput(0, conv.ValueOrDie()); in Compile()
110 ctx->SetOutput(0, in_backprop.ValueOrDie()); in Compile()
168 ctx->SetOutput(0, filter_backprop.ValueOrDie()); in Compile()
Dstateless_random_ops.cc93 ctx->SetOutput(0, uniform); in Compile()
141 ctx->SetOutput(0, uniform); in Compile()
182 ctx->SetOutput(0, normal); in Compile()
224 ctx->SetOutput(0, truncated_normal); in Compile()
Dtensor_list_ops.cc52 ctx->SetOutput(0, index); in Compile()
181 ctx->SetOutput(0, xla::ConstantR1<int64>(b, shape.dim_sizes())); in Compile()
188 ctx->SetOutput(0, xla::ConstantR1<int32>(b, size)); in Compile()
234 ctx->SetOutput(0, xla::Reshape(read, value_shape)); in Compile()
254 ctx->SetOutput(0, buffer); in Compile()
463 ctx->SetOutput(1, xla::Reshape(read, value_shape)); in Compile()
Dtensor_array_ops.cc234 ctx->SetOutput(0, flow); in Compile()
277 ctx->SetOutput(0, xla::Reshape(read, value_shape)); in Compile()
334 ctx->SetOutput(0, xla::Slice(ta, begin, end, strides)); in Compile()
344 ctx->SetOutput(0, gather); in Compile()
431 ctx->SetOutput(0, flow); in Compile()
464 ctx->SetOutput(0, xla::Reshape(ta, shape)); in Compile()
541 ctx->SetOutput(0, flow); in Compile()
/external/tensorflow/tensorflow/tools/proto_text/
Dgen_proto_text_functions_lib.cc80 Generator& SetOutput(Section* section) { in SetOutput() function in tensorflow::__anond395b3d80111::Generator
253 SetOutput(&cc_); in AppendFieldValueAppend()
358 SetOutput(&header_); in AppendEnumFunctions()
362 SetOutput(&cc_); in AppendEnumFunctions()
380 SetOutput(&header_).Print(sig, "\n TF_MUST_USE_RESULT;"); in AppendParseMessageFunction()
382 SetOutput(&cc_); in AppendParseMessageFunction()
415 SetOutput(&header_impl_).Print(sig, ";"); in AppendParseMessageFunction()
418 SetOutput(&cc_); in AppendParseMessageFunction()
613 SetOutput(&header_impl_).Print(); in AppendDebugStringFunctions()
614 SetOutput(&header_).Print().Print("// Message-text conversion for ", in AppendDebugStringFunctions()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry_test.cc31 ctx->SetOutput(0, ctx->Input(0)); in Compile()
40 ctx->SetOutput(0, ctx->Input(0)); in Compile()

12345