/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | batch_norm_op.cc | 67 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 …]
|
D | const_op.cc | 53 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()
|
D | xla_svd_op.cc | 44 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()
|
D | tile_ops.cc | 61 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()
|
D | relu_op.cc | 35 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()
|
D | xla_self_adjoint_eig_op.cc | 34 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()
|
D | sort_ops.cc | 28 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()
|
D | random_ops.cc | 54 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()
|
D | reverse_op.cc | 49 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()
|
D | xla_broadcast_helper_op.cc | 58 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()
|
D | elu_op.cc | 38 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()
|
D | transpose_op.cc | 89 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()
|
D | pad_op.cc | 50 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()
|
D | shape_op.cc | 108 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()
|
D | softmax_op.cc | 81 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()
|
D | qr_op.cc | 34 ctx->SetOutput(0, result.ValueOrDie().q); in Compile() 35 ctx->SetOutput(1, result.ValueOrDie().r); in Compile()
|
D | diag_op.cc | 115 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()
|
D | fake_quantize_ops.cc | 120 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()
|
D | topk_op.cc | 52 context->SetOutput(0, xla::GetTupleElement(output_tuple, 0)); in Compile() 53 context->SetOutput(1, xla::GetTupleElement(output_tuple, 1)); in Compile()
|
D | conv_ops.cc | 57 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()
|
D | stateless_random_ops.cc | 93 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()
|
D | tensor_list_ops.cc | 52 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()
|
D | tensor_array_ops.cc | 234 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/ |
D | gen_proto_text_functions_lib.cc | 80 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/ |
D | xla_op_registry_test.cc | 31 ctx->SetOutput(0, ctx->Input(0)); in Compile() 40 ctx->SetOutput(0, ctx->Input(0)); in Compile()
|