Searched refs:AbstractTensorHandleKind (Results 1 – 3 of 3) sorted by relevance
30 enum AbstractTensorHandleKind { kGraph, kMlir, kEager, kTfrt, kCustomDevice }; enum31 explicit AbstractTensorHandle(AbstractTensorHandleKind kind) : kind_(kind) {} in AbstractTensorHandle()45 AbstractTensorHandleKind getKind() const { return kind_; } in getKind()48 const AbstractTensorHandleKind kind_;
71 explicit ImmediateExecutionTensorHandle(AbstractTensorHandleKind kind) in ImmediateExecutionTensorHandle()
59 explicit TracingTensorHandle(AbstractTensorHandleKind kind) in TracingTensorHandle()