Searched refs:inference_ctx (Results 1 – 3 of 3) sorted by relevance
114 TFE_OpInferenceContext* inference_ctx) in TFE_Op()116 inference_ctx(inference_ctx) {} in TFE_Op()119 std::unique_ptr<TFE_OpInferenceContext> inference_ctx; member
238 TFE_OpInferenceContext* ictx = op->inference_ctx.get(); in OpInferSingleInputAttrs()247 op->inference_ctx.reset(nullptr); in OpInferSingleInputAttrs()262 TFE_OpInferenceContext* ictx = op->inference_ctx.get(); in OpInferSingleTypeInputListAttrs()277 TFE_OpInferenceContext* ictx = op->inference_ctx.get(); in OpInferMixedTypeInputListAttrs()294 TFE_OpInferenceContext* ictx = op->inference_ctx.get(); in OpInferInputListAttrs()616 if (op->inference_ctx) { in TFE_OpAddInput()626 if (op->inference_ctx) { in TFE_OpAddInputList()
1765 CHECK(concatOp->inference_ctx); in TEST()1768 EXPECT_FALSE(concatOp->inference_ctx) << "Inference context is still present"; in TEST()