Searched refs:TFE_OpInferenceContext (Results 1 – 2 of 2) sorted by relevance
102 struct TFE_OpInferenceContext { struct103 explicit TFE_OpInferenceContext(const tensorflow::OpDef* op_def) in TFE_OpInferenceContext() argument114 TFE_OpInferenceContext* inference_ctx) in TFE_Op() argument119 std::unique_ptr<TFE_OpInferenceContext> inference_ctx;
238 TFE_OpInferenceContext* ictx = op->inference_ctx.get(); 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()578 new TFE_OpInferenceContext(op_def)); in TFE_NewOp()