Searched refs:TF_StepId (Results 1 – 3 of 3) sorted by relevance
129 TF_CAPI_EXPORT extern int64_t TF_StepId(TF_OpKernelContext* ctx);
197 int64_t TF_StepId(TF_OpKernelContext* ctx) { in TF_StepId() function
55 EXPECT_EQ(43, TF_StepId(ctx)); in MyComputeFunc()