Home
last modified time | relevance | path

Searched refs:MaybeSetOpName (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/ops/
Dmath_ops.cc24 using tensorflow::tracing::MaybeSetOpName;
33 TF_RETURN_IF_ERROR(MaybeSetOpName(mul_op.get(), name)); in Mul()
51 TF_RETURN_IF_ERROR(MaybeSetOpName(conj_op.get(), name)); in Conj()
67 TF_RETURN_IF_ERROR(MaybeSetOpName(add_op.get(), name)); in Add()
80 TF_RETURN_IF_ERROR(MaybeSetOpName(sub_op.get(), name)); in Sub()
95 TF_RETURN_IF_ERROR(MaybeSetOpName(matmul_op.get(), name)); in MatMul()
111 TF_RETURN_IF_ERROR(MaybeSetOpName(neg_op.get(), name)); in Neg()
122 TF_RETURN_IF_ERROR(MaybeSetOpName(sum_op.get(), name)); in Sum()
135 TF_RETURN_IF_ERROR(MaybeSetOpName(div_op.get(), name)); in Div()
149 TF_RETURN_IF_ERROR(MaybeSetOpName(div_op.get(), name)); in DivNoNan()
[all …]
Darray_ops.cc20 using tensorflow::tracing::MaybeSetOpName;
31 TF_RETURN_IF_ERROR(MaybeSetOpName(identity_op.get(), name)); in Identity()
43 TF_RETURN_IF_ERROR(MaybeSetOpName(identity_n_op.get(), name)); in IdentityN()
54 TF_RETURN_IF_ERROR(MaybeSetOpName(z_op.get(), name)); in ZerosLike()
65 TF_RETURN_IF_ERROR(MaybeSetOpName(shape_op.get(), name)); in Shape()
77 TF_RETURN_IF_ERROR(MaybeSetOpName(op.get(), name)); in ExpandDims()
89 TF_RETURN_IF_ERROR(MaybeSetOpName(op.get(), name)); in OnesLike()
Dnn_ops.cc21 using tensorflow::tracing::MaybeSetOpName;
33 TF_RETURN_IF_ERROR(MaybeSetOpName(sm_loss_op.get(), name)); in SparseSoftmaxCrossEntropyWithLogits()
50 TF_RETURN_IF_ERROR(MaybeSetOpName(relugrad_op.get(), name)); in ReluGrad()
64 TF_RETURN_IF_ERROR(MaybeSetOpName(relu_op.get(), name)); in Relu()
78 TF_RETURN_IF_ERROR(MaybeSetOpName(bias_add_op.get(), name)); in BiasAdd()
95 TF_RETURN_IF_ERROR(MaybeSetOpName(bias_add_grad_op.get(), name)); in BiasAddGrad()
/external/tensorflow/tensorflow/c/eager/
Dtracing_utils.cc25 Status MaybeSetOpName(AbstractOperation* op, const char* op_name) { in MaybeSetOpName() function
30 TF_RETURN_IF_ERROR(MaybeSetOpName( in MaybeSetOpName()
Dtracing_utils.h22 Status MaybeSetOpName(AbstractOperation*, const char* op_name);