TF_Graph* graph, const TF_Operation& op, const char* mutation_type) {

mutex_lock l(graph->mu);

RecordMutation(graph, op, mutation_type);

Node* node = &output.oper->node;

tensorflow::core::CppShapeInferenceResult::HandleData handle_data;

handle_data.set_is_set(true);

{

mutex_lock l(graph->mu);

tensorflow::shape_inference::InferenceContext* ic =

graph->refiner.GetContext(node);

CHECK(ic != nullptr);

CHECK_LT(output.index, ic->num_outputs());

const auto* shapes_and_types =

ic->output_handle_shapes_and_types(output.index);

if (shapes_and_types == nullptr) return "";

for (const auto& p : *shapes_and_types) {

auto* out_shape_and_type = handle_data.add_shape_and_type();

ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape());

out_shape_and_type->set_dtype(p.dtype);

*out_shape_and_type->mutable_type() = p.type;

}

}

string result;

handle_data.SerializeToString(&result);

return result;