Searched refs:TracingOperation (Results 1 – 8 of 8) sorted by relevance
73 class TracingOperation : public AbstractOperation {75 explicit TracingOperation(AbstractOperationKind kind) in TracingOperation() function92 void operator()(TracingOperation* p) const { in operator()101 std::unique_ptr<TracingOperation, internal::TracingOperationDeleter>;
26 if (isa<TracingOperation>(op)) { in MaybeSetOpName()27 TF_RETURN_IF_ERROR(dyn_cast<TracingOperation>(op)->SetOpName(op_name)); in MaybeSetOpName()
108 using tensorflow::tracing::TracingOperation;202 TracingOperation* tracing_op = dyn_cast<TracingOperation>(unwrap(op)); in TF_AbstractOpSetOpName()
47 using tracing::TracingOperation;180 if (isa<TracingOperation>(check_numerics_op.get())) { in TEST_P()181 s = dyn_cast<TracingOperation>(check_numerics_op.get()) in TEST_P()
40 if (isa<tracing::TracingOperation>(op.get())) { in ZerosLike()41 TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingOperation>(op.get())->SetOpName( in ZerosLike()193 if (isa<tracing::TracingOperation>(op.get())) { in BuildOnesLike()194 TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingOperation>(op.get())->SetOpName( in BuildOnesLike()
95 class GraphOperation : public TracingOperation {97 explicit GraphOperation(TF_Graph* g) : TracingOperation(kGraph), g_(g) {} in GraphOperation()355 TracingOperation* CreateOperation() override { in CreateOperation()
55 using tensorflow::tracing::TracingOperation;195 if (isa<TracingOperation>(self)) { in PYBIND11_MODULE()196 auto tracing_op = reinterpret_cast<TracingOperation*>(self); in PYBIND11_MODULE()
74 using tensorflow::tracing::TracingOperation;130 class MlirAbstractOp : public TracingOperation {134 : TracingOperation(kMlir), in MlirAbstractOp()