Searched refs:control_message (Results 1 – 1 of 1) sorted by relevance
491 OpInfo::TensorProperties control_message; in MaybeUpdateInputOutput() local492 control_message.set_dtype(DT_FLOAT); in MaybeUpdateInputOutput()493 control_message.mutable_shape()->add_dim()->set_size(1); in MaybeUpdateInputOutput()494 auto* value = control_message.mutable_value(); in MaybeUpdateInputOutput()496 inputs.push_back(control_message); in MaybeUpdateInputOutput()497 outputs.push_back(control_message); in MaybeUpdateInputOutput()