Searched refs:shape_ctx (Results 1 – 2 of 2) sorted by relevance
327 shape_inference::InferenceContext* shape_ctx = in InferOutputShapes() local331 shape_ctx->MakeShapeFromTensorShape(dtoh_shapes.at(i), &handle)); in InferOutputShapes()332 shape_ctx->set_output(i, handle); in InferOutputShapes()343 shape_inference::InferenceContext* shape_ctx = in InferOutputShapes() local346 shape_inference::ShapeHandle handle = shape_ctx->input(i); in InferOutputShapes()347 if (!shape_ctx->FullyDefined(handle)) { in InferOutputShapes()354 shape_ctx->ShapeHandleToProto(handle, &shape_proto); in InferOutputShapes()
804 if (auto* shape_ctx = refiner_.GetContext(merge)) { in BuildIfNode() local806 shape_ctx->ShapeHandleToProto(shape_ctx->output(0), &shape); in BuildIfNode()