Home
last modified time | relevance | path

Searched refs:AbstractOperationPtr (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/ops/
Dmath_ops.cc31 AbstractOperationPtr mul_op(ctx->CreateOperation()); in Mul()
49 AbstractOperationPtr conj_op(ctx->CreateOperation()); in Conj()
65 AbstractOperationPtr add_op(ctx->CreateOperation()); in Add()
78 AbstractOperationPtr sub_op(ctx->CreateOperation()); in Sub()
93 AbstractOperationPtr matmul_op(ctx->CreateOperation()); in MatMul()
109 AbstractOperationPtr neg_op(ctx->CreateOperation()); in Neg()
120 AbstractOperationPtr sum_op(ctx->CreateOperation()); in Sum()
133 AbstractOperationPtr div_op(ctx->CreateOperation()); in Div()
147 AbstractOperationPtr div_op(ctx->CreateOperation()); in DivNoNan()
161 AbstractOperationPtr exp_op(ctx->CreateOperation()); in Exp()
[all …]
Darray_ops.cc28 AbstractOperationPtr identity_op(ctx->CreateOperation()); in Identity()
40 AbstractOperationPtr identity_n_op(ctx->CreateOperation()); in IdentityN()
52 AbstractOperationPtr z_op(ctx->CreateOperation()); in ZerosLike()
63 AbstractOperationPtr shape_op(ctx->CreateOperation()); in Shape()
75 AbstractOperationPtr op(ctx->CreateOperation()); in ExpandDims()
87 AbstractOperationPtr op(ctx->CreateOperation()); in OnesLike()
Dnn_ops.cc30 AbstractOperationPtr sm_loss_op(ctx->CreateOperation()); in SparseSoftmaxCrossEntropyWithLogits()
47 AbstractOperationPtr relugrad_op(ctx->CreateOperation()); in ReluGrad()
62 AbstractOperationPtr relu_op(ctx->CreateOperation()); in Relu()
75 AbstractOperationPtr bias_add_op(ctx->CreateOperation()); in BiasAdd()
92 AbstractOperationPtr bias_add_grad_op(ctx->CreateOperation()); in BiasAddGrad()
/external/tensorflow/tensorflow/python/framework/experimental/
Dunified_api.cc49 using tensorflow::AbstractOperationPtr;
184 py::class_<AbstractOperation, AbstractOperationPtr>(m, "AbstractOperation") in PYBIND11_MODULE()
/external/tensorflow/tensorflow/c/eager/
Dabstract_operation.h133 using AbstractOperationPtr = variable
Dgradients.cc38 AbstractOperationPtr op(ctx->CreateOperation()); in ZerosLike()
153 AbstractOperationPtr op(ctx_->CreateOperation()); in AggregateGradients()
191 AbstractOperationPtr op(ctx_->CreateOperation()); in BuildOnesLike()
Dunified_api_testutil.cc97 AbstractOperationPtr fn_op(ctx->CreateOperation()); in RunModel()
Dgradients_test.cc175 AbstractOperationPtr check_numerics_op(ctx->CreateOperation()); in TEST_P()
Dgradients_util.cc283 AbstractOperationPtr fn_op(ctx->CreateOperation()); in RunModel()