| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| sort_ops.cc | 28 context->SetOutput(0, xla::Sort(context->Input("input")));
42 context->SetOutput(0, xla::GetTupleElement(result, 0));
43 context->SetOutput(1, xla::GetTupleElement(result, 1));
|
| qr_op.cc | 34 ctx->SetOutput(0, result.ValueOrDie().q);
35 ctx->SetOutput(1, result.ValueOrDie().r);
|
| cholesky_op.cc | 28 ctx->SetOutput(0,
|
| xla_self_adjoint_eig_op.cc | 34 ctx->SetOutput(0, result.w);
35 ctx->SetOutput(1, result.v);
55 ctx->SetOutput(0, result.w);
56 ctx->SetOutput(1, result.v);
|
| const_op.cc | 53 ctx->SetOutput(
61 ctx->SetOutput(0, xla::Broadcast(xla::ConstantR0<float>(
69 ctx->SetOutput(0, xla::Broadcast(xla::ConstantR0<double>(
77 ctx->SetOutput(
88 ctx->SetOutput(
99 ctx->SetOutput(
107 ctx->SetOutput(0, xla::Broadcast(xla::ConstantR0<int64>(
|
| tile_ops.cc | 61 ctx->SetOutput(0, input);
78 ctx->SetOutput(0, input);
98 ctx->SetOutput(0, xla::Or(broadcasted_zero, input));
100 ctx->SetOutput(0, xla::Add(broadcasted_zero, input));
107 ctx->SetOutput(0, result.ValueOrDie());
|
| batch_norm_op.cc | 67 ctx->SetOutput(0, xla::ConvertElementType(xla::GetTupleElement(output, 0),
69 ctx->SetOutput(1, xla::GetTupleElement(output, 1));
70 ctx->SetOutput(2, xla::GetTupleElement(output, 2));
76 ctx->SetOutput(3, xla::GetTupleElement(output, 1));
84 ctx->SetOutput(4, xla::Rsqrt(xla::Add(
87 ctx->SetOutput(4, xla::GetTupleElement(output, 2));
93 ctx->SetOutput(0, xla::ConvertElementType(output, input_type));
95 ctx->SetOutput(1, ctx->Input(3));
96 ctx->SetOutput(2, ctx->Input(4));
97 ctx->SetOutput(3, ctx->Input(3))
[all...] |
| xla_svd_op.cc | 44 ctx->SetOutput(0, result.d);
45 ctx->SetOutput(1, result.u);
46 ctx->SetOutput(2, result.v);
70 ctx->SetOutput(0, result.d);
81 ctx->SetOutput(1, result.u);
82 ctx->SetOutput(2, result.v);
|
| check_numerics_op.cc | 40 ctx->SetOutput(0, ctx->Input(0));
|
| relu_op.cc | 35 ctx->SetOutput(0, xla::Max(zero, ctx->Input(0)));
48 ctx->SetOutput(0, xla::Clamp(zero, ctx->Input(0), six));
62 ctx->SetOutput(0, output);
79 ctx->SetOutput(0, xla::Select(pred, ctx->Input(0), zero));
99 ctx->SetOutput(0, out);
115 ctx->SetOutput(0, output);
|
| pad_op.cc | 50 ctx->SetOutput(0, input);
76 ctx->SetOutput(0, xla::Pad(input, ctx->Input("constant_values"), config));
79 ctx->SetOutput(0, xla::Pad(input, zero, config));
|
| batch_matmul_op.cc | 38 ctx->SetOutput(0, result);
|
| broadcast_to_op.cc | 37 context->SetOutput(0, output.ValueOrDie());
|
| elu_op.cc | 38 ctx->SetOutput(0, xla::Select(pred, ctx->Input(0), expm1));
55 ctx->SetOutput(0, xla::Select(pred, grad, exp_grad));
75 ctx->SetOutput(0, xla::Select(pred, xla::Mul(scale, ctx->Input(0)),
97 ctx->SetOutput(0, xla::Select(pred, lin_grad, exp_grad));
|
| fake_param_op.cc | 40 ctx->SetOutput(0, xla::Zeros(b, shape_));
|
| matrix_triangular_solve_op.cc | 39 ctx->SetOutput(0, result);
|
| reverse_op.cc | 49 ctx->SetOutput(0, ctx->Input(0));
63 ctx->SetOutput(0, xla::Rev(ctx->Input(0), dimensions));
88 ctx->SetOutput(0, ctx->Input(0));
115 ctx->SetOutput(0, xla::Rev(ctx->Input(0), axes));
|
| xla_broadcast_helper_op.cc | 58 context->SetOutput(0, lhs);
59 context->SetOutput(1, rhs);
97 context->SetOutput(0, lhs);
98 context->SetOutput(1, rhs);
|
| topk_op.cc | 52 context->SetOutput(0, xla::GetTupleElement(output_tuple, 0));
53 context->SetOutput(1, xla::GetTupleElement(output_tuple, 1));
|
| einsum_op.cc | 43 ctx->SetOutput(0, xla::Einsum(lhs, rhs, equation_));
|
| empty_op.cc | 54 ctx->SetOutput(0, result);
|
| fill_op.cc | 51 ctx->SetOutput(0, result);
|
| transpose_op.cc | 89 ctx->SetOutput(0, xla::Conj(transposed));
91 ctx->SetOutput(0, transposed);
156 ctx->SetOutput(0, xla::ConstantR1<int32>(ctx->builder(), output));
165 ctx->SetOutput(0, result.ValueOrDie());
|
| xla_dequantize_op.cc | 46 context->SetOutput(0, output);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_op_registry_test.cc | 31 ctx->SetOutput(0, ctx->Input(0));
40 ctx->SetOutput(0, ctx->Input(0));
|
