Searched refs:tfe_op (Results 1 – 1 of 1) sorted by relevance
604 void TFE_InferShapes(TFE_Op* tfe_op, TF_ShapeAndTypeList* input_shapes, in TFE_InferShapes() argument621 tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(tfe_op); in TFE_InferShapes()